document-processor/app/models/extracted_field.py

42 lines
1.7 KiB
Python

from datetime import datetime, date
from decimal import Decimal
from sqlalchemy import String, DateTime, Date, ForeignKey, Numeric, JSON
from sqlalchemy.orm import Mapped, mapped_column, relationship
from app.db.base import Base
class ExtractedField(Base):
__tablename__ = "extracted_fields"
id: Mapped[int] = mapped_column(primary_key=True, index=True)
document_id: Mapped[int] = mapped_column(
ForeignKey("documents.id"), nullable=False, index=True
)
merchant_raw: Mapped[str | None] = mapped_column(String(255), nullable=True)
merchant_normalized: Mapped[str | None] = mapped_column(String(255), nullable=True)
transaction_date: Mapped[date | None] = mapped_column(Date, nullable=True)
subtotal: Mapped[Decimal | None] = mapped_column(Numeric(12, 2), nullable=True)
tax: Mapped[Decimal | None] = mapped_column(Numeric(12, 2), nullable=True)
total: Mapped[Decimal | None] = mapped_column(Numeric(12, 2), nullable=True)
currency: Mapped[str | None] = mapped_column(String(10), nullable=True)
payment_method: Mapped[str | None] = mapped_column(String(100), nullable=True)
receipt_number: Mapped[str | None] = mapped_column(String(255), nullable=True)
location: Mapped[str | None] = mapped_column(String(255), nullable=True)
counterparty: Mapped[str | None] = mapped_column(String(255), nullable=True)
extra_json: 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
)
document: Mapped["Document"] = relationship(back_populates="extracted_fields")