Add PartialUser update_shield_mode_status

Add PartialUser update_shield_mode_status method

Author: chillymosh

docs/changelog.rst

@@ -13,6 +13,7 @@ Master
         - Added ``igdb_ids`` argument to :func:`~twitchio.Client.fetch_games`
         - Added ``tags`` attribute to :class:`~twitchio.Stream`, :class:`~twitchio.ChannelInfo` and :class:`~twitchio.SearchUser`
         - Added :func:`~twitchio.PartialUser.fetch_shield_mode_status`
+        - Added :func:`~twitchio.PartialUser.update_shield_mode_status`
 
     - Bug fixes
         - Fix :func:`~twitchio.PartialUser.fetch_bits_leaderboard` not handling ``started_at`` and :class:`~twitchio.BitsLeaderboard` not correctly parsing

twitchio/http.py

@@ -1114,3 +1114,8 @@ async def get_shield_mode_status(self, token: str, broadcaster_id: str, moderato
         return await self.request(
             Route("GET", "moderation/shield_mode", query=q, token=token), paginate=False, full_body=False
         )
+
+    async def put_shield_mode_status(self, token: str, broadcaster_id: str, moderator_id: str, is_active: bool):
+        q = [("broadcaster_id", broadcaster_id), ("moderator_id", moderator_id)]
+        body = {"is_active": is_active}
+        return await self.request(Route("PUT", "moderation/shield_mode", query=q, body=body, token=token))

twitchio/user.py

@@ -1526,6 +1526,32 @@ async def fetch_shield_mode_status(self, token: str, moderator_id: int):
 
         return ShieldStatus(self._http, data[0])
 
+    async def update_shield_mode_status(self, token: str, moderator_id: int, is_active: bool):
+        """|coro|
+
+        Updates the user's Shield Mode activation status.
+
+        Parameters
+        -----------
+        token: :class:`str`
+            An oauth user token with the ``moderator:read:shield_mode`` or ``moderator:manage:shield_mode`` scope.
+        moderator_id: :class:`int`
+            The ID of the broadcaster or a user that is one of the broadcaster's moderators. This ID must match the user ID in the access token.
+        is_active: :class:`bool`
+            A Boolean value that determines whether to activate Shield Mode. Set to True to activate Shield Mode; otherwise, False to deactivate Shield Mode.
+
+        Returns
+        --------
+        :class:`twitchio.ShieldStatus`
+        """
+        from .models import ShieldStatus
+
+        data = await self._http.put_shield_mode_status(
+            broadcaster_id=str(self.id), moderator_id=str(moderator_id), is_active=is_active, token=token
+        )
+
+        return ShieldStatus(self._http, data[0])
+
 
 class BitLeaderboardUser(PartialUser):