from decimal import Decimal
from sqlalchemy import (
    Boolean,
    ForeignKeyConstraint,
    Integer,
    Numeric,
    PrimaryKeyConstraint,
    String,
)
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy.orm.base import Mapped

from core.database import Base


class QClasses(Base):
    __tablename__ = "q_classes"  # QClases
    __table_args__ = (
        PrimaryKeyConstraint("id", name="qclases_pk"),
        ForeignKeyConstraint(
            ["tenant_id"], ["a76.tenants.id"], name="fk_qclasses_tenants"
        ),
        ForeignKeyConstraint(
            ["company_id"], ["a76.company.id"], name="fk_qclasses_company"
        ),
        ForeignKeyConstraint(["class_id"], ["classes.id"], name="fk_qclasses_classes"),
        {"schema": "a24"},
    )

    id: Mapped[int] = mapped_column(Integer, primary_key=True)
    class_id: Mapped[int] = mapped_column(Integer, nullable=False)
    tenant_id: Mapped[int] = mapped_column(Integer, nullable=False, index=True)
    company_id: Mapped[int] = mapped_column(Integer, nullable=True, index=True)

    import_tariff_code: Mapped[str] = mapped_column(String(10))  # FRACCIONIMPO
    import_tariff_type: Mapped[str] = mapped_column(String(6))  # TIPOFRACIMPO
    export_tariff_code: Mapped[str] = mapped_column(String(10))  # FRACCIONEXPO
    export_tariff_type: Mapped[str] = mapped_column(String(6))  # TIPOFRACEXPO
    depreciation_rate: Mapped[Decimal] = mapped_column(Numeric(5, 2))  # TASADEPRECIA
    fda_code: Mapped[str] = mapped_column(String(20))  # FDA
    eccn_code: Mapped[str] = mapped_column(String(20))  # ECCN
    class_enabled: Mapped[bool] = mapped_column(Boolean)  # HABILITADESHABILITACLASE