stinger-python-utils 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

stinger_python_utils/{message_creator → message_creator.py}

@@ -1,15 +1,13 @@
-
 from pyqttier.message import Message
 from pydantic import BaseModel
 from typing import Union, Optional
 import uuid
 
+
 class MessageCreator:
 
     @classmethod
-    def signal_message(
-        cls, topic: str, payload: BaseModel
-    ) -> Message:
+    def signal_message(cls, topic: str, payload: BaseModel) -> Message:
         return Message(
             topic=topic,
             payload=payload.model_dump_json(by_alias=True).encode("utf-8"),
@@ -34,7 +32,7 @@ class MessageCreator:
         cls,
         topic: str,
         return_code: int,
-        correlation_id: Union[str, bytes],
+        correlation_id: Union[str, bytes, None] = None,
         debug_info: Optional[str] = None,
     ) -> Message:
         """
@@ -62,14 +60,19 @@ class MessageCreator:
     def response_message(
         cls,
         response_topic: str,
-        response_obj: BaseModel|str|bytes,
+        response_obj: BaseModel | str | bytes,
         return_code: int,
-        correlation_id: Union[str, bytes],
+        correlation_id: Union[str, bytes, None] = None,
     ) -> Message:
         """
         This could be used for a successful response to a request.
         """
-        payload = response_obj.model_dump_json(by_alias=True).encode("utf-8") if isinstance(response_obj, BaseModel) else response_obj
+        if isinstance(response_obj, BaseModel):
+            payload = response_obj.model_dump_json(by_alias=True).encode("utf-8")
+        elif isinstance(response_obj, str):
+            payload = response_obj.encode("utf-8")
+        else:
+            payload = response_obj
         msg_obj = Message(
             topic=response_topic,
             payload=payload,
@@ -96,6 +99,7 @@ class MessageCreator:
             payload=state_obj.model_dump_json(by_alias=True).encode("utf-8"),
             qos=1,
             retain=True,
+            content_type="application/json",
         )
         if state_version is not None:
             msg_obj.user_properties = {"PropertyVersion": str(state_version)}
@@ -135,7 +139,7 @@ class MessageCreator:
         property_obj: BaseModel,
         version: str,
         return_code: int,
-        correlation_id: Union[str, bytes],
+        correlation_id: Union[str, bytes, None] = None,
         debug_info: Optional[str] = None,
     ) -> Message:
         """

{stinger_python_utils-0.1.1.dist-info → stinger_python_utils-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stinger-python-utils
-Version: 0.1.1
+Version: 0.1.3
 Summary: Common utilities for Stinger Python services.
 License-Expression: MIT
 License-File: LICENSE

stinger_python_utils-0.1.3.dist-info/RECORD

@@ -0,0 +1,7 @@
+stinger_python_utils/__init__.py,sha256=IjHRV0k2DNwvFrEHebmsXiBvmITE8nQUnsR07h9tVkU,7
+stinger_python_utils/message_creator.py,sha256=hTGXbVIuvc_7pzP73N0z2VzvRiNYFKNwNKLbvh4Kw14,6155
+stinger_python_utils/return_codes.py,sha256=AAwshvHu3SBoctwMX35k3j666J4a7DQ3V8AnR7QBjC8,5114
+stinger_python_utils-0.1.3.dist-info/METADATA,sha256=TKdZvY9ZQZ0X3j97eLRj_jpYV9L0eSp8a7tH6WaHbh8,364
+stinger_python_utils-0.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+stinger_python_utils-0.1.3.dist-info/licenses/LICENSE,sha256=_T-8ExmblJbhv-1AxC6XDVEHg1JdJA-126NvRiMqS-I,1070
+stinger_python_utils-0.1.3.dist-info/RECORD,,

stinger_python_utils-0.1.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-stinger_python_utils/__init__.py,sha256=IjHRV0k2DNwvFrEHebmsXiBvmITE8nQUnsR07h9tVkU,7
-stinger_python_utils/message_creator,sha256=Tho68dvdCje0a4DELlbzH2C9fhEqg7s2TkdBDohoON0,5942
-stinger_python_utils/return_codes.py,sha256=AAwshvHu3SBoctwMX35k3j666J4a7DQ3V8AnR7QBjC8,5114
-stinger_python_utils-0.1.1.dist-info/METADATA,sha256=QbLI1Q4u5TfM4X40JYKi-7W-_0etQNNRkbjFW-O7LZQ,364
-stinger_python_utils-0.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-stinger_python_utils-0.1.1.dist-info/licenses/LICENSE,sha256=_T-8ExmblJbhv-1AxC6XDVEHg1JdJA-126NvRiMqS-I,1070
-stinger_python_utils-0.1.1.dist-info/RECORD,,