add team parameter to blocklist endpoints

Author: vishalnarkhede

stream_chat/async_chat/client.py

@@ -503,23 +503,28 @@ async def send_file(
             return await self._parse_response(response)
 
     async def create_blocklist(
-        self, name: str, words: Iterable[str], type: str = "word"
+        self, name: str, words: Iterable[str], type: str = "word", team: Optional[str] = None
     ) -> StreamResponse:
+        params = {"team": team} if team is not None else {}
         return await self.post(
-            "blocklists", data={"name": name, "words": words, "type": type}
+            "blocklists", data={"name": name, "words": words, "type": type}, params=params
         )
 
-    async def list_blocklists(self) -> StreamResponse:
-        return await self.get("blocklists")
+    async def list_blocklists(self, team: Optional[str] = None) -> StreamResponse:
+        params = {"team": team} if team is not None else {}
+        return await self.get("blocklists", params)
 
-    async def get_blocklist(self, name: str) -> StreamResponse:
-        return await self.get(f"blocklists/{name}")
+    async def get_blocklist(self, name: str, team: Optional[str] = None) -> StreamResponse:
+        params = {"team": team} if team is not None else {}
+        return await self.get(f"blocklists/{name}", params)
 
-    async def update_blocklist(self, name: str, words: Iterable[str]) -> StreamResponse:
-        return await self.put(f"blocklists/{name}", data={"words": words})
+    async def update_blocklist(self, name: str, words: Iterable[str], team: Optional[str] = None) -> StreamResponse:
+        params = {"team": team} if team is not None else {}
+        return await self.put(f"blocklists/{name}", data={"words": words}, params=params)
 
-    async def delete_blocklist(self, name: str) -> StreamResponse:
-        return await self.delete(f"blocklists/{name}")
+    async def delete_blocklist(self, name: str, team: Optional[str] = None) -> StreamResponse:
+        params = {"team": team} if team is not None else {}
+        return await self.delete(f"blocklists/{name}", params)
 
     async def check_push(self, push_data: Dict) -> StreamResponse:
         return await self.post("check_push", data=push_data)

stream_chat/client.py

@@ -480,23 +480,28 @@ def send_file(
         return self._parse_response(response)
 
     def create_blocklist(
-        self, name: str, words: Iterable[str], type: str = "word"
+        self, name: str, words: Iterable[str], type: str = "word", team: Optional[str] = None
     ) -> StreamResponse:
+        params = {"team": team} if team is not None else {}
         return self.post(
-            "blocklists", data={"name": name, "words": words, "type": type}
+            "blocklists", data={"name": name, "words": words, "type": type}, params=params
         )
 
-    def list_blocklists(self) -> StreamResponse:
-        return self.get("blocklists")
+    def list_blocklists(self, team: Optional[str] = None) -> StreamResponse:
+        params = {"team": team} if team is not None else {}
+        return self.get("blocklists", params)
 
-    def get_blocklist(self, name: str) -> StreamResponse:
-        return self.get(f"blocklists/{name}")
+    def get_blocklist(self, name: str, team: Optional[str] = None) -> StreamResponse:
+        params = {"team": team} if team is not None else {}
+        return self.get(f"blocklists/{name}", params)
 
-    def update_blocklist(self, name: str, words: Iterable[str]) -> StreamResponse:
-        return self.put(f"blocklists/{name}", data={"words": words})
+    def update_blocklist(self, name: str, words: Iterable[str], team: Optional[str] = None) -> StreamResponse:
+        params = {"team": team} if team is not None else {}
+        return self.put(f"blocklists/{name}", data={"words": words}, params=params)
 
-    def delete_blocklist(self, name: str) -> StreamResponse:
-        return self.delete(f"blocklists/{name}")
+    def delete_blocklist(self, name: str, team: Optional[str] = None) -> StreamResponse:
+        params = {"team": team} if team is not None else {}
+        return self.delete(f"blocklists/{name}", params)
 
     def check_push(self, push_data: Dict) -> StreamResponse:
         return self.post("check_push", data=push_data)