Add Friend Feed Favoriting functionality (#8)

Author: Ethan Chrisp

callofduty/client.py

@@ -189,7 +189,7 @@ async def SetFeedReaction(
             "category": category,
         }
 
-        return await self.http.SetFeedReaction(reaction.value, json)
+        await self.http.SetFeedReaction(reaction.value, json)
 
     async def RemoveFeedReaction(
         self,
@@ -231,7 +231,91 @@ async def RemoveFeedReaction(
             "category": category,
         }
 
-        return await self.http.SetFeedReaction(Reaction.Remove.value, json)
+        await self.http.SetFeedReaction(Reaction.Remove.value, json)
+
+    async def SetFeedFavorite(
+        self,
+        platform: Platform,
+        username: str,
+        title: Title,
+        date: int,
+        category: str,
+    ) -> None:
+        """
+        Set a Call of Duty Friend Feed item as a favorite.
+
+        Parameters
+        ----------
+        platform : callofduty.Platform
+            Platform to get the player from.
+        username : str
+            Player's username for the designated platform.
+        title : callofduty.Title
+            Title of the feed item.
+        date : int
+            Timstamp of the feed item.
+        category : str
+            Category of the feed item.
+
+        Returns
+        -------
+        None
+        """
+
+        VerifyPlatform(platform)
+        VerifyTitle(title)
+
+        json: dict = {
+            "username": username,
+            "platform": platform.value,
+            "title": title.value,
+            "date": date,
+            "category": category,
+        }
+
+        await self.http.SetFeedFavorite(1, json)
+
+    async def RemoveFeedFavorite(
+        self,
+        platform: Platform,
+        username: str,
+        title: Title,
+        date: int,
+        category: str,
+    ) -> None:
+        """
+        Unset a Call of Duty Friend Feed item as a favorite.
+
+        Parameters
+        ----------
+        platform : callofduty.Platform
+            Platform to get the player from.
+        username : str
+            Player's username for the designated platform.
+        title : callofduty.Title
+            Title of the feed item.
+        date : int
+            Timstamp of the feed item.
+        category : str
+            Category of the feed item.
+
+        Returns
+        -------
+        None
+        """
+
+        VerifyPlatform(platform)
+        VerifyTitle(title)
+
+        json: dict = {
+            "username": username,
+            "platform": platform.value,
+            "title": title.value,
+            "date": date,
+            "category": category,
+        }
+
+        await self.http.SetFeedFavorite(0, json)
 
     async def GetMyIdentities(self) -> list:
         """

callofduty/feed.py

@@ -89,7 +89,41 @@ async def unreact(self) -> None:
         None
         """
 
-        return await self._client.RemoveFeedReaction(
+        await self._client.RemoveFeedReaction(
+            self.player.platform,
+            self.player.username,
+            self.title,
+            (self.date.timestamp() * 1000),
+            self.category,
+        )
+
+    async def favorite(self) -> None:
+        """
+        Set the Call of Duty Friend Feed item as a favorite.
+
+        Returns
+        -------
+        None
+        """
+
+        await self._client.SetFeedFavorite(
+            self.player.platform,
+            self.player.username,
+            self.title,
+            (self.date.timestamp() * 1000),
+            self.category,
+        )
+
+    async def unfavorite(self) -> None:
+        """
+        Unset the Call of Duty Friend Feed item as a favorite.
+
+        Returns
+        -------
+        None
+        """
+
+        await self._client.RemoveFeedFavorite(
             self.player.platform,
             self.player.username,
             self.title,

callofduty/http.py

@@ -175,6 +175,16 @@ async def SetFeedReaction(self, reaction: str, json: dict) -> Union[dict, str]:
             )
         )
 
+    async def SetFeedFavorite(self, set: int, json: dict) -> Union[dict, str]:
+        return await self.Send(
+            Request(
+                "POST",
+                f"api/papi-client/userfeed/v1/favorite/set/{set}/en",
+                baseUrl=Request.myBaseUrl,
+                json=json,
+            )
+        )
+
     async def GetMyIdentities(self) -> Union[dict, str]:
         return await self.Send(Request("GET", "api/papi-client/crm/cod/v2/identities/"))
 

test.py

@@ -96,6 +96,16 @@ async def main():
     #     print(item.text)
     #     await item.unreact()
 
+    # feed = await client.GetFriendFeed(limit=1)
+    # for item in feed:
+    #     print(item.text)
+    #     test = await item.favorite()
+
+    # feed = await client.GetFriendFeed(limit=1)
+    # for item in feed:
+    #     print(item.text)
+    #     await item.unfavorite()
+
     # maps = await client.GetAvailableMaps(Title.ModernWarfare)
     # for mapName in maps:
     #     print(mapName)