feat: add device fields (#107)

GetStream · Mar 28, 2022 · 3b582f5 · 3b582f5
1 parent 9710a90
commit 3b582f5
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 5 deletions.
diff --git a/stream_chat/async_chat/client.py b/stream_chat/async_chat/client.py
@@ -351,11 +351,20 @@ async def update_command(self, name: str, **settings: Any) -> StreamResponse:
         return await self.put(f"commands/{name}", data=settings)
 
     async def add_device(
-        self, device_id: str, push_provider: str, user_id: str
+        self,
+        device_id: str,
+        push_provider: str,
+        user_id: str,
+        push_provider_name: str = None,
     ) -> StreamResponse:
         return await self.post(
             "devices",
-            data={"id": device_id, "push_provider": push_provider, "user_id": user_id},
+            data={
+                "id": device_id,
+                "push_provider": push_provider,
+                "user_id": user_id,
+                "push_provider_name": push_provider_name,
+            },
         )
 
     async def delete_device(self, device_id: str, user_id: str) -> StreamResponse:

diff --git a/stream_chat/base/client.py b/stream_chat/base/client.py
@@ -423,7 +423,11 @@ def update_command(
 
     @abc.abstractmethod
     def add_device(
-        self, device_id: str, push_provider: str, user_id: str
+        self,
+        device_id: str,
+        push_provider: str,
+        user_id: str,
+        push_provider_name: str = None,
     ) -> Union[StreamResponse, Awaitable[StreamResponse]]:
         """
         Add a device to a user

diff --git a/stream_chat/client.py b/stream_chat/client.py
@@ -334,11 +334,20 @@ def update_command(self, name: str, **settings: Any) -> StreamResponse:
         return self.put(f"commands/{name}", data=settings)
 
     def add_device(
-        self, device_id: str, push_provider: str, user_id: str
+        self,
+        device_id: str,
+        push_provider: str,
+        user_id: str,
+        push_provider_name: str = None,
     ) -> StreamResponse:
         return self.post(
             "devices",
-            data={"id": device_id, "push_provider": push_provider, "user_id": user_id},
+            data={
+                "id": device_id,
+                "push_provider": push_provider,
+                "user_id": user_id,
+                "push_provider_name": push_provider_name,
+            },
         )
 
     def delete_device(self, device_id: str, user_id: str) -> StreamResponse: