service.py 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777
  1. from datetime import datetime
  2. from decimal import Decimal
  3. from app.api.v1.module_system.auth.schema import AuthSchema
  4. from app.core.alipay import AlipayClient
  5. from app.core.exceptions import CustomException
  6. from app.core.logger import log
  7. from app.utils.snowflake import get_snowflake_id
  8. from .enums import QuotaStatusEnum
  9. from .schema import (
  10. ExpenseQuotaCreateSchema,
  11. ExpenseQuotaDeleteSchema,
  12. ExpenseQuotaModifySchema,
  13. ExpenseQuotaQueryOutSchema,
  14. ExpenseQuotaQuerySchema,
  15. IssueBatchCancelOutSchema,
  16. IssueBatchCancelSchema,
  17. IssueBatchCreateOutSchema,
  18. IssueBatchCreateSchema,
  19. IssueBatchListOutSchema,
  20. IssueBatchRecordsQueryOutSchema,
  21. IssueBatchRecordsQuerySchema,
  22. IssueQuotaCheckFailedItem,
  23. IssueRecordInfoItem,
  24. QuotaCreateSchema,
  25. QuotaDetailInfoSchema,
  26. QuotaListOutSchema,
  27. QuotaOperationOutSchema,
  28. QuotaOutSchema,
  29. QuotaUpdateSchema,
  30. )
  31. from .crud import IssueBatchCRUD, QuotaCRUD
  32. from .model import IssueBatchModel
  33. class QuotaService:
  34. """额度服务层"""
  35. @classmethod
  36. async def create_expense_quota_service(
  37. cls, auth: AuthSchema, data: ExpenseQuotaCreateSchema
  38. ) -> QuotaOperationOutSchema:
  39. """
  40. 创建余额/点券
  41. 调用: alipay.ebpp.invoice.expensecontrol.quota.create
  42. """
  43. crud = QuotaCRUD(auth)
  44. out_biz_no = data.outer_source_id or str(get_snowflake_id())
  45. try:
  46. from alipay.aop.api.request.AlipayEbppInvoiceExpensecontrolQuotaCreateRequest import (
  47. AlipayEbppInvoiceExpensecontrolQuotaCreateRequest,
  48. )
  49. from alipay.aop.api.domain.AlipayEbppInvoiceExpensecontrolQuotaCreateModel import (
  50. AlipayEbppInvoiceExpensecontrolQuotaCreateModel,
  51. )
  52. from alipay.aop.api.domain.IssueQuotaTarget import (
  53. IssueQuotaTarget,
  54. )
  55. from alipay.aop.api.response.AlipayEbppInvoiceExpensecontrolQuotaCreateResponse import (
  56. AlipayEbppInvoiceExpensecontrolQuotaCreateResponse,
  57. )
  58. except ImportError:
  59. raise CustomException(msg="支付宝SDK未正确安装,请检查alipay-sdk-python依赖")
  60. model = AlipayEbppInvoiceExpensecontrolQuotaCreateModel()
  61. model.target_type = data.target_type
  62. model.target_id = data.target_id
  63. model.enterprise_id = data.enterprise_id
  64. model.outer_source_id = out_biz_no
  65. model.quota_type = data.quota_type or "CAP"
  66. model.share_mode = data.share_mode or "0"
  67. if data.effective_start_date:
  68. model.effective_start_date = data.effective_start_date.strftime("%Y-%m-%d %H:%M:%S")
  69. if data.effective_end_date:
  70. model.effective_end_date = data.effective_end_date.strftime("%Y-%m-%d %H:%M:%S")
  71. if data.issue_name:
  72. model.issue_name = data.issue_name
  73. if data.issue_desc:
  74. model.issue_desc = data.issue_desc
  75. if data.issue_quota_target_list:
  76. target_list = []
  77. for item in data.issue_quota_target_list:
  78. target = IssueQuotaTarget()
  79. target.owner_type = item.owner_type
  80. target.owner_id = item.owner_id
  81. target.quota = item.quota
  82. if item.amount is not None:
  83. target.amount = item.amount
  84. target_list.append(target)
  85. model.issue_quota_target_list = target_list
  86. request = AlipayEbppInvoiceExpensecontrolQuotaCreateRequest()
  87. request.biz_model = model
  88. client = AlipayClient.get_client()
  89. response = client.execute(request)
  90. if not response:
  91. raise CustomException(msg="创建余额/点券失败: 无响应")
  92. result = AlipayEbppInvoiceExpensecontrolQuotaCreateResponse()
  93. result.parse_response_content(response)
  94. if not result.is_success():
  95. log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
  96. raise CustomException(msg=f"创建余额/点券失败: {result.msg}")
  97. quota_data = data.model_dump(exclude_none=True)
  98. quota_data["out_biz_no"] = out_biz_no
  99. quota_data["status"] = QuotaStatusEnum.QUOTA_ACTIVE.value
  100. if result.quota_id:
  101. quota_data["quota_id"] = result.quota_id
  102. quota = await crud.create(quota_data)
  103. if not quota:
  104. raise CustomException(msg="创建额度记录失败")
  105. return QuotaOperationOutSchema(out_biz_no=out_biz_no, quota_id=result.quota_id)
  106. @classmethod
  107. async def create_quota_service(
  108. cls, auth: AuthSchema, data: QuotaCreateSchema
  109. ) -> QuotaOperationOutSchema:
  110. """创建额度"""
  111. crud = QuotaCRUD(auth)
  112. out_biz_no = str(get_snowflake_id())
  113. quota_data = data.model_dump(exclude_none=True)
  114. quota_data["out_biz_no"] = out_biz_no
  115. quota_data["status"] = QuotaStatusEnum.QUOTA_ACTIVE.value
  116. if quota_data.get("available_amount") is None:
  117. quota_data["available_amount"] = quota_data.get("total_amount", 0)
  118. quota = await crud.create(quota_data)
  119. if not quota:
  120. raise CustomException(msg="创建额度记录失败")
  121. return QuotaOperationOutSchema(out_biz_no=out_biz_no, quota_id=quota.quota_id)
  122. @classmethod
  123. async def query_expense_quota_service(
  124. cls, auth: AuthSchema, data: ExpenseQuotaQuerySchema
  125. ) -> ExpenseQuotaQueryOutSchema:
  126. """
  127. 查询余额/点券
  128. 调用: alipay.ebpp.invoice.expensecontrol.quota.query
  129. """
  130. try:
  131. from alipay.aop.api.request.AlipayEbppInvoiceExpensecontrolQuotaQueryRequest import (
  132. AlipayEbppInvoiceExpensecontrolQuotaQueryRequest,
  133. )
  134. from alipay.aop.api.domain.AlipayEbppInvoiceExpensecontrolQuotaQueryModel import (
  135. AlipayEbppInvoiceExpensecontrolQuotaQueryModel,
  136. )
  137. from alipay.aop.api.response.AlipayEbppInvoiceExpensecontrolQuotaQueryResponse import (
  138. AlipayEbppInvoiceExpensecontrolQuotaQueryResponse,
  139. )
  140. except ImportError:
  141. raise CustomException(msg="支付宝SDK未正确安装,请检查alipay-sdk-python依赖")
  142. model = AlipayEbppInvoiceExpensecontrolQuotaQueryModel()
  143. model.owner_type = data.owner_type
  144. model.page_size = data.page_size
  145. model.page_num = data.page_num
  146. if data.target_type:
  147. model.target_type = data.target_type
  148. if data.target_id:
  149. model.target_id = data.target_id
  150. if data.owner_id:
  151. model.owner_id = data.owner_id
  152. if data.owner_open_id:
  153. model.owner_open_id = data.owner_open_id
  154. if data.enterprise_id:
  155. model.enterprise_id = data.enterprise_id
  156. if data.quota_id_list:
  157. model.quota_id_list = data.quota_id_list
  158. if data.quota_type:
  159. model.quota_type = data.quota_type
  160. request = AlipayEbppInvoiceExpensecontrolQuotaQueryRequest()
  161. request.biz_model = model
  162. client = AlipayClient.get_client()
  163. response = client.execute(request)
  164. if not response:
  165. raise CustomException(msg="查询余额/点券失败: 无响应")
  166. result = AlipayEbppInvoiceExpensecontrolQuotaQueryResponse()
  167. result.parse_response_content(response)
  168. if not result.is_success():
  169. log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
  170. raise CustomException(msg=f"查询余额/点券失败: {result.msg}")
  171. return ExpenseQuotaQueryOutSchema(
  172. page_num=result.page_num or data.page_num,
  173. page_size=result.page_size or data.page_size,
  174. total_page_count=result.total_page_count or 0,
  175. )
  176. @classmethod
  177. async def modify_expense_quota_service(
  178. cls, auth: AuthSchema, out_biz_no: str, data: ExpenseQuotaModifySchema
  179. ) -> QuotaOperationOutSchema:
  180. """
  181. 修改余额/点券
  182. 调用: alipay.ebpp.invoice.expensecontrol.quota.modify
  183. """
  184. crud = QuotaCRUD(auth)
  185. quota = await crud.get_by_out_biz_no(out_biz_no)
  186. if not quota:
  187. raise CustomException(msg="额度不存在")
  188. try:
  189. from alipay.aop.api.request.AlipayEbppInvoiceExpensecontrolQuotaModifyRequest import (
  190. AlipayEbppInvoiceExpensecontrolQuotaModifyRequest,
  191. )
  192. from alipay.aop.api.domain.AlipayEbppInvoiceExpensecontrolQuotaModifyModel import (
  193. AlipayEbppInvoiceExpensecontrolQuotaModifyModel,
  194. )
  195. from alipay.aop.api.response.AlipayEbppInvoiceExpensecontrolQuotaModifyResponse import (
  196. AlipayEbppInvoiceExpensecontrolQuotaModifyResponse,
  197. )
  198. except ImportError:
  199. raise CustomException(msg="支付宝SDK未正确安装,请检查alipay-sdk-python依赖")
  200. model = AlipayEbppInvoiceExpensecontrolQuotaModifyModel()
  201. model.quota_id = data.quota_id
  202. model.action = data.action
  203. model.outer_source_id = data.outer_source_id
  204. model.enterprise_id = data.enterprise_id
  205. if data.amount is not None:
  206. model.amount = str(data.amount)
  207. if data.share_mode:
  208. model.share_mode = data.share_mode
  209. request = AlipayEbppInvoiceExpensecontrolQuotaModifyRequest()
  210. request.biz_model = model
  211. client = AlipayClient.get_client()
  212. response = client.execute(request)
  213. if not response:
  214. raise CustomException(msg="修改余额/点券失败: 无响应")
  215. result = AlipayEbppInvoiceExpensecontrolQuotaModifyResponse()
  216. result.parse_response_content(response)
  217. if not result.is_success():
  218. log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
  219. raise CustomException(msg=f"修改余额/点券失败: {result.msg}")
  220. return QuotaOperationOutSchema(
  221. out_biz_no=out_biz_no,
  222. quota_id=data.quota_id,
  223. result=result.success,
  224. )
  225. @classmethod
  226. async def delete_expense_quota_service(
  227. cls, auth: AuthSchema, out_biz_no: str, data: ExpenseQuotaDeleteSchema
  228. ) -> QuotaOperationOutSchema:
  229. """
  230. 删除额度
  231. 调用: alipay.ebpp.invoice.expensecontrol.quota.delete
  232. """
  233. crud = QuotaCRUD(auth)
  234. quota = await crud.get_by_out_biz_no(out_biz_no)
  235. if not quota:
  236. raise CustomException(msg="额度不存在")
  237. try:
  238. from alipay.aop.api.request.AlipayEbppInvoiceExpensecontrolQuotaDeleteRequest import (
  239. AlipayEbppInvoiceExpensecontrolQuotaDeleteRequest,
  240. )
  241. from alipay.aop.api.domain.AlipayEbppInvoiceExpensecontrolQuotaDeleteModel import (
  242. AlipayEbppInvoiceExpensecontrolQuotaDeleteModel,
  243. )
  244. from alipay.aop.api.response.AlipayEbppInvoiceExpensecontrolQuotaDeleteResponse import (
  245. AlipayEbppInvoiceExpensecontrolQuotaDeleteResponse,
  246. )
  247. except ImportError:
  248. raise CustomException(msg="支付宝SDK未正确安装,请检查alipay-sdk-python依赖")
  249. model = AlipayEbppInvoiceExpensecontrolQuotaDeleteModel()
  250. model.enterprise_id = data.enterprise_id
  251. if data.quota_id:
  252. model.quota_id = data.quota_id
  253. if data.issue_batch_id:
  254. model.issue_batch_id = data.issue_batch_id
  255. request = AlipayEbppInvoiceExpensecontrolQuotaDeleteRequest()
  256. request.biz_model = model
  257. client = AlipayClient.get_client()
  258. response = client.execute(request)
  259. if not response:
  260. raise CustomException(msg="删除额度失败: 无响应")
  261. result = AlipayEbppInvoiceExpensecontrolQuotaDeleteResponse()
  262. result.parse_response_content(response)
  263. if not result.is_success():
  264. log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
  265. raise CustomException(msg=f"删除额度失败: {result.msg}")
  266. await crud.delete(id=quota.id)
  267. return QuotaOperationOutSchema(out_biz_no=out_biz_no)
  268. # ========================
  269. # 手工批量发放额度
  270. # ========================
  271. @classmethod
  272. async def issue_batch_create_service(
  273. cls, auth: AuthSchema, data: IssueBatchCreateSchema
  274. ) -> IssueBatchCreateOutSchema:
  275. """
  276. 手工批量发放额度
  277. 调用: alipay.ebpp.invoice.expensecontrol.issuebatch.create
  278. """
  279. try:
  280. from alipay.aop.api.request.AlipayEbppInvoiceExpensecontrolIssuebatchCreateRequest import (
  281. AlipayEbppInvoiceExpensecontrolIssuebatchCreateRequest,
  282. )
  283. from alipay.aop.api.domain.AlipayEbppInvoiceExpensecontrolIssuebatchCreateModel import (
  284. AlipayEbppInvoiceExpensecontrolIssuebatchCreateModel,
  285. )
  286. from alipay.aop.api.domain.IssueTargetInfoContent import (
  287. IssueTargetInfoContent,
  288. )
  289. from alipay.aop.api.response.AlipayEbppInvoiceExpensecontrolIssuebatchCreateResponse import (
  290. AlipayEbppInvoiceExpensecontrolIssuebatchCreateResponse,
  291. )
  292. except ImportError:
  293. raise CustomException(msg="支付宝SDK未正确安装,请检查alipay-sdk-python依赖")
  294. model = AlipayEbppInvoiceExpensecontrolIssuebatchCreateModel()
  295. model.enterprise_id = data.enterprise_id
  296. model.issue_name = data.issue_name
  297. model.quota_type = data.quota_type
  298. model.effective_start_date = data.effective_start_date
  299. model.effective_end_date = data.effective_end_date
  300. model.institution_id = data.institution_id
  301. model.batch_no = data.batch_no
  302. model.share_mode = data.share_mode
  303. if data.issue_desc:
  304. model.issue_desc = data.issue_desc
  305. if data.issue_target_info_list:
  306. target_list = []
  307. for item in data.issue_target_info_list:
  308. target = IssueTargetInfoContent()
  309. target.issue_quota = item.issue_quota
  310. if item.owner_open_id:
  311. target.owner_open_id = item.owner_open_id
  312. if item.owner_id:
  313. target.owner_id = item.owner_id
  314. if item.user_name:
  315. target.user_name = item.user_name
  316. if item.owner_type:
  317. target.owner_type = item.owner_type
  318. target_list.append(target)
  319. model.issue_target_info_list = target_list
  320. request = AlipayEbppInvoiceExpensecontrolIssuebatchCreateRequest()
  321. request.biz_model = model
  322. client = AlipayClient.get_client()
  323. response = client.execute(request)
  324. if not response:
  325. raise CustomException(msg="手工批量发放额度失败: 无响应")
  326. result = AlipayEbppInvoiceExpensecontrolIssuebatchCreateResponse()
  327. result.parse_response_content(response)
  328. if not result.is_success():
  329. log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
  330. raise CustomException(msg=f"手工批量发放额度失败: {result.msg}")
  331. # 保存批次记录到本地
  332. try:
  333. issue_batch_crud = IssueBatchCRUD(auth)
  334. total_amount = Decimal("0")
  335. if data.issue_target_info_list:
  336. for item in data.issue_target_info_list:
  337. try:
  338. total_amount += Decimal(item.issue_quota)
  339. except Exception:
  340. pass
  341. batch_data = {
  342. "issue_batch_id": result.issue_batch_id,
  343. "batch_no": data.batch_no,
  344. "institution_id": data.institution_id,
  345. "issue_name": data.issue_name,
  346. "quota_type": data.quota_type,
  347. "share_mode": data.share_mode,
  348. "total_count": len(data.issue_target_info_list or []),
  349. "total_amount": total_amount,
  350. "status": "ACTIVE",
  351. "effective_start_date": datetime.strptime(data.effective_start_date, "%Y-%m-%d %H:%M:%S"),
  352. "effective_end_date": datetime.strptime(data.effective_end_date, "%Y-%m-%d %H:%M:%S"),
  353. "issue_desc": data.issue_desc,
  354. }
  355. await issue_batch_crud.create(batch_data)
  356. except Exception as e:
  357. log.warning(f"保存发放批次记录失败(不影响发放): {e}")
  358. # 组装校验失败列表(在更新本地记录前获取,用于过滤)
  359. failed_owner_ids: set[str] = set()
  360. if hasattr(result, 'issue_quota_check_failed_list') and result.issue_quota_check_failed_list:
  361. failed_owner_ids = {
  362. str(getattr(f, 'owner_id', ''))
  363. for f in result.issue_quota_check_failed_list
  364. if getattr(f, 'owner_id', None)
  365. }
  366. # 更新本地额度记录(跳过校验失败的员工)
  367. try:
  368. from app.plugin.module_payment.expense.quota.model import QuotaModel
  369. from app.plugin.module_payment.expense.quota.enums import QuotaStatusEnum
  370. from sqlalchemy import insert, select, update as sa_update
  371. tenant_id = auth.user.tenant_id if auth.user else 1
  372. effective_start = datetime.strptime(data.effective_start_date, "%Y-%m-%d %H:%M:%S") if data.effective_start_date else None
  373. effective_end = datetime.strptime(data.effective_end_date, "%Y-%m-%d %H:%M:%S") if data.effective_end_date else None
  374. updated_count = 0
  375. inserted_count = 0
  376. if data.issue_target_info_list:
  377. for item in data.issue_target_info_list:
  378. emp_id = (item.owner_id or "")
  379. # 跳过支付宝校验失败的员工
  380. if emp_id in failed_owner_ids:
  381. log.warning(f"手工发放 - 跳过校验失败员工: owner_id={emp_id}")
  382. continue
  383. try:
  384. quota_amount = Decimal(item.issue_quota)
  385. except Exception:
  386. quota_amount = Decimal("0")
  387. # 查询是否已有该员工在该制度下的额度记录
  388. check = select(QuotaModel).where(
  389. QuotaModel.employee_id == emp_id,
  390. QuotaModel.institution_id == data.institution_id,
  391. )
  392. existing = await auth.db.execute(check)
  393. existing_quota = existing.scalar_one_or_none()
  394. if existing_quota:
  395. # 已有记录 → 更新金额、状态、有效期
  396. upd = sa_update(QuotaModel).where(
  397. QuotaModel.id == existing_quota.id
  398. ).values(
  399. total_amount=quota_amount,
  400. available_amount=quota_amount,
  401. status=QuotaStatusEnum.QUOTA_ACTIVE.value,
  402. quota_type=data.quota_type,
  403. valid_from=effective_start,
  404. valid_to=effective_end,
  405. target_type="INSTITUTION",
  406. target_id=data.institution_id,
  407. )
  408. await auth.db.execute(upd)
  409. updated_count += 1
  410. else:
  411. # 没有记录 → 新增
  412. stmt = insert(QuotaModel).values(
  413. employee_id=emp_id,
  414. institution_id=data.institution_id,
  415. quota_type=data.quota_type,
  416. target_type="INSTITUTION",
  417. target_id=data.institution_id,
  418. out_biz_no=f"batch_{data.batch_no}_{emp_id}",
  419. total_amount=quota_amount,
  420. available_amount=quota_amount,
  421. status=QuotaStatusEnum.QUOTA_ACTIVE.value,
  422. valid_from=effective_start,
  423. valid_to=effective_end,
  424. enterprise_id=data.enterprise_id,
  425. tenant_id=tenant_id,
  426. )
  427. await auth.db.execute(stmt)
  428. inserted_count += 1
  429. await auth.db.flush()
  430. log.info(f"手工发放 - 更新 {updated_count} 条、新增 {inserted_count} 条额度记录到本地(跳过 {len(failed_owner_ids)} 条校验失败)")
  431. # 查询支付宝端的发放记录,获取每个员工的 real quota_id 并更新本地记录
  432. try:
  433. from alipay.aop.api.request.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest import (
  434. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest,
  435. )
  436. from alipay.aop.api.domain.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel import (
  437. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel,
  438. )
  439. from alipay.aop.api.response.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse import (
  440. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse,
  441. )
  442. records_model = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel()
  443. records_model.enterprise_id = data.enterprise_id
  444. records_model.institution_id = data.institution_id
  445. records_model.issue_batch_id = result.issue_batch_id
  446. records_model.page_size = len(data.issue_target_info_list)
  447. records_model.page_num = 1
  448. records_request = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest()
  449. records_request.biz_model = records_model
  450. records_response = client.execute(records_request)
  451. if records_response:
  452. records_result = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse()
  453. records_result.parse_response_content(records_response)
  454. if records_result.is_success() and hasattr(records_result, 'issue_record_info_list') and records_result.issue_record_info_list:
  455. updated_quota_id_count = 0
  456. for r in records_result.issue_record_info_list:
  457. quota_id = getattr(r, 'quota_id', None)
  458. owner_id = getattr(r, 'owner_id', None)
  459. if quota_id and owner_id:
  460. upd_q = sa_update(QuotaModel).where(
  461. QuotaModel.employee_id == owner_id,
  462. QuotaModel.institution_id == data.institution_id,
  463. ).values(quota_id=quota_id)
  464. await auth.db.execute(upd_q)
  465. updated_quota_id_count += 1
  466. if updated_quota_id_count:
  467. await auth.db.flush()
  468. log.info(f"手工发放 - 已从支付宝查询并更新 {updated_quota_id_count} 条记录的 quota_id")
  469. except Exception as e:
  470. log.warning(f"查询支付宝发放记录获取 quota_id 失败(不影响发放): {e}")
  471. except Exception as e:
  472. log.warning(f"保存额度记录到本地失败(不影响发放): {e}")
  473. # 组装校验失败列表
  474. failed_list = None
  475. if hasattr(result, 'issue_quota_check_failed_list') and result.issue_quota_check_failed_list:
  476. failed_list = []
  477. for f in result.issue_quota_check_failed_list:
  478. failed_list.append(IssueQuotaCheckFailedItem(
  479. user_name=getattr(f, 'user_name', None),
  480. owner_type=getattr(f, 'owner_type', None),
  481. owner_id=getattr(f, 'owner_id', None),
  482. owner_open_id=getattr(f, 'owner_open_id', None),
  483. issue_quota=getattr(f, 'issue_quota', None),
  484. message=getattr(f, 'message', None),
  485. result=getattr(f, 'result', None),
  486. ))
  487. return IssueBatchCreateOutSchema(
  488. issue_batch_id=result.issue_batch_id,
  489. issue_quota_check_failed_list=failed_list,
  490. )
  491. @classmethod
  492. async def issue_batch_cancel_service(
  493. cls, auth: AuthSchema, data: IssueBatchCancelSchema
  494. ) -> IssueBatchCancelOutSchema:
  495. """
  496. 作废手工发放批次
  497. 调用: alipay.ebpp.invoice.expensecontrol.issuebatch.cancel
  498. """
  499. try:
  500. from alipay.aop.api.request.AlipayEbppInvoiceExpensecontrolIssuebatchCancelRequest import (
  501. AlipayEbppInvoiceExpensecontrolIssuebatchCancelRequest,
  502. )
  503. from alipay.aop.api.domain.AlipayEbppInvoiceExpensecontrolIssuebatchCancelModel import (
  504. AlipayEbppInvoiceExpensecontrolIssuebatchCancelModel,
  505. )
  506. from alipay.aop.api.response.AlipayEbppInvoiceExpensecontrolIssuebatchCancelResponse import (
  507. AlipayEbppInvoiceExpensecontrolIssuebatchCancelResponse,
  508. )
  509. except ImportError:
  510. raise CustomException(msg="支付宝SDK未正确安装,请检查alipay-sdk-python依赖")
  511. model = AlipayEbppInvoiceExpensecontrolIssuebatchCancelModel()
  512. model.enterprise_id = data.enterprise_id
  513. model.institution_id = data.institution_id
  514. model.issue_batch_id = data.issue_batch_id
  515. request = AlipayEbppInvoiceExpensecontrolIssuebatchCancelRequest()
  516. request.biz_model = model
  517. client = AlipayClient.get_client()
  518. response = client.execute(request)
  519. if not response:
  520. raise CustomException(msg="作废手工发放批次失败: 无响应")
  521. result = AlipayEbppInvoiceExpensecontrolIssuebatchCancelResponse()
  522. result.parse_response_content(response)
  523. if not result.is_success():
  524. log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
  525. raise CustomException(msg=f"作废手工发放批次失败: {result.msg}")
  526. # 更新本地批次记录状态
  527. try:
  528. issue_batch_crud = IssueBatchCRUD(auth)
  529. batch = await issue_batch_crud.get_by_issue_batch_id(data.issue_batch_id)
  530. if batch:
  531. setattr(batch, 'status', 'CANCELLED')
  532. await issue_batch_crud.update(batch.id, {"status": "CANCELLED"})
  533. except Exception as e:
  534. log.warning(f"更新批次本地状态失败(不影响作废): {e}")
  535. return IssueBatchCancelOutSchema(
  536. result=getattr(result, 'result', False),
  537. )
  538. @classmethod
  539. async def issue_batch_records_query_service(
  540. cls, auth: AuthSchema, data: IssueBatchRecordsQuerySchema
  541. ) -> IssueBatchRecordsQueryOutSchema:
  542. """
  543. 查询手工发放发放明细
  544. 调用: alipay.ebpp.invoice.issuebatch.issuerecords.batchquery
  545. """
  546. try:
  547. from alipay.aop.api.request.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest import (
  548. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest,
  549. )
  550. from alipay.aop.api.domain.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel import (
  551. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel,
  552. )
  553. from alipay.aop.api.response.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse import (
  554. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse,
  555. )
  556. except ImportError:
  557. raise CustomException(msg="支付宝SDK未正确安装,请检查alipay-sdk-python依赖")
  558. model = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel()
  559. model.enterprise_id = data.enterprise_id
  560. model.institution_id = data.institution_id
  561. model.issue_batch_id = data.issue_batch_id
  562. model.page_size = data.page_size
  563. model.page_num = data.page_num
  564. request = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest()
  565. request.biz_model = model
  566. client = AlipayClient.get_client()
  567. response = client.execute(request)
  568. if not response:
  569. raise CustomException(msg="查询手工发放明细失败: 无响应")
  570. result = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse()
  571. result.parse_response_content(response)
  572. if not result.is_success():
  573. log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
  574. raise CustomException(msg=f"查询手工发放明细失败: {result.msg}")
  575. record_list = None
  576. if hasattr(result, 'issue_record_info_list') and result.issue_record_info_list:
  577. record_list = []
  578. for r in result.issue_record_info_list:
  579. record_list.append(IssueRecordInfoItem(
  580. quota_id=getattr(r, 'quota_id', None),
  581. issue_quota=getattr(r, 'issue_quota', None),
  582. issue_status=getattr(r, 'issue_status', None),
  583. owner_type=getattr(r, 'owner_type', None),
  584. owner_id=getattr(r, 'owner_id', None),
  585. owner_open_id=getattr(r, 'owner_open_id', None),
  586. user_name=getattr(r, 'user_name', None),
  587. currency=getattr(r, 'currency', None),
  588. ))
  589. return IssueBatchRecordsQueryOutSchema(
  590. page_num=getattr(result, 'page_num', data.page_num),
  591. page_size=getattr(result, 'page_size', data.page_size),
  592. total_page_count=getattr(result, 'total_page_count', "0"),
  593. issue_record_info_list=record_list,
  594. )
  595. @classmethod
  596. async def list_batch_service(
  597. cls,
  598. auth: AuthSchema,
  599. page_no: int = 1,
  600. page_size: int = 20,
  601. search: dict | None = None,
  602. ) -> dict:
  603. """分页查询手工发放批次列表(本地DB)"""
  604. crud = IssueBatchCRUD(auth)
  605. offset = (page_no - 1) * page_size
  606. return await crud.page(
  607. offset=offset,
  608. limit=page_size,
  609. order_by=[{"id": "desc"}],
  610. search=search or {},
  611. out_schema=IssueBatchListOutSchema,
  612. )
  613. @classmethod
  614. async def list_service(
  615. cls,
  616. auth: AuthSchema,
  617. page_no: int = 1,
  618. page_size: int = 20,
  619. search: dict | None = None,
  620. ) -> dict:
  621. crud = QuotaCRUD(auth)
  622. offset = (page_no - 1) * page_size
  623. result = await crud.page(
  624. offset=offset,
  625. limit=page_size,
  626. order_by=[{"id": "desc"}],
  627. search=search or {},
  628. out_schema=QuotaListOutSchema,
  629. )
  630. # 如果指定了 institution_id,同时查询支付宝端自动发放的额度
  631. if search and search.get("institution_id"):
  632. try:
  633. from alipay.aop.api.request.AlipayEbppInvoiceExpensecontrolQuotaQueryRequest import (
  634. AlipayEbppInvoiceExpensecontrolQuotaQueryRequest,
  635. )
  636. from alipay.aop.api.domain.AlipayEbppInvoiceExpensecontrolQuotaQueryModel import (
  637. AlipayEbppInvoiceExpensecontrolQuotaQueryModel,
  638. )
  639. from alipay.aop.api.response.AlipayEbppInvoiceExpensecontrolQuotaQueryResponse import (
  640. AlipayEbppInvoiceExpensecontrolQuotaQueryResponse,
  641. )
  642. alipay_model = AlipayEbppInvoiceExpensecontrolQuotaQueryModel()
  643. # owner_type 为必填字段,查询制度下所有额度时使用通用类型
  644. alipay_model.owner_type = "ENTERPRISE_PAY_UID"
  645. alipay_model.target_type = "INSTITUTION"
  646. alipay_model.target_id = search["institution_id"]
  647. alipay_model.page_size = 100
  648. alipay_model.page_num = 1
  649. request = AlipayEbppInvoiceExpensecontrolQuotaQueryRequest()
  650. request.biz_model = alipay_model
  651. client = AlipayClient.get_client()
  652. response = client.execute(request)
  653. if response:
  654. alipay_result = AlipayEbppInvoiceExpensecontrolQuotaQueryResponse()
  655. alipay_result.parse_response_content(response)
  656. if alipay_result.is_success() and hasattr(alipay_result, 'quota_detail_info_list') and alipay_result.quota_detail_info_list:
  657. # 将支付宝端额度合并到结果中(去重)
  658. existing_ids = {item.get("quota_id") for item in result.get("items", []) if item.get("quota_id")}
  659. for q in alipay_result.quota_detail_info_list:
  660. qid = getattr(q, 'quota_id', None)
  661. if qid and qid not in existing_ids:
  662. result["items"].append({
  663. "quota_id": qid,
  664. "target_type": getattr(q, 'target_type', None),
  665. "target_id": getattr(q, 'target_id', None),
  666. "quota_type": getattr(q, 'quota_type', None),
  667. "employee_id": getattr(q, 'owner_id', None),
  668. "total_amount": getattr(q, 'total_amount', None),
  669. "available_amount": getattr(q, 'available_amount', None),
  670. "status": getattr(q, 'status', "QUOTA_ACTIVE"),
  671. "created_time": getattr(q, 'effective_start_date', None),
  672. })
  673. existing_ids.add(qid)
  674. result["total"] = len(result["items"])
  675. except Exception as e:
  676. log.warning(f"查询支付宝端额度失败(不影响本地数据): {e}")
  677. return result