Add client financial year Accounting Mirror registry

app/modules/accounting/accounting_mirror_registry.py

@@ -0,0 +1,181 @@
+from __future__ import annotations
+
+from datetime import date, datetime, timezone
+from pathlib import Path
+import re
+from typing import Any
+
+from sqlalchemy import select
+
+from app.modules.accounting.accounting_mirror_models import AccountingMirrorRegistry
+
+
+def _utcnow() -> datetime:
+    return datetime.now(timezone.utc)
+
+
+def _fy_key(value: str) -> int:
+    match = re.fullmatch(r"(\d{4})-(\d{2})", str(value or "").strip())
+    return int(match.group(1)) if match else -1
+
+
+def _get_any_mirror(db, tenant_id: int, client_id: int, financial_year: str) -> AccountingMirrorRegistry | None:
+    return db.execute(
+        select(AccountingMirrorRegistry).where(
+            AccountingMirrorRegistry.tenant_id == int(tenant_id),
+            AccountingMirrorRegistry.client_id == int(client_id),
+            AccountingMirrorRegistry.financial_year == str(financial_year or "").strip(),
+        )
+    ).scalar_one_or_none()
+
+
+def get_registered_mirror(db, tenant_id: int, client_id: int, financial_year: str) -> AccountingMirrorRegistry | None:
+    row = _get_any_mirror(db, tenant_id, client_id, financial_year)
+    return row if row and row.is_active and row.status == "active" else None
+
+
+def list_registered_mirrors(db, tenant_id: int, client_id: int) -> list[AccountingMirrorRegistry]:
+    rows = db.execute(
+        select(AccountingMirrorRegistry).where(
+            AccountingMirrorRegistry.tenant_id == int(tenant_id),
+            AccountingMirrorRegistry.client_id == int(client_id),
+            AccountingMirrorRegistry.is_active.is_(True),
+        )
+    ).scalars().all()
+    return sorted(rows, key=lambda row: _fy_key(row.financial_year), reverse=True)
+
+
+def _parse_iso_date(value: Any) -> date | None:
+    text = str(value or "").strip()
+    if not text:
+        return None
+    try:
+        return date.fromisoformat(text[:10])
+    except Exception:
+        return None
+
+
+def _parse_iso_datetime(value: Any) -> datetime | None:
+    text = str(value or "").strip()
+    if not text:
+        return None
+    try:
+        parsed = datetime.fromisoformat(text.replace("Z", "+00:00"))
+        return parsed if parsed.tzinfo else parsed.replace(tzinfo=timezone.utc)
+    except Exception:
+        return None
+
+
+def upsert_registered_mirror(
+    db,
+    *,
+    tenant_id: int,
+    client_id: int,
+    financial_year: str,
+    accounting_relative_dir: str,
+    storage_node_id: int | None,
+    mirror: dict[str, Any] | None = None,
+    job: dict[str, Any] | None = None,
+    requested_by_user_id: int | None = None,
+    replacement: bool = False,
+) -> AccountingMirrorRegistry:
+    mirror = dict(mirror or {})
+    job = dict(job or {})
+    company = dict(mirror.get("company") or {})
+    voucher_period = dict(mirror.get("voucher_period") or {})
+
+    row = _get_any_mirror(db, tenant_id, client_id, financial_year)
+    created = row is None
+    if row is None:
+        row = AccountingMirrorRegistry(
+            tenant_id=int(tenant_id),
+            client_id=int(client_id),
+            financial_year=str(financial_year).strip(),
+            accounting_relative_dir=str(accounting_relative_dir or "").strip(),
+            mirror_file_name=f"client_{int(client_id):08d}.act",
+            created_by_user_id=requested_by_user_id,
+        )
+        db.add(row)
+
+    row.storage_node_id = int(storage_node_id) if storage_node_id else None
+    row.accounting_relative_dir = str(accounting_relative_dir or row.accounting_relative_dir or "").strip()
+    row.mirror_file_name = Path(str(job.get("accounting_db_path") or row.mirror_file_name or f"client_{int(client_id):08d}.act")).name
+    row.company_name = str(
+        job.get("company_name")
+        or company.get("company_name")
+        or row.company_name
+        or ""
+    ).strip()
+    row.company_guid = str(
+        job.get("tally_guid")
+        or company.get("company_guid")
+        or row.company_guid
+        or ""
+    ).strip()
+    row.voucher_from_date = _parse_iso_date(voucher_period.get("from_date")) or row.voucher_from_date
+    row.voucher_to_date = _parse_iso_date(voucher_period.get("to_date")) or row.voucher_to_date
+    try:
+        row.file_size_bytes = int(mirror.get("size_bytes") or row.file_size_bytes or 0)
+    except Exception:
+        pass
+    row.status = "active"
+    row.is_active = True
+    row.last_synced_by_user_id = requested_by_user_id
+    row.last_synced_at_utc = (
+        _parse_iso_datetime(job.get("finished_at_utc"))
+        or _parse_iso_datetime(job.get("updated_at_utc"))
+        or _utcnow()
+    )
+    if replacement and not created:
+        row.replacement_count = int(row.replacement_count or 0) + 1
+    row.updated_at_utc = _utcnow()
+    db.flush()
+    return row
+
+
+def sync_discovered_mirrors(
+    db,
+    *,
+    tenant_id: int,
+    client_id: int,
+    storage_node_id: int | None,
+    discovered: list[dict[str, Any]],
+    requested_by_user_id: int | None = None,
+) -> list[AccountingMirrorRegistry]:
+    output: list[AccountingMirrorRegistry] = []
+    discovered_fys: set[str] = set()
+    for item in discovered or []:
+        fy = str(item.get("financial_year") or "").strip()
+        relative_dir = str(item.get("accounting_relative_dir") or "").strip()
+        mirror = dict(item.get("mirror") or {})
+        if not re.fullmatch(r"\d{4}-\d{2}", fy) or not relative_dir or not mirror.get("ready"):
+            continue
+        discovered_fys.add(fy)
+        row = upsert_registered_mirror(
+            db,
+            tenant_id=tenant_id,
+            client_id=client_id,
+            financial_year=fy,
+            accounting_relative_dir=relative_dir,
+            storage_node_id=storage_node_id,
+            mirror=mirror,
+            job=item,
+            requested_by_user_id=requested_by_user_id,
+            replacement=False,
+        )
+        output.append(row)
+
+    existing = db.execute(
+        select(AccountingMirrorRegistry).where(
+            AccountingMirrorRegistry.tenant_id == int(tenant_id),
+            AccountingMirrorRegistry.client_id == int(client_id),
+            AccountingMirrorRegistry.is_active.is_(True),
+        )
+    ).scalars().all()
+    for row in existing:
+        if row.financial_year not in discovered_fys:
+            row.status = "missing"
+            row.is_active = False
+            row.updated_at_utc = _utcnow()
+    db.commit()
+    return output