Files
arrr-erp/app/modules/bank_statement_analyzer/models.py
T
2026-08-24 15:23:49 +05:30

55 lines
3.7 KiB
Python

from __future__ import annotations
from datetime import datetime, timezone
from sqlalchemy import Boolean, DateTime, ForeignKey, Integer, String, Text
from sqlalchemy.orm import Mapped, mapped_column
from app.core.db.common import CommonBase
class BankStatementAnalysisJob(CommonBase):
__tablename__ = "bank_statement_analysis_jobs"
id: Mapped[str] = mapped_column(String(32), primary_key=True)
tenant_id: Mapped[int | None] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=True, index=True)
branch_id: Mapped[int | None] = mapped_column(ForeignKey("branches.id", ondelete="SET NULL"), nullable=True, index=True)
user_id: Mapped[int] = mapped_column(ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True)
role_bucket: Mapped[str] = mapped_column(String(30), nullable=False)
# Phase 11B: optional client-aware analysis. Null preserves the existing
# standalone Work-folder behavior exactly.
client_id: Mapped[int | None] = mapped_column(ForeignKey("clients.id", ondelete="SET NULL"), nullable=True, index=True)
engagement_id: Mapped[int | None] = mapped_column(ForeignKey("client_service_subscriptions.id", ondelete="SET NULL"), nullable=True, index=True)
ownership_confirmation: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
ownership_status: Mapped[str] = mapped_column(String(30), nullable=False, default="not_checked", index=True)
ownership_validation_json: Mapped[str | None] = mapped_column(Text, nullable=True)
engagement_archive_status: Mapped[str] = mapped_column(String(30), nullable=False, default="not_requested", index=True)
engagement_archive_json: Mapped[str | None] = mapped_column(Text, nullable=True)
selected_bank: Mapped[str] = mapped_column(String(40), nullable=False, default="auto")
financial_year: Mapped[str | None] = mapped_column(String(20), nullable=True)
customer_override: Mapped[str | None] = mapped_column(String(255), nullable=True)
account_override: Mapped[str | None] = mapped_column(String(100), nullable=True)
classification_enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
# Phase 19: one analyzer, three controlled uses. Existing jobs default to
# analyze_only and preserve the historical Bank Analyzer behaviour.
purpose: Mapped[str] = mapped_column(String(30), nullable=False, default="analyze_only", index=True)
accounting_import_status: Mapped[str] = mapped_column(String(30), nullable=False, default="not_requested", index=True)
accounting_import_json: Mapped[str | None] = mapped_column(Text, nullable=True)
status: Mapped[str] = mapped_column(String(20), nullable=False, default="queued", index=True)
progress_percent: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
file_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
input_files_json: Mapped[str] = mapped_column(Text, nullable=False)
job_directory: Mapped[str] = mapped_column(Text, nullable=False)
output_file: Mapped[str | None] = mapped_column(Text, nullable=True)
summary_json: Mapped[str | None] = mapped_column(Text, nullable=True)
error_message: Mapped[str | None] = mapped_column(Text, nullable=True)
submitted_at_utc: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=lambda: datetime.now(timezone.utc), index=True)
started_at_utc: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
completed_at_utc: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
expires_at_utc: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)