diff --git a/stream_chat/async_chat/client.py b/stream_chat/async_chat/client.py index 7711cd35..573a26df 100644 --- a/stream_chat/async_chat/client.py +++ b/stream_chat/async_chat/client.py @@ -503,23 +503,42 @@ 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) diff --git a/stream_chat/client.py b/stream_chat/client.py index 5489035c..e8d0f302 100644 --- a/stream_chat/client.py +++ b/stream_chat/client.py @@ -480,23 +480,36 @@ 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)