favorite() and remove_favorite()

Author: Katistic

vrcpy/aobjects.py

@@ -8,6 +8,15 @@ async def author(self):
         resp = await self.client.api.call("/users/"+self.authorId)
         return User(self.client, resp["data"])
 
+    async def favorite(self):
+        '''
+        Returns favorite object
+        '''
+
+        resp = await self.client.api.call("/favorites", "POST", params={"type": types.FavoriteType.Avatar,\
+            "favoriteId": self.id})
+        return Favorite(resp["data"])
+
 ## User
 
 class LimitedUser(o.LimitedUser):
@@ -44,6 +53,15 @@ async def friend(self):
         resp = await self.client.api.call("/user/"+self.id+"/friendRequest", "POST")
         return o.Notification(self.client, resp["data"])
 
+    async def favorite(self):
+        '''
+        Returns favorite object
+        '''
+
+        resp = await self.client.api.call("/favorites", "POST", params={"type": types.FavoriteType.Friend,\
+            "favoriteId": self.id})
+        return Favorite(resp["data"])
+
 class User(o.User, LimitedUser):
     async def fetch_full(self):
         user = await LimitedUser.fetch_full(self)
@@ -60,6 +78,10 @@ async def friend(self):
         notif = await LimitedUser.friend()
         return notif
 
+    async def favorite(self):
+        resp = await LimitedUser.favorite(self)
+        return resp
+
 class CurrentUser(o.CurrentUser, User):
     obj = "CurrentUser"
 
@@ -77,6 +99,12 @@ async def unfriend(self):
     async def friend(self):
         raise AttributeError("'CurrentUser' object has no attribute 'friend'")
 
+    async def favorite(self):
+        raise AttributeError("'CurrentUser' object has no attribute 'favorite'")
+
+    async def remove_favorite(self, id):
+        resp = await self.client.api.call("/favorites/"+id, "DELETE")
+
     async def update_info(self, email=None, status=None,\
         statusDescription=None, bio=None, bioLinks=None):
 
@@ -153,6 +181,15 @@ async def author(self):
         resp = await self.client.api.call("/users/"+self.authorId)
         return User(self.client, resp["data"])
 
+    async def favorite(self):
+        '''
+        Returns favorite object
+        '''
+
+        resp = await self.client.api.call("/favorites", "POST", params={"type": types.FavoriteType.World,\
+            "favoriteId": self.id})
+        return Favorite(resp["data"])
+
 class World(o.World, LimitedWorld):
     async def author(self):
         resp = await super(LimitedWorld, self).author()
@@ -169,6 +206,10 @@ async def fetch_instance(self, id):
         resp = await self.client.api.call("/instances/"+self.id+":"+id)
         return Instance(self.client, resp["data"])
 
+    async def favorite(self):
+        resp = await LimitedWorld.favorite(self)
+        return resp
+
     async def __cinit__(self):
         instances = []
         for instance in self.instances:

vrcpy/objects.py

@@ -59,6 +59,15 @@ def author(self):
         resp = self.client.api.call("/users/"+self.authorId)
         return User(self.client, resp["data"])
 
+    def favorite(self):
+        '''
+        Returns favorite object
+        '''
+
+        resp = self.client.api.call("/favorites", "POST", params={"type": types.FavoriteType.Avatar,\
+            "favoriteId": self.id})
+        return Favorite(resp["data"])
+
     def __init__(self, client, obj):
         super().__init__(client)
         self.unique += [
@@ -98,7 +107,7 @@ def public_avatars(self):
 
     def unfriend(self):
         '''
-        Returns void
+        Unfriends user
         '''
 
         resp = self.client.api.call("/auth/user/friends/"+self.id, "DELETE")
@@ -111,6 +120,15 @@ def friend(self):
         resp = self.client.api.call("/user/"+self.id+"/friendRequest", "POST")
         return Notification(self.client, resp["data"])
 
+    def favorite(self):
+        '''
+        Returns favorite object
+        '''
+
+        resp = self.client.api.call("/favorites", "POST", params={"type": types.FavoriteType.Friend,\
+            "favoriteId": self.id})
+        return Favorite(resp["data"])
+
     def __init__(self, client, obj=None):
         super().__init__(client)
         self.unique += [
@@ -186,6 +204,12 @@ def fetch_favorites(self, t):
 
         return f
 
+    def remove_favorite(self, id):
+        resp = self.client.api.call("/favorites/"+id, "DELETE")
+
+    def favorite(self):
+        raise AttributeError("'CurrentUser' object has no attribute 'favorite'")
+
     def __cinit__(self):
         if hasattr(self, "currentAvatar"):
             self.currentAvatar = self.client.fetch_avatar(self.currentAvatar)
@@ -253,6 +277,15 @@ def author(self):
         resp = self.client.api.call("/users/"+self.authorId)
         return User(self.client, resp["data"])
 
+    def favorite(self):
+        '''
+        Returns favorite object
+        '''
+
+        resp = self.client.api.call("/favorites", "POST", params={"type": types.FavoriteType.World,\
+            "favoriteId": self.id})
+        return Favorite(resp["data"])
+
     def __init__(self, client, obj=None):
         super().__init__(client)
         self.unique += [