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

from __future__ import annotations

from datetime import datetime, timezone

from sqlalchemy import (
    DateTime,
    Float,
    ForeignKey,
    Integer,
    String,
    Text,
    UniqueConstraint,
)
from sqlalchemy.orm import Mapped, mapped_column

from app.core.db.common import CommonBase


def _utcnow():
    return datetime.now(timezone.utc)


class AccountingStockSyncRun(CommonBase):
    __tablename__ = "accounting_stock_sync_runs"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    tenant_id: Mapped[int] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True)
    client_id: Mapped[int] = mapped_column(ForeignKey("clients.id", ondelete="CASCADE"), nullable=False, index=True)
    tally_guid: Mapped[str] = mapped_column(String(120), nullable=False, index=True)
    company_name: Mapped[str] = mapped_column(String(255), nullable=False, default="")
    status: Mapped[str] = mapped_column(String(30), nullable=False, default="completed", index=True)
    stock_group_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    stock_item_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    unit_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    requested_by_user_id: Mapped[int | None] = mapped_column(ForeignKey("users.id", ondelete="SET NULL"), nullable=True)
    created_at_utc: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=_utcnow, index=True)


class AccountingStockGroup(CommonBase):
    __tablename__ = "accounting_stock_groups"
    __table_args__ = (
        UniqueConstraint("tenant_id", "client_id", "tally_guid", "name", name="uq_accounting_stock_group"),
    )

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    tenant_id: Mapped[int] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True)
    client_id: Mapped[int] = mapped_column(ForeignKey("clients.id", ondelete="CASCADE"), nullable=False, index=True)
    tally_guid: Mapped[str] = mapped_column(String(120), nullable=False, index=True)
    company_name: Mapped[str] = mapped_column(String(255), nullable=False, default="")
    master_guid: Mapped[str] = mapped_column(String(120), nullable=False, default="", index=True)
    name: Mapped[str] = mapped_column(String(255), nullable=False, index=True)
    parent_name: Mapped[str] = mapped_column(String(255), nullable=False, default="", index=True)
    base_units: Mapped[str] = mapped_column(String(80), nullable=False, default="")
    source_payload_json: Mapped[str] = mapped_column(Text, nullable=False, default="{}")
    synced_at_utc: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=_utcnow, index=True)


class AccountingStockItem(CommonBase):
    __tablename__ = "accounting_stock_items"
    __table_args__ = (
        UniqueConstraint("tenant_id", "client_id", "tally_guid", "name", name="uq_accounting_stock_item"),
    )

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    tenant_id: Mapped[int] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True)
    client_id: Mapped[int] = mapped_column(ForeignKey("clients.id", ondelete="CASCADE"), nullable=False, index=True)
    tally_guid: Mapped[str] = mapped_column(String(120), nullable=False, index=True)
    company_name: Mapped[str] = mapped_column(String(255), nullable=False, default="")
    master_guid: Mapped[str] = mapped_column(String(120), nullable=False, default="", index=True)
    name: Mapped[str] = mapped_column(String(255), nullable=False, index=True)
    parent_group_name: Mapped[str] = mapped_column(String(255), nullable=False, default="", index=True)
    category_name: Mapped[str] = mapped_column(String(255), nullable=False, default="")
    base_units: Mapped[str] = mapped_column(String(80), nullable=False, default="", index=True)
    additional_units: Mapped[str] = mapped_column(String(80), nullable=False, default="")
    hsn_code: Mapped[str] = mapped_column(String(40), nullable=False, default="", index=True)
    gst_applicable: Mapped[str] = mapped_column(String(80), nullable=False, default="")
    gst_type_of_supply: Mapped[str] = mapped_column(String(80), nullable=False, default="")
    opening_balance: Mapped[float] = mapped_column(Float, nullable=False, default=0)
    opening_value: Mapped[float] = mapped_column(Float, nullable=False, default=0)
    opening_rate: Mapped[str] = mapped_column(String(80), nullable=False, default="")
    normalized_name: Mapped[str] = mapped_column(String(255), nullable=False, default="", index=True)
    source_payload_json: Mapped[str] = mapped_column(Text, nullable=False, default="{}")
    synced_at_utc: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=_utcnow, index=True)


class AccountingStockUnit(CommonBase):
    __tablename__ = "accounting_stock_units"
    __table_args__ = (
        UniqueConstraint("tenant_id", "client_id", "tally_guid", "name", name="uq_accounting_stock_unit"),
    )

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    tenant_id: Mapped[int] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True)
    client_id: Mapped[int] = mapped_column(ForeignKey("clients.id", ondelete="CASCADE"), nullable=False, index=True)
    tally_guid: Mapped[str] = mapped_column(String(120), nullable=False, index=True)
    company_name: Mapped[str] = mapped_column(String(255), nullable=False, default="")
    master_guid: Mapped[str] = mapped_column(String(120), nullable=False, default="", index=True)
    name: Mapped[str] = mapped_column(String(80), nullable=False, index=True)
    original_name: Mapped[str] = mapped_column(String(80), nullable=False, default="")
    is_simple_unit: Mapped[str] = mapped_column(String(30), nullable=False, default="")
    base_units: Mapped[str] = mapped_column(String(80), nullable=False, default="")
    additional_units: Mapped[str] = mapped_column(String(80), nullable=False, default="")
    conversion: Mapped[str] = mapped_column(String(80), nullable=False, default="")
    normalized_name: Mapped[str] = mapped_column(String(80), nullable=False, default="", index=True)
    source_payload_json: Mapped[str] = mapped_column(Text, nullable=False, default="{}")
    synced_at_utc: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=_utcnow, index=True)


class AccountingStockItemMapping(CommonBase):
    __tablename__ = "accounting_stock_item_mappings"
    __table_args__ = (
        UniqueConstraint(
            "tenant_id", "client_id", "tally_guid", "source_type",
            "party_key", "source_description_norm", "hsn_sac",
            name="uq_accounting_stock_item_mapping_source",
        ),
    )

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    tenant_id: Mapped[int] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True)
    client_id: Mapped[int] = mapped_column(ForeignKey("clients.id", ondelete="CASCADE"), nullable=False, index=True)
    tally_guid: Mapped[str] = mapped_column(String(120), nullable=False, index=True)
    source_type: Mapped[str] = mapped_column(String(50), nullable=False, default="generic", index=True)
    party_key: Mapped[str] = mapped_column(String(255), nullable=False, default="", index=True)
    source_description: Mapped[str] = mapped_column(String(500), nullable=False)
    source_description_norm: Mapped[str] = mapped_column(String(500), nullable=False, index=True)
    hsn_sac: Mapped[str] = mapped_column(String(40), nullable=False, default="", index=True)
    stock_item_id: Mapped[int] = mapped_column(ForeignKey("accounting_stock_items.id", ondelete="CASCADE"), nullable=False, index=True)
    confidence: Mapped[int] = mapped_column(Integer, nullable=False, default=100)
    mapping_origin: Mapped[str] = mapped_column(String(30), nullable=False, default="manual")
    note: Mapped[str] = mapped_column(Text, nullable=False, default="")
    created_by_user_id: Mapped[int | None] = mapped_column(ForeignKey("users.id", ondelete="SET NULL"), nullable=True)
    updated_by_user_id: Mapped[int | None] = mapped_column(ForeignKey("users.id", ondelete="SET NULL"), nullable=True)
    created_at_utc: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=_utcnow)
    updated_at_utc: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=_utcnow)


class AccountingStockUnitMapping(CommonBase):
    __tablename__ = "accounting_stock_unit_mappings"
    __table_args__ = (
        UniqueConstraint(
            "tenant_id", "client_id", "tally_guid", "source_unit_norm",
            name="uq_accounting_stock_unit_mapping_source",
        ),
    )

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    tenant_id: Mapped[int] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True)
    client_id: Mapped[int] = mapped_column(ForeignKey("clients.id", ondelete="CASCADE"), nullable=False, index=True)
    tally_guid: Mapped[str] = mapped_column(String(120), nullable=False, index=True)
    source_unit: Mapped[str] = mapped_column(String(80), nullable=False)
    source_unit_norm: Mapped[str] = mapped_column(String(80), nullable=False, index=True)
    target_unit_id: Mapped[int] = mapped_column(ForeignKey("accounting_stock_units.id", ondelete="CASCADE"), nullable=False, index=True)
    quantity_factor: Mapped[float] = mapped_column(Float, nullable=False, default=1.0)
    note: Mapped[str] = mapped_column(Text, nullable=False, default="")
    created_by_user_id: Mapped[int | None] = mapped_column(ForeignKey("users.id", ondelete="SET NULL"), nullable=True)
    updated_at_utc: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=_utcnow)