stinger-python-utils 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

stinger_python_utils/message_creator.py

@@ -2,7 +2,7 @@ from pyqttier.message import Message
 from pydantic import BaseModel
 from typing import Union, Optional
 import uuid
-
+from .return_codes import MethodReturnCode
 
 class MessageCreator:
 
@@ -31,13 +31,14 @@ class MessageCreator:
     def error_response_message(
         cls,
         topic: str,
-        return_code: int,
+        return_code: Union[int, MethodReturnCode],
         correlation_id: Union[str, bytes, None] = None,
         debug_info: Optional[str] = None,
     ) -> Message:
         """
         This could be used for a response to a request, but where there was an error fulfilling the request.
         """
+        rc = return_code.value if isinstance(return_code, MethodReturnCode) else return_code
         msg_obj = Message(
             topic=topic,
             payload=b"{}",
@@ -48,7 +49,7 @@ class MessageCreator:
                 if isinstance(correlation_id, str)
                 else correlation_id
             ),
-            user_properties={"ReturnCode": str(return_code)},
+            user_properties={"ReturnCode": str(rc)},
         )
         if (
             debug_info is not None and msg_obj.user_properties is not None
@@ -61,7 +62,7 @@ class MessageCreator:
         cls,
         response_topic: str,
         response_obj: Union[BaseModel, str, bytes],
-        return_code: int,
+        return_code: Union[int, MethodReturnCode],
         correlation_id: Union[str, bytes, None] = None,
     ) -> Message:
         """
@@ -73,6 +74,7 @@ class MessageCreator:
             payload = response_obj.encode("utf-8")
         else:
             payload = response_obj
+        rc = return_code.value if isinstance(return_code, MethodReturnCode) else return_code
         msg_obj = Message(
             topic=response_topic,
             payload=payload,
@@ -83,7 +85,7 @@ class MessageCreator:
                 if isinstance(correlation_id, str)
                 else correlation_id
             ),
-            user_properties={"ReturnCode": str(return_code)},
+            user_properties={"ReturnCode": str(rc)},
         )
         return msg_obj
 
@@ -138,13 +140,14 @@ class MessageCreator:
         response_topic: str,
         property_obj: BaseModel,
         version: str,
-        return_code: int,
+        return_code: Union[int, MethodReturnCode],
         correlation_id: Union[str, bytes, None] = None,
         debug_info: Optional[str] = None,
     ) -> Message:
         """
         Creates a message representing a response to a property update request.
         """
+        rc = return_code.value if isinstance(return_code, MethodReturnCode) else return_code
         msg_obj = Message(
             topic=response_topic,
             payload=property_obj.model_dump_json(by_alias=True).encode("utf-8"),
@@ -156,7 +159,7 @@ class MessageCreator:
                 else correlation_id
             ),
             user_properties={
-                "ReturnCode": str(return_code),
+                "ReturnCode": str(rc),
                 "PropertyVersion": str(version),
             },
         )

{stinger_python_utils-0.1.4.dist-info → stinger_python_utils-0.1.5.dist-info}/METADATA

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

stinger_python_utils-0.1.5.dist-info/RECORD

@@ -0,0 +1,7 @@
+stinger_python_utils/__init__.py,sha256=IjHRV0k2DNwvFrEHebmsXiBvmITE8nQUnsR07h9tVkU,7
+stinger_python_utils/message_creator.py,sha256=KHiw7Zk2n8QI96Zv4dT2Isj8ZljwSKaVfyoOyIjlRCA,6529
+stinger_python_utils/return_codes.py,sha256=AAwshvHu3SBoctwMX35k3j666J4a7DQ3V8AnR7QBjC8,5114
+stinger_python_utils-0.1.5.dist-info/METADATA,sha256=6qVtX_zsa7TGr3Ny8myBxm0dhY2OSR_IP6xx5SB5Yog,2769
+stinger_python_utils-0.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+stinger_python_utils-0.1.5.dist-info/licenses/LICENSE,sha256=_T-8ExmblJbhv-1AxC6XDVEHg1JdJA-126NvRiMqS-I,1070
+stinger_python_utils-0.1.5.dist-info/RECORD,,

stinger_python_utils-0.1.4.dist-info/RECORD

@@ -1,7 +0,0 @@
-stinger_python_utils/__init__.py,sha256=IjHRV0k2DNwvFrEHebmsXiBvmITE8nQUnsR07h9tVkU,7
-stinger_python_utils/message_creator.py,sha256=tsGmcw5-iC58lBq4aJQrI3IDZ3cE88C5FOT_5lHLvn0,6160
-stinger_python_utils/return_codes.py,sha256=AAwshvHu3SBoctwMX35k3j666J4a7DQ3V8AnR7QBjC8,5114
-stinger_python_utils-0.1.4.dist-info/METADATA,sha256=Ces8NRPz6g4Vuo2UYlG_3yMSmciZF9XuaAcJqfMHYFA,2769
-stinger_python_utils-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-stinger_python_utils-0.1.4.dist-info/licenses/LICENSE,sha256=_T-8ExmblJbhv-1AxC6XDVEHg1JdJA-126NvRiMqS-I,1070
-stinger_python_utils-0.1.4.dist-info/RECORD,,