fund-tracer/backend/app/models/screenshot.py

"""Screenshot model - 截图记录."""

from __future__ import annotations
from datetime import datetime
from sqlalchemy import String, Text, DateTime, ForeignKey
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.models.database import Base


class Screenshot(Base):
    __tablename__ = "screenshots"

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    case_id: Mapped[int] = mapped_column(ForeignKey("cases.id", ondelete="CASCADE"), index=True)
    filename: Mapped[str] = mapped_column(String(255))
    file_path: Mapped[str] = mapped_column(String(512))
    status: Mapped[str] = mapped_column(String(32), default="pending")  # pending | processing | extracted | failed
    progress_step: Mapped[str | None] = mapped_column(String(64), nullable=True, default=None)
    progress_percent: Mapped[int] = mapped_column(default=0)
    progress_detail: Mapped[str | None] = mapped_column(String(255), nullable=True, default=None)
    started_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True, default=None)
    finished_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True, default=None)
    duration_ms: Mapped[int | None] = mapped_column(nullable=True, default=None)
    error_message: Mapped[str | None] = mapped_column(Text, nullable=True, default=None)
    created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)

    case: Mapped["Case"] = relationship("Case", back_populates="screenshots")
    transactions: Mapped[list["Transaction"]] = relationship(
        "Transaction", back_populates="screenshot", cascade="all, delete-orphan"
    )

    def __repr__(self) -> str:
        return f"<Screenshot(id={self.id}, filename={self.filename})>"