Expose Blast to OBO

Author: anthony-symphony

symphony/bdk/core/service/message/message_service.py

@@ -139,59 +139,7 @@ async def get_attachment_types(self) -> List[str]:
         }
         type_list = await self._pod_api.v1_files_allowed_types_get(**params)
         return type_list.value
-
-
-class MessageService(OboMessageService):
-    """Service class for managing messages.
-    """
-
-    def __init__(self, messages_api: MultiAttachmentsMessagesApi,
-                 message_api: MessageApi,
-                 message_suppression_api: MessageSuppressionApi,
-                 streams_api: StreamsApi,
-                 pod_api: PodApi,
-                 attachment_api: AttachmentsApi,
-                 default_api: DefaultApi,
-                 auth_session: AuthSession,
-                 retry_config: BdkRetryConfig):
-        super().__init__(messages_api, message_suppression_api, auth_session, retry_config)
-        self._message_api = message_api
-        self._message_suppression_api = message_suppression_api
-        self._streams_api = streams_api
-        self._pod_api = pod_api
-        self._attachment_api = attachment_api
-        self._default_api = default_api
-
-    @retry
-    async def list_messages(
-            self,
-            stream_id: str,
-            since: int = 0,
-            skip: int = 0,
-            limit: int = 50
-    ) -> [V4Message]:
-        """Get messages from an existing stream. Additionally returns any attachments associated with the message.
-        See: `Messages <https://developers.symphony.com/restapi/reference/messages-v4>`_
-
-        :param stream_id: The stream where to look for messages
-        :param since: Timestamp of the earliest possible date of the first message returned.
-        :param skip: Number of messages to skip. Default: 0
-        :param limit: Maximum number of messages to return. Default: 50
-
-        :return: the list of matching messages in the stream.
-
-        """
-        params = {
-            "sid": stream_id,
-            "since": since,
-            "session_token": await self._auth_session.session_token,
-            "key_manager_token": await self._auth_session.key_manager_token,
-            "skip": skip,
-            "limit": limit
-        }
-        message_list = await self._messages_api.v4_stream_sid_message_get(**params)
-        return message_list.value
-
+    
     async def blast_message(
             self,
             stream_ids: List[str],
@@ -220,7 +168,7 @@ async def blast_message(
                                                                               version=version, attachments=attachment)
         return await self._blast_message(stream_ids, message_object.content, message_object.data,
                                          message_object.version, message_object.attachments, message_object.previews)
-
+        
     @retry
     async def _blast_message(
             self,
@@ -261,6 +209,58 @@ async def _blast_message(
 
         return await self._messages_api.v4_multi_attachment_message_blast_post(**params)
 
+
+class MessageService(OboMessageService):
+    """Service class for managing messages.
+    """
+
+    def __init__(self, messages_api: MultiAttachmentsMessagesApi,
+                 message_api: MessageApi,
+                 message_suppression_api: MessageSuppressionApi,
+                 streams_api: StreamsApi,
+                 pod_api: PodApi,
+                 attachment_api: AttachmentsApi,
+                 default_api: DefaultApi,
+                 auth_session: AuthSession,
+                 retry_config: BdkRetryConfig):
+        super().__init__(messages_api, message_suppression_api, auth_session, retry_config)
+        self._message_api = message_api
+        self._message_suppression_api = message_suppression_api
+        self._streams_api = streams_api
+        self._pod_api = pod_api
+        self._attachment_api = attachment_api
+        self._default_api = default_api
+
+    @retry
+    async def list_messages(
+            self,
+            stream_id: str,
+            since: int = 0,
+            skip: int = 0,
+            limit: int = 50
+    ) -> [V4Message]:
+        """Get messages from an existing stream. Additionally returns any attachments associated with the message.
+        See: `Messages <https://developers.symphony.com/restapi/reference/messages-v4>`_
+
+        :param stream_id: The stream where to look for messages
+        :param since: Timestamp of the earliest possible date of the first message returned.
+        :param skip: Number of messages to skip. Default: 0
+        :param limit: Maximum number of messages to return. Default: 50
+
+        :return: the list of matching messages in the stream.
+
+        """
+        params = {
+            "sid": stream_id,
+            "since": since,
+            "session_token": await self._auth_session.session_token,
+            "key_manager_token": await self._auth_session.key_manager_token,
+            "skip": skip,
+            "limit": limit
+        }
+        message_list = await self._messages_api.v4_stream_sid_message_get(**params)
+        return message_list.value
+
     @retry
     async def import_messages(
             self,