| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959 |
- from datetime import datetime
- from decimal import Decimal
- from typing import Any, Optional
- from redis.asyncio import Redis
- from app.api.v1.module_system.auth.schema import AuthSchema
- from app.core.alipay import AlipayClient
- from app.core.exceptions import CustomException
- from app.core.logger import log
- from app.utils.snowflake import get_snowflake_id_str
- from app.plugin.module_payment.enterprise.crud import EnterpriseCRUD
- from .crud import AccountCRUD, TransferCRUD, DepositCRUD, WithdrawCRUD
- from .enums import (
- DepositStatusEnum,
- WithdrawStatusEnum,
- )
- from .schema import (
- AccountAuthorizeApplySchema,
- AccountAuthorizeApplyOutSchema,
- AccountCreateSchema,
- AccountDepositSchema,
- AccountDepositOutSchema,
- AccountOperationOutSchema,
- AccountQuerySchema,
- AccountTransferSchema,
- AccountTransferOutSchema,
- AccountWithdrawSchema,
- ReceiptApplySchema,
- TransferListOutSchema,
- TransferOutSchema,
- TenantTransferCreate,
- TenantTransferResponse,
- )
- from ..openapi.crud import OpenTransferCRUD
- class AccountService:
- """资金专户服务层"""
- @classmethod
- async def stat_transfer_amount_service(
- cls,
- auth: AuthSchema,
- enterprise_id: str,
- start_date: Optional[datetime] = None,
- end_date: Optional[datetime] = None,
- ) -> Decimal:
- """
- 统计转账金额(✅)
- 统计企业在指定时间范围内的转账总金额以及每天的转账金额。
- """
- crud = TransferCRUD(auth)
-
- return await crud.get_transfer_amount(
- enterprise_id=enterprise_id,
- start_date=start_date,
- end_date=end_date,
- )
- @classmethod
- async def authorize_apply_service(
- cls,
- auth: AuthSchema,
- data: AccountAuthorizeApplySchema
- ) -> AccountAuthorizeApplyOutSchema:
- """
- 申请转账授权签约(✅)
- 调用: alipay.commerce.ec.trans.authorize.apply
- """
- from alipay.aop.api.request.AlipayCommerceEcTransAuthorizeApplyRequest import (
- AlipayCommerceEcTransAuthorizeApplyRequest,
- )
- from alipay.aop.api.domain.AlipayCommerceEcTransAuthorizeApplyModel import (
- AlipayCommerceEcTransAuthorizeApplyModel,
- )
- from alipay.aop.api.response.AlipayCommerceEcTransAuthorizeApplyResponse import (
- AlipayCommerceEcTransAuthorizeApplyResponse,
- )
- model = AlipayCommerceEcTransAuthorizeApplyModel()
- model.enterprise_id = data.enterprise_id
- request = AlipayCommerceEcTransAuthorizeApplyRequest()
- request.biz_model = model
- client = AlipayClient.get_client()
- response = client.execute(request)
- if not response:
- raise CustomException(msg="申请转账授权失败: 无响应")
- result = AlipayCommerceEcTransAuthorizeApplyResponse()
- result.parse_response_content(response)
- if not result.is_success():
- log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
- raise CustomException(msg=f"申请转账授权失败: {result.msg}")
- return AccountAuthorizeApplyOutSchema(
- sign_url=result.sign_url,
- )
- @classmethod
- async def create_account_service(
- cls,
- auth: AuthSchema,
- data: AccountCreateSchema
- ) -> AccountOperationOutSchema:
- """
- 开通资金专户(✅)
- 调用: alipay.commerce.ec.trans.account.create
- """
- from alipay.aop.api.request.AlipayCommerceEcTransAccountCreateRequest import (
- AlipayCommerceEcTransAccountCreateRequest,
- )
- from alipay.aop.api.domain.AlipayCommerceEcTransAccountCreateModel import (
- AlipayCommerceEcTransAccountCreateModel,
- )
- from alipay.aop.api.response.AlipayCommerceEcTransAccountCreateResponse import (
- AlipayCommerceEcTransAccountCreateResponse,
- )
- model = AlipayCommerceEcTransAccountCreateModel()
- model.enterprise_id = data.enterprise_id
- # model.account_type = data.account_type or "ALL" # 收支全能户
- # model.scene = data.scene or "B2B_TRANS" # ToB转账场景
- model.account_type = "ALL"
- model.scene = "B2B_TRANS"
- request = AlipayCommerceEcTransAccountCreateRequest()
- request.biz_model = model
- client = AlipayClient.get_client()
- response = client.execute(request)
- if not response:
- raise CustomException(msg="开通资金专户失败: 无响应")
- result = AlipayCommerceEcTransAccountCreateResponse()
- result.parse_response_content(response)
- if not result.is_success():
- log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
- raise CustomException(msg=f"开通资金专户失败: {result.msg}")
- account_data = AccountCreateSchema(
- enterprise_id=model.enterprise_id,
- account_book_id=result.account_book_id,
- account_type=model.account_type,
- scene=model.scene,
- )
- if result.account_book_id:
- account_data.account_book_id = result.account_book_id
- await AccountCRUD(auth).create(account_data)
- return AccountOperationOutSchema(
- enterprise_id=account_data.enterprise_id,
- account_book_id=account_data.account_book_id,
- )
- @classmethod
- async def deposit_service(
- cls,
- auth: AuthSchema,
- data: AccountDepositSchema
- ) -> AccountDepositOutSchema:
- """
- 资金专户充值(✅)
- 调用: alipay.commerce.ec.trans.account.deposit
- """
- from alipay.aop.api.request.AlipayCommerceEcTransAccountDepositRequest import (
- AlipayCommerceEcTransAccountDepositRequest,
- )
- from alipay.aop.api.domain.AlipayCommerceEcTransAccountDepositModel import (
- AlipayCommerceEcTransAccountDepositModel,
- )
- from alipay.aop.api.response.AlipayCommerceEcTransAccountDepositResponse import (
- AlipayCommerceEcTransAccountDepositResponse,
- )
- model = AlipayCommerceEcTransAccountDepositModel()
- model.enterprise_id = data.enterprise_id
- model.account_book_id = data.account_book_id
- model.amount = str(data.amount)
- model.out_biz_no = get_snowflake_id_str(auth.tenant_id)
- request = AlipayCommerceEcTransAccountDepositRequest()
- request.biz_model = model
- client = AlipayClient.get_client()
- response = client.execute(request)
- if not response:
- raise CustomException(msg="充值失败: 无响应")
- result = AlipayCommerceEcTransAccountDepositResponse()
- result.parse_response_content(response)
- if not result.is_success():
- log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
- raise CustomException(msg=f"充值失败: {result.msg}")
- deposit_crud = DepositCRUD(auth)
- deposit_data = {
- "enterprise_id": data.enterprise_id,
- "out_biz_no": model.out_biz_no,
- "account_book_id": data.account_book_id,
- "amount": data.amount,
- "url": result.url,
- "status": DepositStatusEnum.DEALING.value,
- "remark": data.remark,
- }
- await deposit_crud.create(deposit_data)
- return AccountDepositOutSchema(
- url=result.url,
- )
- @classmethod
- async def transfer_service(
- cls,
- auth: AuthSchema,
- data: AccountTransferSchema
- ) -> AccountTransferOutSchema:
- """
- 资金专户转账(✅)
- 调用: alipay.commerce.ec.trans.account.transfer
- """
- from alipay.aop.api.request.AlipayCommerceEcTransAccountTransferRequest import (
- AlipayCommerceEcTransAccountTransferRequest,
- )
- from alipay.aop.api.domain.AlipayCommerceEcTransAccountTransferModel import (
- AlipayCommerceEcTransAccountTransferModel,
- )
- from alipay.aop.api.response.AlipayCommerceEcTransAccountTransferResponse import (
- AlipayCommerceEcTransAccountTransferResponse,
- )
- from alipay.aop.api.domain.TransParticipant import (
- TransParticipant,
- )
- from alipay.aop.api.domain.BankCardExtInfoDTO import (
- BankCardExtInfoDTO,
- )
-
- # 检查资金专户是否存在
- account = await AccountCRUD(auth).get_by_account_book_id(data.account_book_id)
- if not account:
- raise CustomException(msg="资金账户不存在")
- if account.tenant_id != auth.tenant_id:
- raise CustomException(msg="无权限操作")
- if data.enterprise_id and account.enterprise_id != data.enterprise_id:
- raise CustomException(msg="参数错误")
-
- if not data.order_title and account.enterprise_id:
- enterprise = await EnterpriseCRUD(auth).get_by_enterprise_id(account.enterprise_id)
- if not enterprise:
- raise CustomException(msg="资金账户所属企业不存在")
- data.order_title = f"来自{enterprise.name}转账"
- model = AlipayCommerceEcTransAccountTransferModel()
- model.enterprise_id = account.enterprise_id
- model.account_book_id = account.account_book_id
- model.out_biz_no = get_snowflake_id_str(auth.tenant_id)
- # 转账总金额,单位为元,精确到小数点后两位
- model.amount = str(data.amount)
- model.order_title = data.order_title
- payee_info = TransParticipant()
- payee_info.identity_type = data.payee_info.identity_type
- payee_info.name = data.payee_info.name
- payee_info.identity = data.payee_info.identity
- if data.payee_info.bankcard_ext_info:
- payee_info.bankcard_ext_info = BankCardExtInfoDTO.from_alipay_dict(
- data.payee_info.bankcard_ext_info.model_dump(exclude_none=True)
- )
- model.payee_info = payee_info
- request = AlipayCommerceEcTransAccountTransferRequest()
- request.biz_model = model
- client = AlipayClient.get_client()
- response = client.execute(request)
- if not response:
- raise CustomException(msg="转账失败: 无响应")
- result = AlipayCommerceEcTransAccountTransferResponse()
- result.parse_response_content(response)
- if not result.is_success():
- log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
- raise CustomException(msg=f"转账失败: {result.sub_msg or result.msg or result.code}")
- transfer_crud = TransferCRUD(auth)
- transfer_data = {
- "enterprise_id": model.enterprise_id,
- "out_biz_no": model.out_biz_no,
- "account_book_id": model.account_book_id,
- "amount": model.amount,
- "order_title": model.order_title,
- "payee_info": data.payee_info.model_dump() if data.payee_info else None,
- "status": result.status,
- "order_no": result.order_no,
- "fund_order_id": result.fund_order_id,
- }
- await transfer_crud.create(transfer_data)
- return AccountTransferOutSchema(
- status=result.status,
- order_no=result.order_no,
- fund_order_id=result.fund_order_id,
- out_biz_no=model.out_biz_no,
- )
- @classmethod
- async def tenant_transfer_service(
- cls,
- auth: AuthSchema,
- tenant_id: int,
- data: TenantTransferCreate,
- request_ip: str,
- api_key_id: int | None = None,
- ) -> TenantTransferResponse:
- """
- 租户API转账(通过API Key认证)
- 调用: alipay.commerce.ec.trans.account.transfer
- """
- from alipay.aop.api.request.AlipayCommerceEcTransAccountTransferRequest import (
- AlipayCommerceEcTransAccountTransferRequest,
- )
- from alipay.aop.api.domain.AlipayCommerceEcTransAccountTransferModel import (
- AlipayCommerceEcTransAccountTransferModel,
- )
- from alipay.aop.api.response.AlipayCommerceEcTransAccountTransferResponse import (
- AlipayCommerceEcTransAccountTransferResponse,
- )
- from alipay.aop.api.domain.TransParticipant import (
- TransParticipant,
- )
- from alipay.aop.api.domain.BankCardExtInfoDTO import (
- BankCardExtInfoDTO,
- )
-
- # 检查资金专户是否存在
- account = await AccountCRUD(auth).get_by_account_book_id(data.account_book_id)
- if not account:
- raise CustomException(msg="资金账户不存在")
- if account.tenant_id != tenant_id:
- raise CustomException(msg="无权限操作")
- if data.enterprise_id and account.enterprise_id != data.enterprise_id:
- raise CustomException(msg="参数错误")
-
- if not data.order_title and account.enterprise_id:
- enterprise = await EnterpriseCRUD(auth).get_by_enterprise_id(account.enterprise_id)
- if not enterprise:
- raise CustomException(msg="资金账户所属企业不存在")
- data.order_title = f"来自{enterprise.name}转账"
- model = AlipayCommerceEcTransAccountTransferModel()
- model.enterprise_id = account.enterprise_id
- model.account_book_id = account.account_book_id
- model.out_biz_no = get_snowflake_id_str(tenant_id)
- # 转账总金额,单位为元,精确到小数点后两位
- model.amount = str(data.amount)
- model.order_title = data.order_title
- payee_info = TransParticipant()
- payee_info.identity_type = data.payee_info.identity_type
- payee_info.name = data.payee_info.name
- payee_info.identity = data.payee_info.identity
- if data.payee_info.bankcard_ext_info:
- payee_info.bankcard_ext_info = BankCardExtInfoDTO.from_alipay_dict(
- data.payee_info.bankcard_ext_info.model_dump(exclude_none=True)
- )
- model.payee_info = payee_info
- request = AlipayCommerceEcTransAccountTransferRequest()
- request.biz_model = model
- client = AlipayClient.get_client()
- response = client.execute(request)
- if not response:
- raise CustomException(msg="转账失败: 无响应")
- result = AlipayCommerceEcTransAccountTransferResponse()
- result.parse_response_content(response)
- if not result.is_success():
- log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
- raise CustomException(msg=f"转账失败: {result.sub_msg or result.msg or result.code}")
- transfer_crud = TransferCRUD(auth)
- transfer_data = {
- "enterprise_id": model.enterprise_id,
- "out_biz_no": model.out_biz_no,
- "account_book_id": model.account_book_id,
- "amount": model.amount,
- "order_title": model.order_title,
- "payee_info": data.payee_info.model_dump() if data.payee_info else None,
- "status": result.status,
- "order_no": result.order_no,
- "fund_order_id": result.fund_order_id,
- }
- await transfer_crud.create(transfer_data)
- return TenantTransferResponse(
- status=result.status,
- order_no=result.order_no,
- fund_order_id=result.fund_order_id,
- )
- @classmethod
- async def withdraw_service(
- cls,
- auth: AuthSchema,
- data: AccountWithdrawSchema
- ) -> AccountOperationOutSchema:
- """
- 资金专户提现
- 调用: alipay.commerce.ec.trans.account.withdraw
- 接口文档: https://opendocs.alipay.com/pre-open/d651859b_alipay.commerce.ec.trans.account.withdraw
- 参数说明:
- - enterprise_id: 企业ID
- - account_book_id: 资金专户号
- - amount: 提现金额
- - out_biz_no: 商家侧订单号(唯一)
- """
- from alipay.aop.api.request.AlipayCommerceEcTransAccountWithdrawRequest import (
- AlipayCommerceEcTransAccountWithdrawRequest,
- )
- from alipay.aop.api.domain.AlipayCommerceEcTransAccountWithdrawModel import (
- AlipayCommerceEcTransAccountWithdrawModel,
- )
- from alipay.aop.api.response.AlipayCommerceEcTransAccountWithdrawResponse import (
- AlipayCommerceEcTransAccountWithdrawResponse,
- )
- crud = AccountCRUD(auth)
- enterprise = await crud.get_by_enterprise_id(data.enterprise_id)
- if not enterprise:
- raise CustomException(msg="企业不存在")
- model = AlipayCommerceEcTransAccountWithdrawModel()
- model.enterprise_id = enterprise.enterprise_id
- model.account_book_id = data.account_book_id
- model.amount = str(data.amount)
- model.out_biz_no = get_snowflake_id_str(auth.tenant_id)
- request = AlipayCommerceEcTransAccountWithdrawRequest()
- request.biz_model = model
- client = AlipayClient.get_client()
- response = client.execute(request)
- if not response:
- raise CustomException(msg="提现失败: 无响应")
- result = AlipayCommerceEcTransAccountWithdrawResponse()
- result.parse_response_content(response)
- if not result.is_success():
- log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
- raise CustomException(msg=f"提现失败: {result.msg}")
- withdraw_crud = WithdrawCRUD(auth)
- withdraw_data = {
- "enterprise_id": data.enterprise_id,
- "out_biz_no": model.out_biz_no,
- "account_book_id": data.account_book_id,
- "amount": data.amount,
- # 专户提现到余额户是同步操作,要么执行成功,要么执行异常,
- # 出参status设计多余,遵循规范使用业务码区分成功与失败
- "status": WithdrawStatusEnum.SUCCESS.value,
- "order_no": result.order_no,
- }
- await withdraw_crud.create(withdraw_data)
- log.info(f"资金专户提现发起成功: 企业: {data.enterprise_id}, 金额: {data.amount}")
- return AccountOperationOutSchema(
- enterprise_id=data.enterprise_id,
- account_book_id=data.account_book_id,
- )
- @classmethod
- async def query_account_service(
- cls,
- auth: AuthSchema,
- data: AccountQuerySchema
- ) -> list[Any]:
- """
- 查询资金专户(调用支付宝接口)
- 调用: alipay.commerce.ec.trans.account.query
- """
- from alipay.aop.api.request.AlipayCommerceEcTransAccountQueryRequest import (
- AlipayCommerceEcTransAccountQueryRequest,
- )
- from alipay.aop.api.domain.AlipayCommerceEcTransAccountQueryModel import (
- AlipayCommerceEcTransAccountQueryModel,
- )
- from alipay.aop.api.response.AlipayCommerceEcTransAccountQueryResponse import (
- AlipayCommerceEcTransAccountQueryResponse,
- )
- from alipay.aop.api.domain.FundAccountApiDTO import (
- FundAccountApiDTO,
- )
- model = AlipayCommerceEcTransAccountQueryModel()
- model.enterprise_id = data.enterprise_id
- request = AlipayCommerceEcTransAccountQueryRequest()
- request.biz_model = model
- client = AlipayClient.get_client()
- response = client.execute(request)
- if not response:
- raise CustomException(msg="查询资金专户失败: 无响应")
- result = AlipayCommerceEcTransAccountQueryResponse()
- result.parse_response_content(response)
- if not result.is_success():
- log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
- raise CustomException(msg=f"查询资金专户失败: {result.msg}")
- collect = []
- for v in list(result.account_list or []):
- if not hasattr(v, "account_book_id"):
- continue
- if not hasattr(v, "scene") or v.scene != "B2B_TRANS":
- continue
- account = FundAccountApiDTO.to_alipay_dict(v)
- collect.append(account)
-
- return collect
- @classmethod
- async def transfer_detail_service(
- cls,
- auth: AuthSchema,
- out_biz_no: str
- ) -> TransferOutSchema:
- """
- 查询转账记录详情
- """
- crud = TransferCRUD(auth)
- transfer = await crud.get_by_out_biz_no(out_biz_no)
- if not transfer:
- raise CustomException(msg="转账记录不存在")
- transfer_result = TransferOutSchema.model_validate(transfer)
- # 查询三方订单号
- open_transfer_crud = OpenTransferCRUD(auth)
- open_transfer_data = await open_transfer_crud.get(out_biz_no=transfer.out_biz_no)
- if open_transfer_data:
- transfer_result.third_biz_no = open_transfer_data.third_biz_no
- return transfer_result
- @classmethod
- async def transfer_list_service(
- cls,
- auth: AuthSchema,
- page_no: int = 1,
- page_size: int = 20,
- search: dict | None = None,
- ) -> dict:
- """
- 查询转账记录列表
- """
- log.info(f"查询转账记录列表: {page_no}, {page_size}, {search}")
- crud = TransferCRUD(auth)
- offset = (page_no - 1) * page_size
- return await crud.page(
- offset=offset,
- limit=page_size,
- order_by=[{"id": "desc"}],
- search=search or {},
- out_schema=TransferListOutSchema,
- )
- @classmethod
- async def transfer_export_service(
- cls,
- auth: AuthSchema,
- start_time: str,
- end_time: str,
- enterprise_id: Optional[str] = None,
- ) -> bytes:
- """
- 导出转账记录报表为Excel文件
- """
- log.info(f"导出转账记录报表: {start_time} -> {end_time}")
-
- crud = TransferCRUD(auth)
-
- search = {
- "created_time__gte": start_time,
- "created_time__lte": end_time,
- }
-
- if enterprise_id:
- search["enterprise_id"] = enterprise_id
-
- records = await crud.list(
- search=search,
- order_by=[{"id": "desc"}],
- )
-
- from app.utils.excel_util import ExcelUtil
-
- status_map = {
- "DEALING": "处理中",
- "SUCCESS": "成功",
- "FAIL": "失败",
- "REFUND": "退票",
- }
-
- payee_type_map = {
- "ALIPAY_ACCOUNT": "支付宝账户",
- "BANK_CARD": "银行卡",
- }
-
- list_data = []
- for i, record in enumerate(records, start=1):
- payee_info = record.payee_info or {}
- list_data.append({
- "序号": i,
- "订单号": record.out_biz_no or "",
- "商户订单号": record.order_no or "",
- "金额(元)": str(record.amount or 0),
- "收款方姓名": payee_info.get("name", ""),
- "收款方类型": payee_type_map.get(payee_info.get("identity_type", ""), ""),
- "状态": status_map.get(record.status, record.status),
- "转账标题": record.order_title or "",
- "创建时间": record.created_time.strftime("%Y-%m-%d %H:%M:%S") if record.created_time else "",
- })
-
- mapping_dict = {
- "序号": "序号",
- "订单号": "订单号",
- "商户订单号": "商户订单号",
- "金额(元)": "金额(元)",
- "收款方姓名": "收款方姓名",
- "收款方类型": "收款方类型",
- "状态": "状态",
- "转账标题": "转账标题",
- "创建时间": "创建时间",
- }
-
- return ExcelUtil.export_list2excel(list_data, mapping_dict)
- @classmethod
- async def apply_receipt_service(
- cls,
- auth: AuthSchema,
- redis: Redis,
- data: ReceiptApplySchema,
- ) -> str:
- """
- 申请转账业务回单
-
- 调用: alipay.commerce.ec.trans.receipt.apply
-
- 参数:
- - enterprise_id: 企业ID
- - order_no: 支付宝转账单号
-
- 返回: file_id
- """
- from app.core.redis_crud import RedisCURD
-
- redis_crud = RedisCURD(redis)
-
- cache_key = f"receipt:{data.enterprise_id}:{data.order_no}"
- cached_file_id = await redis_crud.get(cache_key)
-
- if cached_file_id:
- log.info(f"使用缓存的 file_id: {cached_file_id}")
- return cached_file_id
-
- crud = EnterpriseCRUD(auth)
- enterprise = await crud.get_by_enterprise_id(data.enterprise_id)
- if not enterprise:
- raise CustomException(msg="企业不存在")
-
- from alipay.aop.api.request.AlipayCommerceEcTransReceiptApplyRequest import (
- AlipayCommerceEcTransReceiptApplyRequest,
- )
- from alipay.aop.api.domain.AlipayCommerceEcTransReceiptApplyModel import (
- AlipayCommerceEcTransReceiptApplyModel,
- )
- from alipay.aop.api.response.AlipayCommerceEcTransReceiptApplyResponse import (
- AlipayCommerceEcTransReceiptApplyResponse,
- )
-
- model = AlipayCommerceEcTransReceiptApplyModel()
- model.enterprise_id = data.enterprise_id
- model.order_no = data.order_no
-
- request = AlipayCommerceEcTransReceiptApplyRequest()
- request.biz_model = model
-
- client = AlipayClient.get_client()
- response = client.execute(request)
-
- if not response:
- raise CustomException(msg="申请回单失败: 无响应")
-
- result = AlipayCommerceEcTransReceiptApplyResponse()
- result.parse_response_content(response)
-
- if not result.is_success():
- # 清除缓存
- await redis_crud.delete(cache_key)
- raise CustomException(msg=f"申请回单失败: {result.msg}")
-
- file_id = str(result.file_id)
- await redis_crud.set(cache_key, file_id, expire=172800)
-
- log.info(f"申请回单成功: order_no={data.order_no}, file_id={file_id}")
- return file_id
- @classmethod
- async def query_receipt_service(cls, enterprise_id: str, file_id: str) -> dict:
- """
- 查询回单状态
-
- 调用: alipay.commerce.ec.trans.receipt.query
-
- 参数:
- - file_id: 文件申请号
-
- 返回: {file_id, status, download_url, error_message}
- """
- from alipay.aop.api.request.AlipayCommerceEcTransReceiptQueryRequest import (
- AlipayCommerceEcTransReceiptQueryRequest,
- )
- from alipay.aop.api.response.AlipayCommerceEcTransReceiptQueryResponse import (
- AlipayCommerceEcTransReceiptQueryResponse,
- )
- from alipay.aop.api.domain.AlipayCommerceEcTransReceiptQueryModel import (
- AlipayCommerceEcTransReceiptQueryModel,
- )
-
- model = AlipayCommerceEcTransReceiptQueryModel()
- model.enterprise_id = enterprise_id
- model.file_id = file_id
-
- request = AlipayCommerceEcTransReceiptQueryRequest()
- request.biz_model = model
- client = AlipayClient.get_client()
- response = client.execute(request)
-
- if not response:
- raise CustomException(msg="查询回单失败: 无响应")
-
- result = AlipayCommerceEcTransReceiptQueryResponse()
- result.parse_response_content(response)
-
- if not result.is_success():
- raise CustomException(msg=f"查询回单失败: {result.msg}")
-
- data = {
- "file_id": file_id,
- "status": result.status,
- "download_url": result.download_url,
- "error_message": result.error_message,
- }
-
- return data
- @classmethod
- async def update_transfer_status_service(
- cls,
- auth: AuthSchema,
- order_no: str,
- status: str,
- ext_info: dict = {}
- ) -> None:
- """
- 更新转账状态(由通知处理器调用)
- """
- crud = TransferCRUD(auth)
- transfer = await crud.get_by_order_no(order_no)
- if not transfer:
- log.warning(f"转账记录不存在: {order_no}")
- return
- update_data = {}
- update_data["status"] = status
- if ext_info:
- update_data["ext_info"] = ext_info
- await crud.update_by_order_no(order_no, update_data)
- @classmethod
- async def update_deposit_status_service(
- cls,
- auth: AuthSchema,
- out_biz_no: str,
- status: str,
- ) -> None:
- """
- 更新充值状态(由通知处理器调用)
- """
- crud = DepositCRUD(auth)
- deposit = await crud.get_by_out_biz_no(out_biz_no)
- if not deposit:
- log.warning(f"充值记录不存在: {out_biz_no}")
- return
- update_data = {"status": status}
- await crud.update_by_out_biz_no(out_biz_no, update_data)
- @classmethod
- async def update_withdraw_status_service(
- cls,
- auth: AuthSchema,
- out_biz_no: str,
- status: str,
- error_code: str | None = None,
- error_msg: str | None = None,
- ) -> None:
- """
- 更新提现状态(由通知处理器调用)
- """
- crud = WithdrawCRUD(auth)
- withdraw = await crud.get_by_out_biz_no(out_biz_no)
- if not withdraw:
- log.warning(f"提现记录不存在: {out_biz_no}")
- return
- update_data = {"status": status}
- if error_code:
- update_data["error_code"] = error_code
- if error_msg:
- update_data["error_msg"] = error_msg
- await crud.update_by_out_biz_no(out_biz_no, update_data)
- @classmethod
- async def consume_detail_query_service(
- cls,
- auth: AuthSchema,
- pay_no: str,
- enterprise_id: str | None = None,
- ant_shop_id: str | None = None,
- query_options: list[str] | None = None,
- ) -> dict:
- """
- 账单详情查询(✅)
- 调用: alipay.commerce.ec.consume.detail.query
- 用于查询企业码账单详情,支持查询关联退款、订单、票据等信息。
- """
- from alipay.aop.api.request.AlipayCommerceEcConsumeDetailQueryRequest import (
- AlipayCommerceEcConsumeDetailQueryRequest,
- )
- from alipay.aop.api.domain.AlipayCommerceEcConsumeDetailQueryModel import (
- AlipayCommerceEcConsumeDetailQueryModel,
- )
- from alipay.aop.api.response.AlipayCommerceEcConsumeDetailQueryResponse import (
- AlipayCommerceEcConsumeDetailQueryResponse,
- )
- model = AlipayCommerceEcConsumeDetailQueryModel()
- model.pay_no = pay_no
- if enterprise_id:
- model.enterprise_id = enterprise_id
- if ant_shop_id:
- model.ant_shop_id = ant_shop_id
- if query_options:
- model.query_options = query_options
- request = AlipayCommerceEcConsumeDetailQueryRequest()
- request.biz_model = model
- client = AlipayClient.get_client()
- response = client.execute(request)
- if not response:
- raise CustomException(msg="账单详情查询失败: 无响应")
- result = AlipayCommerceEcConsumeDetailQueryResponse()
- result.parse_response_content(response)
- if not result.is_success():
- log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
- raise CustomException(msg=f"账单详情查询失败: {result.msg}")
- consume_info = result.consume_info
- if not consume_info:
- raise CustomException(msg="账单详情查询失败: 无账单信息")
- return {
- "account_id": consume_info.account_id,
- "pay_no": consume_info.pay_no,
- "consume_type": consume_info.consume_type,
- "gmt_biz_create": consume_info.gmt_biz_create,
- "consume_biz_type": consume_info.consume_biz_type,
- "consume_amount": consume_info.consume_amount,
- "order_complete_label": consume_info.order_complete_label,
- "refund_status": consume_info.refund_status,
- "refund_amount": consume_info.refund_amount,
- "peer_payer_card_name": consume_info.peer_payer_card_name,
- "user_id": getattr(consume_info, 'user_id', None),
- "open_id": getattr(consume_info, 'open_id', None),
- "enterprise_id": consume_info.enterprise_id,
- "employee_id": consume_info.employee_id,
- "enterprise_name": getattr(consume_info, 'enterprise_name', None),
- "employee_name": getattr(consume_info, 'employee_name', None),
- "consume_scene_code": getattr(consume_info, 'consume_scene_code', None),
- "consume_type_sub_category": getattr(consume_info, 'consume_type_sub_category', None),
- "consume_title": getattr(consume_info, 'consume_title', None),
- "gmt_pay": getattr(consume_info, 'gmt_pay', None),
- "gmt_refund": getattr(consume_info, 'gmt_refund', None),
- "pay_amount": getattr(consume_info, 'pay_amount', None),
- "invoice_amount": getattr(consume_info, 'invoice_amount', None),
- "peer_pay_amount": getattr(consume_info, 'peer_pay_amount', None),
- "subsidy_amount": getattr(consume_info, 'subsidy_amount', None),
- "ext_infos": getattr(consume_info, 'ext_infos', None),
- }
|