feat: add options to export channel (#100)

Author: peterdeme

stream_chat/async_chat/client.py

@@ -552,6 +552,7 @@ async def export_channel(
         channel_id: str,
         messages_since: Union[str, datetime.datetime] = None,
         messages_until: Union[str, datetime.datetime] = None,
+        **options: Any,
     ) -> StreamResponse:
         if isinstance(messages_since, datetime.datetime):
             messages_since = messages_since.isoformat()
@@ -566,11 +567,16 @@ async def export_channel(
                     "messages_since": messages_since,
                     "messages_until": messages_until,
                 }
-            ]
+            ],
+            **options,
         )
 
-    async def export_channels(self, channels: Iterable[Dict]) -> StreamResponse:
-        return await self.post("export_channels", data={"channels": channels})
+    async def export_channels(
+        self, channels: Iterable[Dict], **options: Any
+    ) -> StreamResponse:
+        return await self.post(
+            "export_channels", data={"channels": channels, **options}
+        )
 
     async def get_export_channel_status(self, task_id: str) -> StreamResponse:
         return await self.get(f"export_channels/{task_id}")

stream_chat/base/client.py

@@ -823,6 +823,7 @@ def export_channel(
         channel_id: str,
         messages_since: Union[str, datetime.datetime] = None,
         messages_until: Union[str, datetime.datetime] = None,
+        **options: Any,
     ) -> Union[StreamResponse, Awaitable[StreamResponse]]:
         """
         Requests a channel export
@@ -831,7 +832,7 @@ def export_channel(
 
     @abc.abstractmethod
     def export_channels(
-        self, channels: Iterable[Dict]
+        self, channels: Iterable[Dict], **options: Any
     ) -> Union[StreamResponse, Awaitable[StreamResponse]]:
         """
         Requests a channels export

stream_chat/client.py

@@ -524,6 +524,7 @@ def export_channel(
         channel_id: str,
         messages_since: Union[str, datetime.datetime] = None,
         messages_until: Union[str, datetime.datetime] = None,
+        **options: Any,
     ) -> StreamResponse:
         if isinstance(messages_since, datetime.datetime):
             messages_since = messages_since.isoformat()
@@ -538,11 +539,14 @@ def export_channel(
                     "messages_since": messages_since,
                     "messages_until": messages_until,
                 }
-            ]
+            ],
+            **options,
         )
 
-    def export_channels(self, channels: Iterable[Dict]) -> StreamResponse:
-        return self.post("export_channels", data={"channels": channels})
+    def export_channels(
+        self, channels: Iterable[Dict], **options: Any
+    ) -> StreamResponse:
+        return self.post("export_channels", data={"channels": channels, **options})
 
     def get_export_channel_status(self, task_id: str) -> StreamResponse:
         return self.get(f"export_channels/{task_id}")

stream_chat/tests/async_chat/test_channel.py

@@ -340,7 +340,9 @@ async def test_export_channel(
     ):
         await channel.send_message({"text": "Hey Joni"}, random_users[0]["id"])
 
-        resp = await client.export_channel(channel.channel_type, channel.id)
+        resp = await client.export_channel(
+            channel.channel_type, channel.id, include_truncated_messages=False
+        )
         task_id = resp["task_id"]
         assert task_id != ""
 

stream_chat/tests/test_channel.py

@@ -328,7 +328,9 @@ def test_export_channel(
     ):
         channel.send_message({"text": "Hey Joni"}, random_users[0]["id"])
 
-        resp = client.export_channel(channel.channel_type, channel.id)
+        resp = client.export_channel(
+            channel.channel_type, channel.id, include_truncated_messages=False
+        )
         task_id = resp["task_id"]
         assert task_id != ""