ergon-framework-python 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

ergon/connector/ergon_platform/workflows/utils.py

@@ -0,0 +1,250 @@
+import base64
+import json
+from typing import Any, Dict, List, Optional
+
+from ...transaction import Transaction
+from .models import (
+    FAILURE_STATUSES,
+    PROCESSING_STATUSES,
+    SUCCESS_STATUSES,
+    CreateItemInput,
+    CreateItemPayload,
+)
+
+
+def get_value(obj: Any, key: str, default: Any = None) -> Any:
+    """Read a key from a dict or an attribute from an object, tolerating both shapes."""
+    if isinstance(obj, dict):
+        return obj.get(key, default)
+    return getattr(obj, key, default)
+
+
+def as_payload(obj: Any) -> Any:
+    """Convert SDK models to JSON-friendly payloads while leaving plain data untouched."""
+    if hasattr(obj, "model_dump"):
+        return obj.model_dump(mode="json")
+    return obj
+
+
+def serialize_object(obj: Any) -> Any:
+    if obj is None:
+        return None
+    if isinstance(obj, (str, int, float, bool, bytes)):
+        return obj
+    if isinstance(obj, dict):
+        return {k: serialize_object(v) for k, v in obj.items()}
+    if isinstance(obj, (list, tuple)):
+        return [serialize_object(item) for item in obj]
+    if hasattr(obj, "model_dump"):
+        return obj.model_dump(mode="json")
+    if hasattr(obj, "to_dict"):
+        return obj.to_dict()
+    if hasattr(obj, "__dict__"):
+        return {k: serialize_object(v) for k, v in obj.__dict__.items() if not k.startswith("_")}
+    return obj
+
+
+def classify_status(raw_status: str) -> str:
+    """Map a raw pipeline status string to a coarse state."""
+    status = (raw_status or "").lower()
+    if status in SUCCESS_STATUSES:
+        return "success"
+    if status in FAILURE_STATUSES:
+        return "failed"
+    if status in PROCESSING_STATUSES:
+        return "processing"
+    return "unknown"
+
+
+def find_status_entry(statuses: Any, buckets_file_id: str) -> Any:
+    for status in extract_status_entries(statuses):
+        if extract_status_file_id(status) == str(buckets_file_id):
+            return status
+    return {}
+
+
+def extract_status_entries(statuses: Any) -> List[Any]:
+    """Normalize SDK attachment status responses to a list of status entries."""
+    if statuses is None:
+        return []
+    if isinstance(statuses, list):
+        return statuses
+    if isinstance(statuses, tuple):
+        return list(statuses)
+    if isinstance(statuses, dict):
+        for key in ("items", "data", "results", "statuses", "attachments", "files"):
+            value = statuses.get(key)
+            if isinstance(value, list):
+                return value
+        if get_value(statuses, "status") is not None or extract_status_file_id(statuses):
+            return [statuses]
+        return []
+    for key in ("items", "data", "results", "statuses", "attachments", "files"):
+        value = getattr(statuses, key, None)
+        if isinstance(value, (list, tuple)):
+            return list(value)
+    if get_value(statuses, "status") is not None or extract_status_file_id(statuses):
+        return [statuses]
+    return []
+
+
+def extract_status_file_id(status: Any) -> Optional[str]:
+    for key in ("buckets_file_id", "bucketsFileId", "file_id", "fileId"):
+        value = get_value(status, key)
+        if value:
+            return str(value)
+    return None
+
+
+def latest_status_entry(statuses: Any) -> Any:
+    entries = extract_status_entries(statuses)
+    return entries[-1] if entries else {}
+
+
+def extract_buckets_file_id(item: Any, field_id: str) -> Optional[str]:
+    field_values = get_value(item, "field_values", []) or []
+    if isinstance(field_values, dict):
+        return extract_buckets_file_id_from_attachments(field_values.get(field_id))
+
+    for field_value in field_values:
+        if str(get_value(field_value, "field_id")) != str(field_id):
+            continue
+        buckets_file_id = extract_buckets_file_id_from_attachments(get_value(field_value, "value", []))
+        if buckets_file_id:
+            return buckets_file_id
+    return None
+
+
+def extract_buckets_file_id_from_attachments(value: Any) -> Optional[str]:
+    attachments = value or []
+    if isinstance(attachments, dict):
+        for key in ("attachments", "files", "items", "data", "results", "value"):
+            nested = attachments.get(key)
+            if isinstance(nested, list):
+                attachments = nested
+                break
+        else:
+            attachments = [attachments]
+    if not isinstance(attachments, list):
+        return None
+    for attachment in reversed(attachments):
+        buckets_file_id = get_value(attachment, "buckets_file_id")
+        if buckets_file_id:
+            return str(buckets_file_id)
+    return None
+
+
+def item_to_transaction(item: Any, workflow_id: str) -> Transaction:
+    payload = serialize_object(item)
+    payload = payload if isinstance(payload, dict) else {"value": payload}
+    return Transaction(
+        id=str(payload.get("id", "")),
+        payload=payload,
+        metadata={
+            "workflow_id": workflow_id,
+            "phase_id": payload.get("phase_id"),
+            "company_id": payload.get("company_id"),
+            "title": payload.get("title"),
+        },
+    )
+
+
+def normalize_create_payload(payload: CreateItemPayload) -> Dict[str, Any]:
+    """Flatten a CreateItemInput/dict into kwargs accepted by the service create_item."""
+    if isinstance(payload, CreateItemInput):
+        data: Dict[str, Any] = {
+            "title": payload.title,
+            "workflow_id": payload.workflow_id,
+            "phase_id": payload.phase_id,
+            "parent_item_id": payload.parent_item_id,
+            "field_values": payload.field_values,
+            "attachment": payload.attachment,
+            "attachment_field_id": payload.attachment_field_id,
+            "content_type": payload.content_type,
+            "fields": dict(payload.extra_fields),
+        }
+        return data
+    if isinstance(payload, dict):
+        data = dict(payload)
+        data.setdefault("fields", {})
+        return data
+    raise TypeError(f"Unsupported create payload type: {type(payload)}")
+
+
+def requeue(transaction: Transaction) -> CreateItemInput:
+    """Build a CreateItemInput from a transaction payload for re-dispatch."""
+    data = normalize_create_payload(transaction.payload)
+    workflow_id = data.get("workflow_id") or transaction.metadata.get("workflow_id")
+    phase_id = data.get("phase_id") or transaction.metadata.get("phase_id")
+    title = data.get("title")
+    if not isinstance(title, str) or not title.strip():
+        raise ValueError("title is required to requeue a transaction")
+
+    return CreateItemInput(
+        title=title,
+        workflow_id=workflow_id,
+        phase_id=phase_id,
+        parent_item_id=data.get("parent_item_id"),
+        field_values=data.get("field_values"),
+        attachment=data.get("attachment"),
+        attachment_field_id=data.get("attachment_field_id"),
+        content_type=data.get("content_type"),
+        extra_fields=dict(data.get("fields") or {}),
+    )
+
+
+def build_redispatch_transaction(transaction: Transaction) -> Transaction:
+    """Prepare a transaction payload suitable for re-dispatch."""
+    return Transaction(
+        id=transaction.id,
+        payload=requeue(transaction),
+        metadata=dict(transaction.metadata or {}),
+    )
+
+
+def decode_jwt_claims(jwt_token: str) -> Dict[str, Any]:
+    """Decode JWT payload claims without signature validation.
+
+    Raises:
+        ValueError: When the token shape or payload encoding is invalid.
+    """
+    parts = jwt_token.split(".")
+    if len(parts) < 2 or not parts[1]:
+        raise ValueError("Invalid JWT format: payload segment is missing")
+
+    payload = parts[1]
+    payload += "=" * (-len(payload) % 4)
+    try:
+        decoded = base64.urlsafe_b64decode(payload.encode("ascii"))
+        claims = json.loads(decoded.decode("utf-8"))
+    except Exception as exc:  # noqa: BLE001
+        raise ValueError("Invalid JWT payload encoding") from exc
+
+    if not isinstance(claims, dict):
+        raise ValueError("Invalid JWT payload: expected a JSON object")
+    return claims
+
+
+def extract_items(response: Any) -> List[Any]:
+    """Pull the item list out of a Page, dict, or raw list response."""
+    if isinstance(response, list):
+        return response
+    if isinstance(response, dict):
+        for key in ("items", "data", "results"):
+            value = response.get(key)
+            if isinstance(value, list):
+                return value
+        return []
+    items = getattr(response, "items", None)
+    if isinstance(items, (list, tuple)):
+        return list(items)
+    return []
+
+
+def extract_total(response: Any) -> int:
+    """Read total count from a paginated Page, dict, or SDK response."""
+    for key in ("total", "total_count", "count"):
+        value = get_value(response, key)
+        if value is not None:
+            return int(value)
+    return 0

ergon/task/mixins/consumer.py

@@ -136,7 +136,6 @@ class ConsumerMixin(ABC):
             # -----------------------
             # 3) SUCCESS HANDLER
             # -----------------------
-            logger.info(f"Invoking success handler for transaction {transaction.id} with outcome: '{process_result}'")
             success_ok, success_result = self._handle_success(transaction, process_result, policy.success.retry)
 
             if not success_ok:
@@ -679,7 +678,6 @@ class AsyncConsumerMixin(ABC):
             # -----------------------
             # 3) SUCCESS HANDLER
             # -----------------------
-            logger.info(f"Invoking success handler for transaction {transaction.id} with outcome: '{process_result}'")
             success_ok, success_result = await self._handle_success(transaction, process_result, policy.success.retry)
 
             if not success_ok:

{ergon_framework_python-0.2.1.dist-info → ergon_framework_python-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ergon-framework-python
-Version: 0.2.1
+Version: 0.2.2
 Summary: Ergon internal task-oriented project bootstrapper
 Author-email: Ergondata Technologies <anza.vossos@protonmail.com>
 License-Expression: MIT
@@ -29,14 +29,14 @@ Requires-Dist: aio-pika>=9.0.0; extra == "all"
 Requires-Dist: aiobotocore>=2.13.0; extra == "all"
 Requires-Dist: asyncpg>=0.29.0; extra == "all"
 Requires-Dist: boto3>=1.35.0; extra == "all"
-Requires-Dist: ergon-platform-sdk>=0.1.0; extra == "all"
+Requires-Dist: ergon-platform-sdk>=0.2.0; extra == "all"
 Requires-Dist: httpx>=0.28.0; extra == "all"
 Requires-Dist: nylas>=6.0.0; extra == "all"
 Requires-Dist: openpyxl>=3.1.5; extra == "all"
 Requires-Dist: pika>=1.3.0; extra == "all"
 Requires-Dist: requests>=2.32.5; extra == "all"
 Provides-Extra: ergon-platform
-Requires-Dist: ergon-platform-sdk>=0.1.0; extra == "ergon-platform"
+Requires-Dist: ergon-platform-sdk>=0.2.0; extra == "ergon-platform"
 Requires-Dist: httpx>=0.28.0; extra == "ergon-platform"
 Provides-Extra: excel
 Requires-Dist: openpyxl>=3.1.5; extra == "excel"

{ergon_framework_python-0.2.1.dist-info → ergon_framework_python-0.2.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ergon/__init__.py,sha256=OB0tsZ-GmQWa6cIaSGHCsnf6rXE5PtWw1Dz0DQsdPbA,218
+ergon/__init__.py,sha256=o88UpQYh3RcjogyA8PsDjekDAK6Xfth93UCFdDWixyo,218
 ergon/cli.py,sha256=CF8PvOjIguR18a657aEBEyhFXUoSVtSRVXv5PfnuWtA,6021
 ergon/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ergon/bootstrap/src/__project__/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -22,17 +22,34 @@ ergon/bootstrap/src/__project__/tasks/example_task/exceptions.py,sha256=LTWwquhM
 ergon/bootstrap/src/__project__/tasks/example_task/helpers.py,sha256=uR_XZEtxR6JzbJloi7OhJb77f7Wc6pnOe1W3CHvl8vE,122
 ergon/bootstrap/src/__project__/tasks/example_task/schemas.py,sha256=3_qANKlJIQ90gaRNrDuBAM42UNzfDbLI4-F0kkjjWHc,155
 ergon/bootstrap/src/__project__/tasks/example_task/task.py,sha256=XnmVg-B0N8Ld48FMLJgGfTTJLjmpy5Nog_Gaj1ruVEQ,36
-ergon/connector/__init__.py,sha256=ybPKXBpRaOIWl6TM998vLYalKAGFGGVE5Mh6nbiEdIA,5658
+ergon/connector/__init__.py,sha256=qBgpM6yArVb-ns7l09RN6hYU84Rxx1_3aD4ZZiCFTRE,6648
 ergon/connector/_lazy.py,sha256=ANsKY3JTtmVo8Z8p3ey3LwkQGVVYITKO1eEK66IvA3U,1690
 ergon/connector/connector.py,sha256=jaF1iXKALE1IHHfcP54mTD5mvj-wzk5iNaLo96IOcDI,3944
 ergon/connector/transaction.py,sha256=LMGEFfisdjGGCPrWps3nYJ-SRBKa-P2pOFmwbSgDN-E,291
-ergon/connector/ergon_platform/__init__.py,sha256=j4CMZYLg6EgLpfMNr8dSPELC3V-ou7zoWfnZ1KIhrbY,1026
+ergon/connector/ergon_platform/__init__.py,sha256=BYp8roDI5JyKdwLaVGeE9Qnx3VWt4gpPdAdb258YKuw,3807
 ergon/connector/ergon_platform/_client.py,sha256=RQbTRanK0pF_JzJi7EF9_6d4veCC4_xfXHk5A0nL5jE,812
-ergon/connector/ergon_platform/_operations.py,sha256=-HpkrTjJm7D_rw2K4YW0s1e7B0sXxod1g0fKMDs3Cdg,9614
-ergon/connector/ergon_platform/async_connector.py,sha256=B6VArBW_e-qYV5ZK8RCnH6ZkrbTe8u_5VbISs9Xkd3M,8398
-ergon/connector/ergon_platform/connector.py,sha256=tp-ogXNlZDmt90X0wpRj5C06SHuIiO6sXmPY99Vuii4,7511
-ergon/connector/ergon_platform/models.py,sha256=qxUZE9Gu-F9n8pJz0K73NlltYJ2tLYmQfqZpXkefGf8,4366
-ergon/connector/ergon_platform/utils.py,sha256=z0pMSPCzTBeQfHYaLCQInzJuulkMVFx_gLpvRu9XKUM,8778
+ergon/connector/ergon_platform/async_connector.py,sha256=Is4wpMO9dtPNJoj1Y8WeyZg8bpskHbcpApdVYl45OYM,184
+ergon/connector/ergon_platform/connector.py,sha256=pIRBPoWTLs8q_5_2Dby7fNEMQJWsnVOrt7dTJdFZdMo,162
+ergon/connector/ergon_platform/models.py,sha256=ydrowIUaMhiZ08iJPKynkJU19RvqmogIreQFeJKhCuw,1248
+ergon/connector/ergon_platform/utils.py,sha256=eZU8PHwarpz--EWgz74MkfPLD2AbKp5pmNveijd13oc,987
+ergon/connector/ergon_platform/channels/__init__.py,sha256=wEpy-8p1mwQNZmosO17hOZ3DIPisX3P14T6WVbyMeBo,1480
+ergon/connector/ergon_platform/channels/adapters.py,sha256=LoylPshBCMi5R32Ms-IHe_H2rCO4QFrUWMvscRWkmQ8,8123
+ergon/connector/ergon_platform/channels/async_connector.py,sha256=iKqj-I_2k6DPRd-r0FZ7li659L0Rf3k5--lT3VCBd-8,12853
+ergon/connector/ergon_platform/channels/connector.py,sha256=UEgT7so-CWEVPUmxhBzkX4WnxC5PJVlhs-fDd1o1qYk,11637
+ergon/connector/ergon_platform/channels/models.py,sha256=00WTLbYdputWd9FHy9D5h3X4AAiAIBeYlBeH5cGY6vA,18292
+ergon/connector/ergon_platform/channels/services/__init__.py,sha256=EQBc2Sc2mL3gwitd1cPL2wpYnio7C7nHzAA1ml45UzQ,1091
+ergon/connector/ergon_platform/channels/services/activity.py,sha256=SxN3y2VZDulPxiqy6XNDipVvhCk9uDzsIeSXs9Hd6WA,9756
+ergon/connector/ergon_platform/channels/services/addresses.py,sha256=YNjMTqZidfqo5aOXa32F_rXIukVKxE04ULrdBBm2L4k,11288
+ergon/connector/ergon_platform/channels/services/attachments.py,sha256=IT9bU8p3X93StRFpQAofcf2nuBTpqfpqEuabAccj05M,10216
+ergon/connector/ergon_platform/channels/services/messages.py,sha256=us0gemfUAk3GKdC7-mXW0V7r7oCIbD73qIYXgy6W9Oc,4474
+ergon/connector/ergon_platform/channels/services/records.py,sha256=yv_xl0lFhksk0baHpHkFlqA8X3YaoEOQ9jpOUT5guk0,2483
+ergon/connector/ergon_platform/channels/services/service.py,sha256=BjqRYeLUZtzXpPeOLjgNdlq7TJXFc28SJtMH-Uha2DY,1160
+ergon/connector/ergon_platform/workflows/__init__.py,sha256=W5TDhYmldZqfQRJ4G8siFis7AAasIAuZhtqrwlR5QFk,1315
+ergon/connector/ergon_platform/workflows/_operations.py,sha256=IJPsiCEo27UvRKhGTrxn1HqCxbaPfVLwV0mFGgs0sOw,9750
+ergon/connector/ergon_platform/workflows/async_connector.py,sha256=PYqLxDAw1M0T6dR9SckohT1nhibPyRoBhdu9JFJ2r90,8463
+ergon/connector/ergon_platform/workflows/connector.py,sha256=gooEFbkzVB0ig-a29sD__hZJeNEQ9JfDi2mxBuLQGB0,7576
+ergon/connector/ergon_platform/workflows/models.py,sha256=_a2qH9Bkx5f2IqWb0LdqML0D088aH7AwApe204REoEE,3591
+ergon/connector/ergon_platform/workflows/utils.py,sha256=---WoYGjyAqKLl-OWOqpkvFDK8HCeiApmz47nJ3z958,8779
 ergon/connector/excel/__init__.py,sha256=YHrcMRwW96DtvRs2CNpKK5GW3lJXZSdClLUWCDMgzsA,1013
 ergon/connector/excel/connector.py,sha256=Tx2T9Ggo8pPAaSmua4j_PMu_QiiiezWm1Nru3Jr0E94,5327
 ergon/connector/excel/models.py,sha256=9RSCaFJAaFTTI0BMZ8S87SWvRqpEsMA50Mw7Gc47_dE,905
@@ -82,7 +99,7 @@ ergon/task/policies.py,sha256=3x1SV-yt3FXNgmnATdQdpRtsDtlskjx582SpbC6UH6A,7358
 ergon/task/runner.py,sha256=xUaMSO6V_iOq6QJdiyW1IA5kfWuwQjh1IJl6HUh06hY,20344
 ergon/task/utils.py,sha256=OmlP2AhLM32V8O9OMky6DdSvxwmMNn9x2LTguf0y-M0,2501
 ergon/task/mixins/__init__.py,sha256=6A8vNgdjaQYNJvymWNF-kBN2oTaQBYRLctS3Sq1w_-Y,320
-ergon/task/mixins/consumer.py,sha256=pImeLqO3tGnoV9nUOaRNbusBWj-gGg170eRAzloT_08,52508
+ergon/task/mixins/consumer.py,sha256=DhN2qBtNtB6xdKEdCokw1Z5ic2W8QfoUCm1QH_YPIFA,52270
 ergon/task/mixins/metrics.py,sha256=4bavwo8koWiN5ADD8t5qd2BVeVcgxo6ddzq-xOybeTg,13159
 ergon/task/mixins/producer.py,sha256=xENTZdZ7t0dpw5YsreQT20iTycSaj1EA4ny-60u-SWM,20427
 ergon/telemetry/__init__.py,sha256=8waeIAHEx6tMXWfw-zOEKgjrg7V8S1poG070MdAmi_8,100
@@ -92,9 +109,9 @@ ergon/telemetry/metrics.py,sha256=0T_CuNw9P5AG1U9k1-bNp1g7McyiNLUIt_zKyLOtrnA,28
 ergon/telemetry/tracing.py,sha256=_ktiZWAgB1DU-uN91Mzr0iYNXxbCN21LnKiRo5SOJxY,4454
 ergon/utils/__init__.py,sha256=M0_VCjMN7jvSPlRZLZHeFKTHAl1zu3BnAhjrzJjDfiI,57
 ergon/utils/env.py,sha256=lji4xENvfHqg2GTGnzG95VKpxxM4f3DHIKKP61xQG08,621
-ergon_framework_python-0.2.1.dist-info/licenses/LICENSE,sha256=bWCIhHzaadz_ecLdCdtwyMafeQ17s2d7_FSSuSuEcDs,1079
-ergon_framework_python-0.2.1.dist-info/METADATA,sha256=Xwenx1poSM2NQRTfwm622NA0t0YE7A6pfljS-Z3OcAI,16914
-ergon_framework_python-0.2.1.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
-ergon_framework_python-0.2.1.dist-info/entry_points.txt,sha256=reZ934txb5OrtcNBgTGqCsXa1biUDQWubD2a9fGZynE,42
-ergon_framework_python-0.2.1.dist-info/top_level.txt,sha256=KSPt6N7RzsjU7PRAalWQ9Cul7YTUxbo84yAJ-y2vLm8,6
-ergon_framework_python-0.2.1.dist-info/RECORD,,
+ergon_framework_python-0.2.2.dist-info/licenses/LICENSE,sha256=bWCIhHzaadz_ecLdCdtwyMafeQ17s2d7_FSSuSuEcDs,1079
+ergon_framework_python-0.2.2.dist-info/METADATA,sha256=VNSh11YTgbAxLcsXPSj44s5pvfEniE9QST6-JEmsK1w,16914
+ergon_framework_python-0.2.2.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+ergon_framework_python-0.2.2.dist-info/entry_points.txt,sha256=reZ934txb5OrtcNBgTGqCsXa1biUDQWubD2a9fGZynE,42
+ergon_framework_python-0.2.2.dist-info/top_level.txt,sha256=KSPt6N7RzsjU7PRAalWQ9Cul7YTUxbo84yAJ-y2vLm8,6
+ergon_framework_python-0.2.2.dist-info/RECORD,,