finalsa-common-models 2.0.0__tar.gz → 2.0.2__tar.gz

{finalsa_common_models-2.0.0 → finalsa_common_models-2.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: finalsa-common-models
-Version: 2.0.0
+Version: 2.0.2
 Summary: Common models for Finalsa
 Project-URL: Homepage, https://github.com/finalsa/finalsa-common-models
 Author-email: Luis Jimenez <luis@finalsa.com>

finalsa_common_models-2.0.2/finalsa/common/models/__init__.py

@@ -0,0 +1,18 @@
+from finalsa.common.models.models import (
+    Meta,
+    SqsReponse,
+    parse_sns_message_attributes,
+    parse_sqs_message_attributes,
+    to_sqs_message_attributes,
+    to_sns_message_attributes,
+)
+
+
+__all__ = [
+    "Meta",
+    "SqsReponse",
+    "parse_sns_message_attributes",
+    "parse_sqs_message_attributes",
+    "to_sqs_message_attributes",
+    "to_sns_message_attributes",
+]

finalsa_common_models-2.0.2/finalsa/common/models/models/__init__.py

@@ -0,0 +1,15 @@
+from .functions import parse_sns_message_attributes, parse_sqs_message_attributes, to_sqs_message_attributes, to_sns_message_attributes
+from .sqs_response import SqsReponse
+from .meta import Meta, AsyncMeta, Authorization, HttpMeta
+
+__all__ = [
+    "Meta",
+    "AsyncMeta",
+    "Authorization",
+    "HttpMeta",
+    "SqsReponse",
+    "parse_sns_message_attributes",
+    "parse_sqs_message_attributes",
+    "to_sqs_message_attributes",
+    "to_sns_message_attributes"
+]

finalsa_common_models-2.0.2/finalsa/common/models/models/functions.py

@@ -0,0 +1,79 @@
+from typing import Dict
+from datetime import datetime
+from decimal import Decimal
+from uuid import UUID
+
+
+def parse_sns_message_attributes(attributes: Dict) -> Dict:
+    message_attributes = {}
+    if not attributes:
+        return message_attributes
+    for key, value in attributes.items():
+        if value['Type'] == 'String':
+            message_attributes[key] = value['Value']
+        elif value['Type'] == 'Number':
+            message_attributes[key] = int(value['Value'])
+        elif value['Type'] == 'Binary':
+            message_attributes[key] = bytes(value['Value'], 'utf-8')
+    return message_attributes
+
+def parse_sqs_message_attributes(attributes: Dict) -> Dict:
+    message_attributes = {}
+    if not attributes:
+        return message_attributes
+    for key, value in attributes.items():
+        if value['DataType'] == 'String':
+            message_attributes[key] = value['StringValue']
+        elif value['DataType'] == 'Number':
+            message_attributes[key] = int(value['StringValue'])
+        elif value['DataType'] == 'Binary':
+            message_attributes[key] = bytes(value['StringValue'], 'utf-8')
+    return message_attributes
+
+def to_sqs_message_attributes(attributes: Dict) -> Dict:
+    att_dict = {}
+    for key, value in attributes.items():
+        if isinstance(value, str):
+            att_dict[key] = {
+                'DataType': 'String', 'StringValue': value}
+        elif isinstance(value, Decimal):
+            att_dict[key] = {
+                'DataType': 'Number', 'StringValue': str(value)}
+        elif isinstance(value, UUID):
+            att_dict[key] = {
+                'DataType': 'String', 'StringValue': str(value)}
+        elif isinstance(value, datetime):
+            att_dict[key] = {
+                'DataType': 'String', 'StringValue': value.isoformat()}
+        elif isinstance(value, int):
+            att_dict[key] = {
+                'DataType': 'Number', 'StringValue': str(value)}
+        elif isinstance(value, bytes):
+            att_dict[key] = {
+                'DataType': 'Binary', 'BinaryValue': value}
+    return att_dict
+
+
+
+def to_sns_message_attributes(attributes: Dict) -> Dict:
+    att_dict = {}
+    for key, value in attributes.items():
+        if isinstance(value, str):
+            att_dict[key] = {
+                'Type': 'String', 'Value': value}
+        elif isinstance(value, Decimal):
+            att_dict[key] = {
+                'Type': 'Number', 'Value': str(value)}
+        elif isinstance(value, UUID):
+            att_dict[key] = {
+                'Type': 'String', 'Value': str(value)}
+        elif isinstance(value, datetime):
+            att_dict[key] = {
+                'Type': 'String', 'Value': value.isoformat()}
+        elif isinstance(value, int):
+            att_dict[key] = {
+                'Type': 'Number', 'Value': str(value)}
+        elif isinstance(value, bytes):
+            att_dict[key] = {
+                'Type': 'Binary', 'Value': value}
+    return att_dict

finalsa_common_models-2.0.2/finalsa/common/models/models/meta.py

@@ -0,0 +1,30 @@
+from pydantic import BaseModel
+from datetime import datetime
+from typing import Optional, List
+
+class Authorization(BaseModel):
+    token: str
+    user_id: str
+    roles: Optional[List[str]] = None
+    groups: Optional[List[str]] = None
+    permissions: Optional[List[str]] = None
+    scopes: Optional[List[str]] = None
+    expiration: Optional[datetime] = None
+    
+
+class Meta(BaseModel):
+    authorization: Optional[Authorization] = None
+    timestamp: datetime
+    correlation_id: str
+
+
+class HttpMeta(Meta):
+    ip: str
+
+
+class AsyncMeta(Meta):
+    topic: str
+    subtopic: Optional[str] = None
+    produced_at: Optional[datetime] = None
+    consumed_at: Optional[datetime] = None
+    retry_count: int = 0

{finalsa_common_models-2.0.0 → finalsa_common_models-2.0.2}/finalsa/common/models/models/sqs_response.py

@@ -1,7 +1,7 @@
 from typing import Dict, Optional, Union
 from pydantic import BaseModel
 from orjson import loads
-from .functions import parse_message_attributes
+from .functions import parse_sqs_message_attributes, parse_sns_message_attributes
 
 
 class SqsReponse(BaseModel):
@@ -26,7 +26,7 @@ class SqsReponse(BaseModel):
 
     def __parse_from_sns__(self, payload: Dict) -> Union[str, Dict]:
         self.topic = str(payload['TopicArn'].split(':')[-1]).lower()
-        self.message_attributes = parse_message_attributes(
+        self.message_attributes = parse_sns_message_attributes(
             payload.get('MessageAttributes', {}))
         try:
             return loads(payload['Message'])
@@ -59,6 +59,6 @@ class SqsReponse(BaseModel):
             attributes=response['Attributes'],
             md5_of_message_attributes=response.get(
                 'MD5OfMessageAttributes', ''),
-            message_attributes=parse_message_attributes(
+            message_attributes=parse_sqs_message_attributes(
                 response.get('MessageAttributes', {}))
         )

{finalsa_common_models-2.0.0 → finalsa_common_models-2.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "finalsa-common-models"
-version = "2.0.0"
+version = "2.0.2"
 description = "Common models for Finalsa"
 readme = "README.md"
 requires-python = ">=3.10"

finalsa_common_models-2.0.0/finalsa/common/models/__init__.py

@@ -1,14 +0,0 @@
-from finalsa.common.models.models import (
-    Meta,
-    SqsReponse,
-    parse_message_attributes,
-    to_message_attributes
-)
-
-
-__all__ = [
-    "Meta",
-    "SqsReponse",
-    "parse_message_attributes",
-    "to_message_attributes",
-]

finalsa_common_models-2.0.0/finalsa/common/models/models/__init__.py

@@ -1,10 +0,0 @@
-from .functions import parse_message_attributes, to_message_attributes
-from .sqs_response import SqsReponse
-from .meta import Meta
-
-__all__ = [
-    "Meta",
-    "SqsReponse",
-    "parse_message_attributes",
-    "to_message_attributes",
-]

finalsa_common_models-2.0.0/finalsa/common/models/models/functions.py

@@ -1,42 +0,0 @@
-from typing import Dict
-from datetime import datetime
-from decimal import Decimal
-from uuid import UUID
-
-
-def parse_message_attributes(attributes: Dict) -> Dict:
-    message_attributes = {}
-    if not attributes:
-        return message_attributes
-    for key, value in attributes.items():
-        if value['Type'] == 'String':
-            message_attributes[key] = value['Value']
-        elif value['Type'] == 'Number':
-            message_attributes[key] = int(value['Value'])
-        elif value['Type'] == 'Binary':
-            message_attributes[key] = bytes(value['Value'], 'utf-8')
-    return message_attributes
-
-
-def to_message_attributes(attributes: Dict) -> Dict:
-    att_dict = {}
-    for key, value in attributes.items():
-        if isinstance(value, str):
-            att_dict[key] = {
-                'DataType': 'String', 'StringValue': value}
-        elif isinstance(value, Decimal):
-            att_dict[key] = {
-                'DataType': 'Number', 'StringValue': str(value)}
-        elif isinstance(value, UUID):
-            att_dict[key] = {
-                'DataType': 'String', 'StringValue': str(value)}
-        elif isinstance(value, datetime):
-            att_dict[key] = {
-                'DataType': 'String', 'StringValue': value.isoformat()}
-        elif isinstance(value, int):
-            att_dict[key] = {
-                'DataType': 'Number', 'StringValue': str(value)}
-        elif isinstance(value, bytes):
-            att_dict[key] = {
-                'DataType': 'Binary', 'BinaryValue': value}
-    return att_dict

finalsa_common_models-2.0.0/finalsa/common/models/models/meta.py

@@ -1,10 +0,0 @@
-from pydantic import BaseModel
-from datetime import datetime
-from typing import Optional
-
-
-class Meta(BaseModel):
-    authorization: Optional[dict] = {}
-    timestamp: datetime
-    correlation_id: str
-    ip: str