shanxia/bx/bx-backend/app/models/reimburse.py

from datetime import datetime
from decimal import Decimal

from sqlalchemy import BigInteger, DateTime, ForeignKey, Integer, Numeric, String, func
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.database import Base
from app.models.base import BaseModel, SoftDeleteMixin


class ReimburseApply(BaseModel, SoftDeleteMixin):
    __tablename__ = "bx_reimburse_apply"

    apply_code: Mapped[str] = mapped_column(String(32), nullable=False, unique=True)
    title: Mapped[str] = mapped_column(String(256), nullable=False)
    applicant_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
    applicant_name: Mapped[str] = mapped_column(String(64), nullable=False)
    expense_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
    expense_name: Mapped[str] = mapped_column(String(128), nullable=False)
    amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), nullable=False)
    invoice_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0, server_default="0")
    attach_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0, server_default="0")
    status: Mapped[str] = mapped_column(String(32), nullable=False, default="draft", server_default="draft")
    current_node: Mapped[str | None] = mapped_column(String(128))
    wf_id: Mapped[int | None] = mapped_column(BigInteger)
    pay_status: Mapped[str | None] = mapped_column(String(32))
    pay_time: Mapped[datetime | None] = mapped_column(DateTime)
    pay_user_id: Mapped[int | None] = mapped_column(BigInteger)
    pay_user_name: Mapped[str | None] = mapped_column(String(64))
    actual_amount: Mapped[Decimal | None] = mapped_column(Numeric(14, 2))
    offset_advance_id: Mapped[int | None] = mapped_column(BigInteger)
    offset_amount: Mapped[Decimal | None] = mapped_column(Numeric(14, 2))
    remark: Mapped[str | None] = mapped_column(String(512))

    details: Mapped[list["ReimburseDetail"]] = relationship(
        back_populates="reimburse",
        primaryjoin="ReimburseApply.id == ReimburseDetail.reimburse_id",
        foreign_keys="[ReimburseDetail.reimburse_id]",
        order_by="ReimburseDetail.id",
    )
    approvals: Mapped[list["ReimburseApproval"]] = relationship(back_populates="apply")
    offsets: Mapped[list["ReimburseOffset"]] = relationship(back_populates="reimburse")


class ReimburseDetail(BaseModel, SoftDeleteMixin):
    __tablename__ = "bx_reimburse_detail"

    reimburse_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("bx_reimburse_apply.id"), nullable=False)
    expense_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
    expense_name: Mapped[str] = mapped_column(String(128), nullable=False)
    amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), nullable=False)
    remark: Mapped[str | None] = mapped_column(String(512))

    reimburse: Mapped["ReimburseApply"] = relationship(back_populates="details")


class ReimburseApproval(Base):
    __tablename__ = "bx_reimburse_approval"

    id: Mapped[int] = mapped_column(BigInteger, primary_key=True, autoincrement=True)
    apply_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("bx_reimburse_apply.id"), nullable=False)
    node_order: Mapped[int] = mapped_column(Integer, nullable=False)
    node_name: Mapped[str] = mapped_column(String(128), nullable=False)
    approver_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
    approver_name: Mapped[str] = mapped_column(String(64), nullable=False)
    approval_status: Mapped[str] = mapped_column(String(32), nullable=False)
    approval_opinion: Mapped[str | None] = mapped_column(String(1024))
    approval_time: Mapped[datetime] = mapped_column(DateTime, nullable=False)

    apply: Mapped["ReimburseApply"] = relationship(back_populates="approvals")


class ReimburseOffset(Base):
    __tablename__ = "bx_reimburse_offset"

    id: Mapped[int] = mapped_column(BigInteger, primary_key=True, autoincrement=True)
    reimburse_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("bx_reimburse_apply.id"), nullable=False)
    advance_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("bx_advance_apply.id"), nullable=False)
    offset_amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), nullable=False)
    remark: Mapped[str | None] = mapped_column(String(512))
    create_by: Mapped[str | None] = mapped_column(String(64))
    create_time: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())

    reimburse: Mapped["ReimburseApply"] = relationship(back_populates="offsets")
    advance: Mapped["AdvanceApply"] = relationship()