40 lines
2.1 KiB
Python
40 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import date, datetime
|
|
|
|
from sqlalchemy import String, Integer, DateTime, Boolean, ForeignKey, JSON, Date, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from adapters.db.session import Base
|
|
|
|
|
|
class Document(Base):
|
|
__tablename__ = "documents"
|
|
__table_args__ = (
|
|
UniqueConstraint('business_id', 'code', name='uq_documents_business_code'),
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(50), nullable=False, index=True)
|
|
business_id: Mapped[int] = mapped_column(Integer, ForeignKey("businesses.id", ondelete="CASCADE"), nullable=False, index=True)
|
|
fiscal_year_id: Mapped[int] = mapped_column(Integer, ForeignKey("fiscal_years.id", ondelete="RESTRICT"), nullable=False, index=True)
|
|
currency_id: Mapped[int] = mapped_column(Integer, ForeignKey("currencies.id", ondelete="RESTRICT"), nullable=False, index=True)
|
|
created_by_user_id: Mapped[int] = mapped_column(Integer, ForeignKey("users.id", ondelete="RESTRICT"), nullable=False, index=True)
|
|
registered_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
|
document_date: Mapped[date] = mapped_column(Date, nullable=False)
|
|
document_type: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
is_proforma: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
|
extra_info: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
developer_settings: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)
|
|
|
|
# Relationships
|
|
business = relationship("Business", back_populates="documents")
|
|
fiscal_year = relationship("FiscalYear", back_populates="documents")
|
|
currency = relationship("Currency", back_populates="documents")
|
|
created_by = relationship("User", foreign_keys=[created_by_user_id])
|
|
lines = relationship("DocumentLine", back_populates="document", cascade="all, delete-orphan")
|
|
|
|
|