52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
import enum
|
|
|
|
from sqlalchemy import String, Integer, DateTime, ForeignKey, Enum as SAEnum, func
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.core.database import Base
|
|
|
|
|
|
class SourceApp(str, enum.Enum):
|
|
wechat = "wechat"
|
|
alipay = "alipay"
|
|
bank = "bank"
|
|
digital_wallet = "digital_wallet"
|
|
other = "other"
|
|
|
|
|
|
class PageType(str, enum.Enum):
|
|
bill_list = "bill_list"
|
|
bill_detail = "bill_detail"
|
|
transfer_receipt = "transfer_receipt"
|
|
sms_notice = "sms_notice"
|
|
balance = "balance"
|
|
unknown = "unknown"
|
|
|
|
|
|
class OcrStatus(str, enum.Enum):
|
|
pending = "pending"
|
|
processing = "processing"
|
|
done = "done"
|
|
failed = "failed"
|
|
|
|
|
|
class EvidenceImage(Base):
|
|
__tablename__ = "evidence_images"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
case_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("cases.id"), index=True)
|
|
file_path: Mapped[str] = mapped_column(String(512))
|
|
thumb_path: Mapped[str] = mapped_column(String(512), default="")
|
|
source_app: Mapped[SourceApp] = mapped_column(SAEnum(SourceApp), default=SourceApp.other)
|
|
page_type: Mapped[PageType] = mapped_column(SAEnum(PageType), default=PageType.unknown)
|
|
ocr_status: Mapped[OcrStatus] = mapped_column(SAEnum(OcrStatus), default=OcrStatus.pending)
|
|
file_hash: Mapped[str] = mapped_column(String(128), unique=True, index=True)
|
|
file_size: Mapped[int] = mapped_column(Integer, default=0)
|
|
uploaded_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
case = relationship("Case", back_populates="images")
|
|
ocr_blocks = relationship("OcrBlock", back_populates="image", lazy="selectin")
|