payment-platform/backend/app/plugin/module_payment/account/schema.py

from datetime import datetime
from decimal import Decimal
from typing import List, Optional

from pydantic import BaseModel, ConfigDict, Field, field_serializer, field_validator, model_validator


class BankcardExtInfoSchema(BaseModel):
    """银行卡信息"""

    # account_type｜收款账户类型
    # 【描述】收款账户类型。 1：对公（在金融机构开设的公司账户）,如果银行卡为对公，必须传递省市支行信息或者联行号
    #  2：对私（在金融机构开设的个人账户）
    account_type: str = Field(description="收款账户类型: 1/2。对公: 1，对私: 2")
    # inst_name｜机构名称
    # 【描述】当收款账户为对公账户时，机构名称必填；当收款账户为对私账户时，机构名称可为空。
    inst_name: Optional[str] = Field(default=None, description="机构名称")
    # 【描述】银行所在省份
    inst_province: Optional[str] = Field(default=None, description="银行所在省份")
    # inst_city｜收款银行所在市
    # 【描述】收款银行所在市
    inst_city: Optional[str] = Field(default=None, description="收款银行所在市")
    # inst_branch_name｜收款银行所属支行
    # 【描述】收款银行所属支行
    inst_branch_name: Optional[str] = Field(default=None, description="收款银行所属支行")
    # bank_code｜银行支行联行号
    # 【描述】银行支行联行号
    bank_code: Optional[str] = Field(default=None, description="银行支行联行号")

    @field_validator("account_type")
    def validate_account_type(cls, v: str) -> str:
        if v not in ["1", "2"]:
            raise ValueError(f"account_type 必须是 {['1', '2']}")
        return v
    
    @model_validator(mode='after')
    def validate_inst_name(self) -> 'BankcardExtInfoSchema':
        """验证对公账户时inst_name必填"""
        if self.account_type == "1" and self.inst_name is None:
            raise ValueError("当 account_type 是 1 时，inst_name 必填")
        return self

class PayeeInfoSchema(BaseModel):
    """收款方信息"""

    identity_type: str = Field(description="收款方类型: ALIPAY_ACCOUNT/BANK_CARD/BOOK")
    name: str = Field(description="收款方真实姓名")
    identity: str = Field(description="收款方唯一标识")
    bankcard_ext_info: Optional[BankcardExtInfoSchema] = Field(default=None, description="银行卡信息")

    @field_validator("identity_type")
    def validate_identity_type(cls, v: str) -> str:
        if v == "alipay":
            return "ALIPAY_ACCOUNT"
        elif v == "bank":
            return "BANK_CARD"
        elif v == "book":
            return "BOOK"
        # else:
        #     raise ValueError(f"identity_type must be one of {['alipay', 'bank', 'book']}")
        return v

    @model_validator(mode='after')
    def validate_bankcard_ext_info(self) -> 'PayeeInfoSchema':
        """验证银行卡类型时bankcard_ext_info必填"""
        if self.identity_type == "BANK_CARD" and self.bankcard_ext_info is None:
            raise ValueError("当 identity_type 是银行卡时，bankcard_ext_info 必填")
        return self


class AccountAuthorizeApplySchema(BaseModel):
    """申请转账授权签约请求"""

    enterprise_id: str = Field(description="企业ID")


class AccountAuthorizeCallbackSchema(BaseModel):
    """转账授权回调通知请求"""

    enterprise_id: str = Field(description="企业ID")
    status: str = Field(description="协议状态: NORMAL")
    sign_time: str = Field(description="签约时间")


class AccountCreateSchema(BaseModel):
    """开通资金专户请求"""

    enterprise_id: str = Field(description="企业ID")
    account_type: Optional[str] = Field(default="ALL", description="账户类型: INCOME/DISBURSE/ALL")
    scene: Optional[str] = Field(default="B2B_TRANS", description="场景类型: ENT_CREDIT/B2B_TRANS")
    account_book_id: Optional[str] = Field(default=None, description="资金专户号")
    remark: Optional[str] = Field(default=None, description="备注")


class AccountDepositSchema(BaseModel):
    """资金专户充值请求"""

    enterprise_id: str = Field(description="企业ID")
    account_book_id: str = Field(description="资金专户号")
    amount: Decimal = Field(max_digits=10, decimal_places=2, gt=0, description="充值金额")
    out_biz_no: Optional[str] = Field(default=None, description="外部订单号")
    remark: Optional[str] = Field(default=None, description="充值备注")


class AccountTransferSchema(BaseModel):
    """资金专户转账请求"""

    account_book_id: str = Field(description="付款方资金专户号")
    enterprise_id: Optional[str] = Field(default=None, description="企业ID")
    out_biz_no: Optional[str] = Field(default=None, description="商家侧订单号")
    # 转账总金额，单位为元，精确到小数点后两位
    amount: Decimal = Field(max_digits=10, decimal_places=2, gt=0, description="转账金额")
    order_title: Optional[str] = Field(default=None, description="转账标题")
    remark: Optional[str] = Field(default=None, description="转账备注")
    payee_info: PayeeInfoSchema = Field(description="收款方信息")
    employee_id: Optional[str] = Field(default=None, description="员工ID")
    priority: Optional[str] = Field(default="NORMAL", description="优先级: HIGH/NORMAL/LOW")
    ext_info: Optional[dict] = Field(default=None, description="扩展信息")

    @model_validator(mode="after")
    def validate_ext_info(self):
        if self.ext_info and not isinstance(self.ext_info, dict):
            raise ValueError("ext_info 需要是字典类型")
        return self


class TenantTransferCreate(BaseModel):
    """租户API转账请求"""

    account_book_id: str = Field(description="付款方资金专户号")
    enterprise_id: Optional[str] = Field(description="企业ID")
    # 转账总金额，单位为元，精确到小数点后两位
    amount: Decimal = Field(max_digits=10, decimal_places=2, gt=0, description="转账金额")
    order_title: Optional[str] = Field(default=None, description="转账标题")
    remark: Optional[str] = Field(default=None, description="转账备注")
    payee_info: PayeeInfoSchema = Field(description="收款方信息")


class TenantTransferResponse(BaseModel):
    """租户API转账响应"""

    status: Optional[str] = Field(description="转账状态，SUCCESS：成功（对转账到银行卡的单据, 该状态可能变为退票[REFUND]）； FAIL：失败； DEALING：处理中（转账到支付宝账户不涉及）； REFUND：退票（转账到支付宝账户不涉及）")
    order_no: Optional[str] = Field(default=None, description="支付宝转账单号")
    fund_order_id: Optional[str] = Field(default=None, description="宝支付资金流水号")


class AccountWithdrawSchema(BaseModel):
    """资金专户提现请求"""

    enterprise_id: str = Field(description="企业ID")
    account_book_id: str = Field(description="资金专户号")
    amount: Decimal = Field(max_digits=10, decimal_places=2, gt=0, description="提现金额")
    out_biz_no: str = Field(description="商家侧订单号")


class ReceiptApplySchema(BaseModel):
    """申请资金回单请求"""

    enterprise_id: str = Field(description="企业ID")
    order_no: str = Field(description="支付宝转账单号")


class AccountQuerySchema(BaseModel):
    """资金专户查询请求"""

    enterprise_id: str = Field(description="企业ID")


class AccountOutSchema(BaseModel):
    """资金专户响应模型"""

    model_config = ConfigDict(from_attributes=True)

    id: int = Field(description="主键ID")
    enterprise_id: Optional[str] = Field(default=None, description="所属企业ID")
    account_book_id: Optional[str] = Field(default=None, description="资金专户号")
    account_name: Optional[str] = Field(default=None, description="专户名称")
    scene: str = Field(description="场景类型")
    sign_status: str = Field(description="授权状态")
    sign_time: Optional[datetime] = Field(default=None, description="授权时间")
    status: str = Field(description="专户状态")
    created_time: datetime = Field(description="创建时间")
    updated_time: datetime = Field(description="更新时间")


class AccountAuthorizeApplyOutSchema(BaseModel):
    """申请转账授权响应"""

    sign_url: Optional[str] = Field(description="签约链接")


class AccountDepositOutSchema(BaseModel):
    """充值响应"""

    url: Optional[str] = Field(description="充值确认页面URL")


class AccountTransferOutSchema(BaseModel):
    """转账响应"""

    status: Optional[str] = Field(description="转账状态，SUCCESS：成功（对转账到银行卡的单据, 该状态可能变为退票[REFUND]）； FAIL：失败； DEALING：处理中（转账到支付宝账户不涉及）； REFUND：退票（转账到支付宝账户不涉及）")
    order_no: Optional[str] = Field(default=None, description="支付宝转账单号")
    fund_order_id: Optional[str] = Field(default=None, description="宝支付资金流水号")
    out_biz_no: Optional[str] = Field(default=None, description="平台流水号")

class AccountTransferBatchSchema(BaseModel):
    """批量资金专户转账请求"""

    transfers: List[AccountTransferSchema] = Field(
        description="转账列表"
    )
    callback_url: Optional[str] = Field(
        default=None,
        description="转账结果回调URL"
    )


class AccountTransferBatchOutSchema(BaseModel):
    """批量转账响应"""

    batch_id: str = Field(description="批次ID")
    status: str = Field(description="状态: PENDING/PROCESSING/COMPLETED")
    total: int = Field(description="总笔数")
    processed: int = Field(description="已处理笔数")
    success: int = Field(description="成功笔数")
    failed: int = Field(description="失败笔数")


class TransferResultSchema(BaseModel):
    """单笔转账结果"""

    out_biz_no: str = Field(description="商家侧订单号")
    status: str = Field(description="状态: SUCCESS/FAIL")
    order_no: Optional[str] = Field(default=None, description="支付宝转账单号")
    error_msg: Optional[str] = Field(default=None, description="错误信息")


class TransferTaskSchema(BaseModel):
    """转账任务"""

    task_id: str = Field(description="任务ID")
    enterprise_id: str = Field(description="企业ID")
    account_book_id: str = Field(description="资金专户号")
    out_biz_no: str = Field(description="商家侧订单号")
    amount: Decimal = Field(max_digits=10, decimal_places=2, gt=0, description="转账金额")
    order_title: str = Field(description="转账标题")
    payee_info: PayeeInfoSchema = Field(description="收款方信息")
    priority: str = Field(default="NORMAL", description="优先级: HIGH/NORMAL/LOW")
    status: str = Field(default="PENDING", description="状态: PENDING/PROCESSING/SUCCESS/FAILED/RETRYING")
    retries: int = Field(default=0, description="重试次数")
    created_at: str = Field(description="创建时间")


class TransferTaskOutSchema(BaseModel):
    """转账任务响应"""

    task_id: str = Field(description="任务ID")
    status: str = Field(description="任务状态")
    enterprise_id: str = Field(description="企业ID")
    out_biz_no: str = Field(description="商家侧订单号")
    amount: Decimal = Field(max_digits=10, decimal_places=2, gt=0, description="转账金额")
    priority: str = Field(description="优先级")
    created_at: str = Field(description="创建时间")


class TransferQueueStatusSchema(BaseModel):
    """队列状态"""

    enterprise_id: str = Field(description="企业ID")
    pending_count: int = Field(description="待处理任务数")
    processing_count: int = Field(description="处理中任务数")
    completed_count: int = Field(description="已完成任务数")
    high_priority_count: int = Field(description="高优先级任务数")
    normal_priority_count: int = Field(description="普通优先级任务数")
    low_priority_count: int = Field(description="低优先级任务数")



class TransferOutSchema(BaseModel):
    """转账记录响应"""

    model_config = ConfigDict(from_attributes=True)

    # id: int = Field(description="主键ID")
    enterprise_id: Optional[str] = Field(default=None, description="所属企业ID")
    out_biz_no: Optional[str] = Field(default=None, description="商家侧订单号")
    account_book_id: Optional[str] = Field(default=None, description="付款方资金专户号")
    amount: Decimal = Field(max_digits=10, decimal_places=2, gt=0, description="转账金额")
    order_title: Optional[str] = Field(default=None, description="转账标题")
    payee_info: Optional[PayeeInfoSchema] = Field(default=None, description="收款方信息")
    status: str = Field(description="转账状态")
    order_no: Optional[str] = Field(default=None, description="支付宝转账单号")
    error_code: Optional[str] = Field(default=None, description="错误码")
    error_msg: Optional[str] = Field(default=None, description="错误信息")
    # ext_info: Optional[dict] = Field(default=None, description="扩展信息")
    created_time: datetime = Field(description="创建时间")

    third_biz_no: Optional[str] = Field(default=None, description="三方侧订单号")

    @field_validator("payee_info", mode="after")
    def normal_payee_info(cls, v: Optional["PayeeInfoSchema"]) -> Optional["PayeeInfoSchema"]:
        if v is None:
            return v
        # 脱敏处理
        if v.identity_type == "BANK_CARD" and v.identity:
            # 银行卡号：保留前4位和后4位
            if len(v.identity) >= 8:
                v.identity = v.identity[:4] + "*" * (len(v.identity) - 8) + v.identity[-4:]
        elif v.identity_type == "ALIPAY_ACCOUNT" and v.identity:
            # 支付宝账号脱敏
            if "@" in v.identity:
                # 邮箱格式：保留@前第一个字符和域名
                parts = v.identity.split("@")
                if len(parts[0]) > 1:
                    v.identity = parts[0][0] + "*" * 3 + "@" + parts[1]
            else:
                # 手机号格式：保留前3位和后4位
                if len(v.identity) >= 7:
                    v.identity = v.identity[:3] + "*" * (len(v.identity) - 7) + v.identity[-4:]
        # 姓名脱敏：只保留第一个字
        if v.name and len(v.name) > 1:
            v.name = v.name[0] + "*" * (len(v.name) - 1)
        # 银行卡扩展信息脱敏
        if v.bankcard_ext_info:
            if v.bankcard_ext_info.inst_name:
                v.bankcard_ext_info.inst_name = "***"
        return v

    @field_serializer('amount')
    def serialize_amount(self, value: Decimal) -> str:
        """将 Decimal 序列化为保留两位小数的字符串"""
        return f"{value:.2f}"


class TransferListOutSchema(BaseModel):
    """转账记录列表响应"""

    model_config = ConfigDict(from_attributes=True)

    id: int = Field(description="主键ID")
    out_biz_no: Optional[str] = Field(default=None, description="商家侧订单号")
    account_book_id: Optional[str] = Field(default=None, description="付款方资金专户号")
    amount: Optional[float] = Field(default=None, description="转账金额")
    order_title: Optional[str] = Field(default=None, description="转账标题")
    status: str = Field(description="转账状态")
    order_no: Optional[str] = Field(default=None, description="支付宝转账单号")
    fund_order_id: Optional[str] = Field(default=None, description="宝支付资金流水号")
    payee_info: Optional[PayeeInfoSchema] = Field(default=None, description="收款方信息")
    created_time: datetime = Field(description="创建时间")

    @field_validator("payee_info", mode="after")
    def normal_payee_info(cls, v: Optional["PayeeInfoSchema"]) -> Optional["PayeeInfoSchema"]:
        if v is None:
            return v
        # 脱敏处理
        if v.identity_type == "BANK_CARD" and v.identity:
            # 银行卡号：保留前4位和后4位
            if len(v.identity) >= 8:
                v.identity = v.identity[:4] + "*" * (len(v.identity) - 8) + v.identity[-4:]
        elif v.identity_type == "ALIPAY_ACCOUNT" and v.identity:
            # 支付宝账号脱敏
            if "@" in v.identity:
                # 邮箱格式：保留@前第一个字符和域名
                parts = v.identity.split("@")
                if len(parts[0]) > 1:
                    v.identity = parts[0][0] + "*" * 3 + "@" + parts[1]
            else:
                # 手机号格式：保留前3位和后4位
                if len(v.identity) >= 7:
                    v.identity = v.identity[:3] + "*" * (len(v.identity) - 7) + v.identity[-4:]
        # 姓名脱敏：只保留第一个字
        if v.name and len(v.name) > 1:
            v.name = v.name[0] + "*" * (len(v.name) - 1)
        # 银行卡扩展信息脱敏
        if v.bankcard_ext_info:
            if v.bankcard_ext_info.inst_name:
                v.bankcard_ext_info.inst_name = "***"
        return v
    
    @field_serializer('amount')
    def serialize_amount(self, value: Decimal) -> str:
        """将 Decimal 序列化为保留两位小数的字符串"""
        return f"{value:.2f}"


class DepositOutSchema(BaseModel):
    """充值记录响应"""

    model_config = ConfigDict(from_attributes=True)

    id: int = Field(description="主键ID")
    enterprise_id: Optional[str] = Field(default=None, description="所属企业ID")
    out_biz_no: Optional[str] = Field(default=None, description="商家侧订单号")
    account_book_id: Optional[str] = Field(default=None, description="资金专户号")
    amount: Optional[float] = Field(default=None, description="充值金额")
    deposit_url: Optional[str] = Field(default=None, description="充值确认页面URL")
    status: str = Field(description="充值状态")
    created_time: datetime = Field(description="创建时间")


class WithdrawOutSchema(BaseModel):
    """提现记录响应"""

    model_config = ConfigDict(from_attributes=True)

    id: int = Field(description="主键ID")
    enterprise_id: Optional[str] = Field(default=None, description="所属企业ID")
    out_biz_no: Optional[str] = Field(default=None, description="商家侧订单号")
    account_book_id: Optional[str] = Field(default=None, description="资金专户号")
    amount: Optional[Decimal] = Field(default=None, description="提现金额")
    status: str = Field(description="提现状态")
    order_no: Optional[str] = Field(default=None, description="支付宝提现单号")
    error_code: Optional[str] = Field(default=None, description="错误码")
    error_msg: Optional[str] = Field(default=None, description="错误信息")
    created_time: datetime = Field(description="创建时间")


class ReceiptApplyOutSchema(BaseModel):
    """申请回单响应"""

    model_config = ConfigDict(from_attributes=True)

    file_id: str = Field(description="文件申请号")


class ReceiptQueryOutSchema(BaseModel):
    """查询回单响应"""

    model_config = ConfigDict(from_attributes=True)

    file_id: str = Field(description="文件申请号")
    status: str = Field(description="处理状态: INIT/PROCESS/SUCCESS/FAIL")
    download_url: Optional[str] = Field(default=None, description="下载链接")
    error_message: Optional[str] = Field(default=None, description="错误信息")


class AccountOperationOutSchema(BaseModel):
    """资金专户操作响应"""

    model_config = ConfigDict(from_attributes=True)

    enterprise_id: Optional[str] = Field(default=None, description="企业ID")
    account_book_id: Optional[str] = Field(default=None, description="资金专户号")


class ConsumeDetailQuerySchema(BaseModel):
    """账单详情查询请求"""

    pay_no: str = Field(..., description="支付宝账单号")
    enterprise_id: Optional[str] = Field(default=None, description="企业ID（2.0接口签约企业必填）")
    ant_shop_id: Optional[str] = Field(default=None, description="蚂蚁门店ID（商户服务商必填）")
    query_options: Optional[List[str]] = Field(
        default=None,
        description="查询选项: Refund(关联退款账单)/Order(关联订单)/Ticket(关联票据)"
    )


class EcConsumeInfoSchema(BaseModel):
    """账单信息子属性"""

    model_config = ConfigDict(from_attributes=True)

    account_id: str = Field(description="共同账户ID")
    pay_no: str = Field(description="交易流水号")
    consume_type: str = Field(description="账单类型: CONSUME(消费账单)/REFUND(退款账单)/TRANSFER(转账账单)")
    gmt_biz_create: str = Field(description="账单创建时间，格式：yyyy-MM-dd HH:mm:ss")
    consume_biz_type: str = Field(description="账单业务类型: EC_PAY(因公支付)/EC_CLLCT(因公收款)")
    consume_amount: str = Field(description="账单金额，单位：元")
    order_complete_label: str = Field(description="订单完结标识: 0(未完结)/1(已完结)")
    refund_status: str = Field(description="退款状态: INIT(未退款)/REFUND_PART(部分退款)/REFUND_FULL(全额退款)")
    refund_amount: str = Field(description="退款金额，单位：元")
    peer_payer_card_name: str = Field(description="实际出资人名称")
    user_id: Optional[str] = Field(default=None, description="员工支付宝UID")
    open_id: Optional[str] = Field(default=None, description="员工open_id")
    enterprise_id: str = Field(description="企业ID")
    employee_id: str = Field(description="员工ID")
    enterprise_name: Optional[str] = Field(default=None, description="企业名称")
    employee_name: Optional[str] = Field(default=None, description="员工名称")
    consume_scene_code: Optional[str] = Field(default=None, description="消费场景编码")
    consume_type_sub_category: Optional[str] = Field(default=None, description="消费类型子类")
    consume_title: Optional[str] = Field(default=None, description="账单标题")
    gmt_pay: Optional[str] = Field(default=None, description="账单支付时间")
    gmt_refund: Optional[str] = Field(default=None, description="退款时间")
    pay_amount: Optional[str] = Field(default=None, description="用户实际支付金额")
    invoice_amount: Optional[str] = Field(default=None, description="用户实际承担金额")
    peer_pay_amount: Optional[str] = Field(default=None, description="企业代付金额")
    subsidy_amount: Optional[str] = Field(default=None, description="补贴金额")
    ext_infos: Optional[dict] = Field(default=None, description="扩展信息")


class ConsumeDetailOutSchema(BaseModel):
    """账单详情查询响应"""

    model_config = ConfigDict(from_attributes=True)

    consume_info: EcConsumeInfoSchema = Field(description="账单信息")