Initial Commit 🚀 🚀

api/services/workflow/disposition_mapper.py

@@ -0,0 +1,77 @@
+"""Utility module for applying disposition code mapping."""
+
+from typing import Optional
+
+from loguru import logger
+
+from api.db import db_client
+from api.enums import OrganizationConfigurationKey
+
+
+async def apply_disposition_mapping(value: str, organization_id: Optional[int]) -> str:
+    """Apply disposition code mapping if configured.
+
+    Args:
+        value: The original disposition value to map
+        organization_id: The organization ID
+
+    Returns:
+        The mapped value if found in configuration, otherwise the original value
+    """
+    if not organization_id or not value:
+        return value
+
+    try:
+        disposition_mapping = await db_client.get_configuration_value(
+            organization_id,
+            OrganizationConfigurationKey.DISPOSITION_CODE_MAPPING.value,
+            default={},
+        )
+
+        if not disposition_mapping:
+            return value
+
+        # Return mapped value if exists, otherwise original
+        # DISPOSITION_CODE_MAPPING looks like {"user_idle_max_duration_exceeded": "DAIR"} etc.
+        mapped_value = disposition_mapping.get(value, value)
+
+        if mapped_value != value:
+            logger.debug(
+                f"Mapped disposition code from '{value}' to '{mapped_value}' "
+                f"for organization {organization_id}"
+            )
+
+        return mapped_value
+
+    except Exception as e:
+        logger.error(f"Error applying disposition mapping: {e}")
+        return value
+
+
+async def get_organization_id_from_workflow_run(
+    workflow_run_id: Optional[int],
+) -> Optional[int]:
+    """Get organization_id from workflow_run_id through the model relationships.
+
+    Args:
+        workflow_run_id: The workflow run ID
+
+    Returns:
+        The organization ID if found, otherwise None
+    """
+    if not workflow_run_id:
+        return None
+
+    try:
+        workflow_run = await db_client.get_workflow_run_by_id(workflow_run_id)
+        if not workflow_run or not workflow_run.workflow:
+            return None
+
+        workflow = workflow_run.workflow
+        if not workflow.user:
+            return None
+
+        return workflow.user.selected_organization_id
+    except Exception as e:
+        logger.error(f"Error getting organization_id from workflow_run: {e}")
+        return None