feat: add premium members methods

Author: New-dev0

swibots/api/community/controllers/community_controller.py

@@ -33,9 +33,7 @@ async def get_community(self, community_id: str = "", username: str = ""):
         return self.client.build_object(Community, response.data.get("result"))
 
     async def update_community(self, community: Community):
-        response = await self.client.put(
-            BASE_PATH, data=community.to_json_request()
-        )
+        response = await self.client.put(BASE_PATH, data=community.to_json_request())
         return self.client.build_object(Community, response.data.get("result"))
 
     async def deduct_xp(
@@ -152,3 +150,54 @@ async def get_messaging_bots(
         return self.client.build_list(InstantMessaging, response.data.get("result"))
 
     # endregion
+
+    async def add_premium_member(
+        self,
+        user_id: str,
+        community_id: str,
+        channel_id: str = None,
+        group_id: str = None,
+    ):
+        data = {"userId": user_id, "communityId": community_id}
+        if channel_id:
+            data["channelId"] = channel_id
+        if group_id:
+            data["groupId"] = group_id
+        response = await self.client.post(
+            f"{BASE_PATH}/premium/member?{urlencode(data)}"
+        )
+        return response.data.get("success")
+
+    async def delete_premium_member(
+        self,
+        user_id: str,
+        community_id: str,
+        channel_id: str = None,
+        group_id: str = None,
+    ):
+        data = {"userId": user_id, "communityId": community_id}
+        if channel_id:
+            data["channelId"] = channel_id
+        if group_id:
+            data["groupId"] = group_id
+        response = await self.client.delete(
+            f"{BASE_PATH}/premium/member?{urlencode(data)}"
+        )
+        return response.data.get("success")
+
+    async def get_premium_members(
+        self,
+        user_id: str,
+        community_id: str,
+        channel_id: str = None,
+        group_id: str = None,
+    ):
+        data = {"userId": user_id, "communityId": community_id}
+        if channel_id:
+            data["channelId"] = channel_id
+        if group_id:
+            data["groupId"] = group_id
+        response = await self.client.get(
+            f"{BASE_PATH}/premium/member?{urlencode(data)}"
+        )
+        return response.data

swibots/api/community/methods/community_methods.py

@@ -198,4 +198,67 @@ async def update_community(self: "swibots.ApiClient", community: Community) -> C
         Returns:
             Community: new community response
         """
-        return await self.community_service.communities.update_community(community=community)
+        return await self.community_service.communities.update_community(community=community)
+
+    async def add_premium_member(self: "swibots.ApiClient", user_id: int,
+                               community_id: str,
+                               channel_id: str = None,
+                               group_id: str = None):
+        """Add premium member to the community
+
+        Args:
+            user_id (int): User id
+            community_id (str): Community id
+            channel_id (str, optional): Channel id
+            group_id (str, optional): Group id
+
+        Returns:
+            bool
+        """
+        return await self.community_service.communities.add_premium_member(
+            user_id=user_id,
+            community_id=community_id,
+            channel_id=channel_id,
+            group_id=group_id
+        )
+
+    async def delete_premium_member(self: "swibots.ApiClient", user_id: int,
+                               community_id: str,
+                               channel_id: str = None,
+                               group_id: str = None):
+        """Remove premium member to the community
+
+        Args:
+            user_id (int): User id
+            community_id (str): Community id
+            channel_id (str, optional): Channel id
+            group_id (str, optional): Group id
+
+        Returns:
+            bool
+        """
+        return await self.community_service.communities.delete_premium_member(
+            user_id=user_id,
+            community_id=community_id,
+            channel_id=channel_id,
+            group_id=group_id
+        )
+
+    async def get_premium_members(self: "swibots.ApiClient", user_id: int,
+                               community_id: str,
+                               channel_id: str = None,
+                               group_id: str = None):
+        """Get all premium members in the community
+
+        Args:
+            user_id (int): User id
+            community_id (str): Community id
+            channel_id (str, optional): Channel id
+            group_id (str, optional): Group id
+        """
+        return await self.community_service.communities.get_premium_members(
+            user_id=user_id,
+            community_id=community_id,
+            channel_id=channel_id,
+            group_id=group_id
+        )