36 lines
1.9 KiB
Python
36 lines
1.9 KiB
Python
|
"""Transaction model - 交易记录."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
from datetime import datetime
|
||
|
|
from decimal import Decimal
|
||
|
|
from sqlalchemy import String, Text, DateTime, Numeric, ForeignKey
|
||
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||
|
|
|
||
|
|
from app.models.database import Base
|
||
|
|
|
||
|
|
|
||
|
|
class Transaction(Base):
|
||
|
|
__tablename__ = "transactions"
|
||
|
|
|
||
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
||
|
|
case_id: Mapped[int] = mapped_column(ForeignKey("cases.id", ondelete="CASCADE"), index=True)
|
||
|
|
screenshot_id: Mapped[int] = mapped_column(ForeignKey("screenshots.id", ondelete="CASCADE"), index=True)
|
||
|
|
app_source: Mapped[str] = mapped_column(String(128))
|
||
|
|
transaction_type: Mapped[str] = mapped_column(String(32)) # 转出/转入/消费/收款/提现/充值
|
||
|
|
amount: Mapped[Decimal] = mapped_column(Numeric(18, 2))
|
||
|
|
currency: Mapped[str] = mapped_column(String(16), default="CNY")
|
||
|
|
counterparty_name: Mapped[str | None] = mapped_column(String(256), nullable=True)
|
||
|
|
counterparty_account: Mapped[str | None] = mapped_column(String(512), nullable=True)
|
||
|
|
order_number: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
||
|
|
transaction_time: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
||
|
|
remark: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
|
|
raw_text: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
|
|
confidence: Mapped[str] = mapped_column(String(16), default="medium") # high | medium | low
|
||
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||
|
|
|
||
|
|
case: Mapped["Case"] = relationship("Case", back_populates="transactions")
|
||
|
|
screenshot: Mapped["Screenshot"] = relationship("Screenshot", back_populates="transactions")
|
||
|
|
|
||
|
|
def __repr__(self) -> str:
|
||
|
|
return f"<Transaction(id={self.id}, amount={self.amount}, app={self.app_source})>"
|