payment-platform/backend/app/plugin/module_payment/expense/quota/model.py

from datetime import datetime
from decimal import Decimal

from sqlalchemy import DateTime, Numeric, String, Text
from sqlalchemy.orm import Mapped, mapped_column

from app.common.enums import PermissionFilterStrategy
from app.core.base_model import PaymentModelMixin, TenantMixin, EnterpriseMixin

from .enums import QuotaStatusEnum


class QuotaModel(PaymentModelMixin, TenantMixin, EnterpriseMixin):
    """额度模型"""

    __tablename__ = "pay_expense_quota"
    __table_args__ = {"comment": "额度表"}
    __permission_strategy__ = PermissionFilterStrategy.ENTERPRISE_BASED

    employee_id: Mapped[str] = mapped_column(
        String(64), index=True, comment="员工ID"
    )
    institution_id: Mapped[str] = mapped_column(
        String(64), index=True, comment="制度ID"
    )
    out_biz_no: Mapped[str | None] = mapped_column(
        String(64), unique=True, index=True, comment="外部业务编号"
    )
    quota_id: Mapped[str | None] = mapped_column(
        String(64), comment="额度ID(支付宝返回)或关联的发放规则ID(issue_rule_id)"
    )
    total_amount: Mapped[Decimal | None] = mapped_column(
        Numeric(12, 2), default=0, comment="总金额"
    )
    available_amount: Mapped[Decimal | None] = mapped_column(
        Numeric(12, 2), default=0, comment="可用金额"
    )
    valid_from: Mapped[datetime | None] = mapped_column(
        DateTime, comment="有效期开始"
    )
    valid_to: Mapped[datetime | None] = mapped_column(
        DateTime, comment="有效期结束"
    )
    status: Mapped[str] = mapped_column(
        String(32),
        default=QuotaStatusEnum.QUOTA_ACTIVE.value,
        comment="状态: QUOTA_ACTIVE/QUOTA_FROZEN/QUOTA_EXHAUSTED/QUOTA_EXPIRED"
    )