arrr-erp/app/modules/accounting/accounting_mirror_service.py

from __future__ import annotations

from typing import Any

from app.modules.accounting.agent_bridge import request_agent_command


DEFAULT_QUERY_TIMEOUT_SECONDS = 45
DEFAULT_SYNC_TIMEOUT_SECONDS = 330


class AccountingMirrorError(RuntimeError):
    pass


def _unwrap(response: dict[str, Any]) -> dict[str, Any]:
    if not isinstance(response, dict):
        raise AccountingMirrorError("ERP Local Agent returned an invalid response.")
    if not response.get("ok"):
        raise AccountingMirrorError(str(response.get("error") or "ERP Local Agent command failed."))
    result = response.get("result")
    if not isinstance(result, dict):
        raise AccountingMirrorError("ERP Local Agent returned an invalid mirror result.")
    return result


def mirror_status(*, node_code: str, accounting_payload: dict[str, Any]) -> dict[str, Any]:
    response = request_agent_command(
        node_code,
        "accounting_mirror_status",
        accounting_payload,
        timeout_seconds=DEFAULT_QUERY_TIMEOUT_SECONDS,
    )
    return _unwrap(response)


def mirror_sync(
    *,
    node_code: str,
    accounting_payload: dict[str, Any],
    company_name: str = "",
    company_guid: str = "",
    dsn: str = "TallyODBC64_9000",
    timeout_seconds: int = DEFAULT_SYNC_TIMEOUT_SECONDS,
) -> dict[str, Any]:
    payload = dict(accounting_payload or {})
    payload.update(
        {
            "company_name": str(company_name or "").strip(),
            "company_guid": str(company_guid or "").strip(),
            "dsn": str(dsn or "TallyODBC64_9000").strip(),
            "timeout_seconds": max(60, int(timeout_seconds)),
        }
    )
    response = request_agent_command(
        node_code,
        "accounting_mirror_sync",
        payload,
        timeout_seconds=max(60, int(timeout_seconds)),
    )
    return _unwrap(response)


def mirror_query(
    *,
    node_code: str,
    accounting_payload: dict[str, Any],
    query: str,
    filters: dict[str, Any] | None = None,
    timeout_seconds: int = DEFAULT_QUERY_TIMEOUT_SECONDS,
) -> dict[str, Any]:
    payload = dict(accounting_payload or {})
    payload.update({"query": str(query or "").strip(), "filters": filters or {}})
    response = request_agent_command(
        node_code,
        "accounting_mirror_query",
        payload,
        timeout_seconds=max(10, int(timeout_seconds)),
    )
    return _unwrap(response)


def daybook(*, node_code: str, accounting_payload: dict[str, Any], from_date="", to_date="", limit=500):
    return mirror_query(
        node_code=node_code,
        accounting_payload=accounting_payload,
        query="daybook",
        filters={"from_date": from_date, "to_date": to_date, "limit": limit},
    )


def trial_balance(*, node_code: str, accounting_payload: dict[str, Any], limit=5000):
    return mirror_query(
        node_code=node_code,
        accounting_payload=accounting_payload,
        query="trial_balance",
        filters={"limit": limit},
    )


def ledger_transactions(
    *,
    node_code: str,
    accounting_payload: dict[str, Any],
    ledger_name: str,
    from_date="",
    to_date="",
    limit=5000,
):
    return mirror_query(
        node_code=node_code,
        accounting_payload=accounting_payload,
        query="ledger_transactions",
        filters={
            "ledger_name": ledger_name,
            "from_date": from_date,
            "to_date": to_date,
            "limit": limit,
        },
    )


def inventory(
    *,
    node_code: str,
    accounting_payload: dict[str, Any],
    stock_item_name: str = "",
    from_date="",
    to_date="",
    limit=5000,
):
    return mirror_query(
        node_code=node_code,
        accounting_payload=accounting_payload,
        query="inventory",
        filters={
            "stock_item_name": stock_item_name,
            "from_date": from_date,
            "to_date": to_date,
            "limit": limit,
        },
    )


def stock_items(*, node_code: str, accounting_payload: dict[str, Any], limit=5000):
    return mirror_query(
        node_code=node_code,
        accounting_payload=accounting_payload,
        query="stock_items",
        filters={"limit": limit},
    )


def hsn_history(*, node_code: str, accounting_payload: dict[str, Any], stock_item_name: str):
    return mirror_query(
        node_code=node_code,
        accounting_payload=accounting_payload,
        query="hsn_history",
        filters={"stock_item_name": stock_item_name},
    )


def gst_history(*, node_code: str, accounting_payload: dict[str, Any], stock_item_name: str):
    return mirror_query(
        node_code=node_code,
        accounting_payload=accounting_payload,
        query="gst_history",
        filters={"stock_item_name": stock_item_name},
    )


def cash_transactions(
    *,
    node_code: str,
    accounting_payload: dict[str, Any],
    ledger_name: str = "Cash",
    from_date="",
    to_date="",
    limit=5000,
):
    return mirror_query(
        node_code=node_code,
        accounting_payload=accounting_payload,
        query="cash_transactions",
        filters={
            "ledger_name": ledger_name,
            "from_date": from_date,
            "to_date": to_date,
            "limit": limit,
        },
    )


def voucher(*, node_code: str, accounting_payload: dict[str, Any], voucher_guid: str):
    return mirror_query(
        node_code=node_code,
        accounting_payload=accounting_payload,
        query="voucher",
        filters={"voucher_guid": voucher_guid},
    )


def mirror_exceptions(*, node_code: str, accounting_payload: dict[str, Any], limit=5000):
    return mirror_query(
        node_code=node_code,
        accounting_payload=accounting_payload,
        query="exceptions",
        filters={"limit": limit},
    )