98 lines
4.1 KiB
Python
98 lines
4.1 KiB
Python
from datetime import datetime
|
|
|
|
from sqlalchemy import Boolean, DateTime, Integer, String, Text
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.db.base import Base
|
|
from app.models.document_additional_field_version import DocumentAdditionalFieldVersion
|
|
from app.models.extracted_field_version import ExtractedFieldVersion
|
|
from app.models.document_line_item_set import DocumentLineItemSet
|
|
from app.models.document_line_item_set_version import DocumentLineItemSetVersion
|
|
|
|
|
|
class Document(Base):
|
|
__tablename__ = "documents"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True, index=True)
|
|
document_id: Mapped[str] = mapped_column(String(64), unique=True, index=True, nullable=False)
|
|
|
|
document_type: Mapped[str | None] = mapped_column(String(50), nullable=True)
|
|
|
|
source_path: Mapped[str] = mapped_column(Text, nullable=False)
|
|
original_path: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
current_path: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
share_path: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
original_filename: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
canonical_filename: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
|
|
mime_type: Mapped[str | None] = mapped_column(String(100), nullable=True)
|
|
file_size: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
page_count: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
|
|
sha256_original: Mapped[str | None] = mapped_column(String(64), nullable=True)
|
|
sha256_current: Mapped[str | None] = mapped_column(String(64), nullable=True)
|
|
|
|
storage_status: Mapped[str] = mapped_column(String(50), default="ingested", nullable=False)
|
|
review_status: Mapped[str] = mapped_column(String(50), default="ingested", nullable=False)
|
|
|
|
is_trashed: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)
|
|
trashed_at: Mapped[datetime | None] = mapped_column(DateTime, 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,
|
|
)
|
|
|
|
versions: Mapped[list["DocumentVersion"]] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
)
|
|
text_versions: Mapped[list["TextVersion"]] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
)
|
|
extracted_fields: Mapped[list["ExtractedField"]] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
)
|
|
layer1_candidates: Mapped[list["Layer1Candidate"]] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
)
|
|
receipt_line_items: Mapped[list["ReceiptLineItem"]] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
)
|
|
naming_fields: Mapped[list["DocumentNamingField"]] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
)
|
|
additional_fields: Mapped[list["DocumentAdditionalField"]] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
)
|
|
extracted_field_versions: Mapped[list["ExtractedFieldVersion"]] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
order_by="ExtractedFieldVersion.version_number",
|
|
)
|
|
additional_field_versions: Mapped[list["DocumentAdditionalFieldVersion"]] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
order_by="DocumentAdditionalFieldVersion.version_number",
|
|
)
|
|
line_item_set: Mapped["DocumentLineItemSet | None"] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
uselist=False,
|
|
)
|
|
line_item_set_versions: Mapped[list["DocumentLineItemSetVersion"]] = relationship(
|
|
back_populates="document",
|
|
cascade="all, delete-orphan",
|
|
order_by="DocumentLineItemSetVersion.version_number",
|
|
)
|