From 3640a0c678a74e8f681fa3f780f6aae03830a088 Mon Sep 17 00:00:00 2001 From: ferhat elmas Date: Fri, 17 Apr 2020 16:00:21 +0200 Subject: [PATCH] Support customizing base url --- stream_chat/client.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/stream_chat/client.py b/stream_chat/client.py index 4239dfb1..1679de32 100644 --- a/stream_chat/client.py +++ b/stream_chat/client.py @@ -31,7 +31,9 @@ def __init__(self, api_key, api_secret, timeout=6.0, **options): self.api_secret = api_secret self.timeout = timeout self.options = options - self.base_url = "https://chat-us-east-1.stream-io-api.com" + self.base_url = options.get( + "base_url", "https://chat-us-east-1.stream-io-api.com" + ) self.auth_token = jwt.encode( {"server": True}, self.api_secret, algorithm="HS256" ) @@ -111,7 +113,7 @@ def update_users_partial(self, updates): def update_user_partial(self, update): return self.update_users_partial([update]) - + def delete_user(self, user_id, **options): return self.delete("users/{}".format(user_id), options) @@ -183,7 +185,7 @@ def mark_all_read(self, user_id): def update_message(self, message): if message.get("id") is None: raise ValueError("message must have an id") - return self.post("messages/{}".format(message['id']), data={"message": message}) + return self.post("messages/{}".format(message["id"]), data={"message": message}) def delete_message(self, message_id, **options): return self.delete("messages/{}".format(message_id), options) @@ -301,7 +303,7 @@ def send_file(self, uri, url, name, user, content_type=None): headers["stream-auth-type"] = "jwt" headers["X-Stream-Client"] = get_user_agent() parts = urlparse(url) - if parts[0] == '': + if parts[0] == "": url = "file://" + url if content_type: file_tuple = (name, urllib.request.urlopen(url), content_type) @@ -312,5 +314,6 @@ def send_file(self, uri, url, name, user, content_type=None): params=self.get_default_params(), data={"user": json.dumps(user)}, files={"file": file_tuple}, - headers=headers) + headers=headers, + ) return self._parse_response(response)