From fc2f5fdd27feb584682f49ae9bcc204185e23786 Mon Sep 17 00:00:00 2001 From: Vishal Narkhede Date: Thu, 23 Jan 2025 23:04:58 +0100 Subject: [PATCH 1/2] add team parameter to blocklist endpoints --- stream_chat/async_chat/client.py | 25 +++++++++++++++---------- stream_chat/client.py | 25 +++++++++++++++---------- 2 files changed, 30 insertions(+), 20 deletions(-) diff --git a/stream_chat/async_chat/client.py b/stream_chat/async_chat/client.py index 7711cd3..30ec42e 100644 --- a/stream_chat/async_chat/client.py +++ b/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) diff --git a/stream_chat/client.py b/stream_chat/client.py index 5489035..bd1762c 100644 --- a/stream_chat/client.py +++ b/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) From dbd1742984084ac1db5b3c5e5f6f6e07370801e5 Mon Sep 17 00:00:00 2001 From: Vishal Narkhede Date: Thu, 23 Jan 2025 23:22:11 +0100 Subject: [PATCH 2/2] lint fix --- stream_chat/async_chat/client.py | 26 ++++++++++++++++++++------ stream_chat/client.py | 14 +++++++++++--- 2 files changed, 31 insertions(+), 9 deletions(-) diff --git a/stream_chat/async_chat/client.py b/stream_chat/async_chat/client.py index 30ec42e..573a26d 100644 --- a/stream_chat/async_chat/client.py +++ b/stream_chat/async_chat/client.py @@ -503,26 +503,40 @@ async def send_file( return await self._parse_response(response) async def create_blocklist( - self, name: str, words: Iterable[str], type: str = "word", team: Optional[str] = None + 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}, params=params + "blocklists", + data={"name": name, "words": words, "type": type}, + params=params, ) 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, team: Optional[str] = None) -> StreamResponse: + 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], team: Optional[str] = None) -> StreamResponse: + 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) + return await self.put( + f"blocklists/{name}", data={"words": words}, params=params + ) - async def delete_blocklist(self, name: str, team: Optional[str] = None) -> StreamResponse: + 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) diff --git a/stream_chat/client.py b/stream_chat/client.py index bd1762c..e8d0f30 100644 --- a/stream_chat/client.py +++ b/stream_chat/client.py @@ -480,11 +480,17 @@ def send_file( return self._parse_response(response) def create_blocklist( - self, name: str, words: Iterable[str], type: str = "word", team: Optional[str] = None + 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}, params=params + "blocklists", + data={"name": name, "words": words, "type": type}, + params=params, ) def list_blocklists(self, team: Optional[str] = None) -> StreamResponse: @@ -495,7 +501,9 @@ 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], team: Optional[str] = None) -> StreamResponse: + 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)