dograh/api/services/workflow/disposition_mapper.py

"""Utility module for applying disposition code mapping."""

from loguru import logger

from api.db import db_client
from api.enums import OrganizationConfigurationKey


async def apply_disposition_mapping(value: str, organization_id: int | None) -> 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