service.py 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949
  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. # 本地检查批次号是否已存在,避免无效调用支付宝
  295. try:
  296. issue_batch_crud = IssueBatchCRUD(auth)
  297. existing_batch = await issue_batch_crud.get_by_batch_no(data.batch_no)
  298. if existing_batch:
  299. raise CustomException(msg=f"批次号 {data.batch_no} 已存在,请勿重复创建")
  300. except CustomException:
  301. raise
  302. except Exception:
  303. pass
  304. model = AlipayEbppInvoiceExpensecontrolIssuebatchCreateModel()
  305. model.enterprise_id = data.enterprise_id
  306. model.issue_name = data.issue_name
  307. model.quota_type = data.quota_type
  308. model.effective_start_date = data.effective_start_date
  309. model.effective_end_date = data.effective_end_date
  310. model.institution_id = data.institution_id
  311. model.batch_no = data.batch_no
  312. model.share_mode = data.share_mode
  313. if data.issue_desc:
  314. model.issue_desc = data.issue_desc
  315. if data.issue_target_info_list:
  316. target_list = []
  317. for item in data.issue_target_info_list:
  318. target = IssueTargetInfoContent()
  319. target.issue_quota = item.issue_quota
  320. if item.owner_open_id:
  321. target.owner_open_id = item.owner_open_id
  322. if item.owner_id:
  323. target.owner_id = item.owner_id
  324. if item.user_name:
  325. target.user_name = item.user_name
  326. if item.owner_type:
  327. target.owner_type = item.owner_type
  328. target_list.append(target)
  329. model.issue_target_info_list = target_list
  330. request = AlipayEbppInvoiceExpensecontrolIssuebatchCreateRequest()
  331. request.biz_model = model
  332. client = AlipayClient.get_client()
  333. response = client.execute(request)
  334. if not response:
  335. raise CustomException(msg="手工批量发放额度失败: 无响应")
  336. result = AlipayEbppInvoiceExpensecontrolIssuebatchCreateResponse()
  337. result.parse_response_content(response)
  338. if not result.is_success():
  339. sub_msg = getattr(result, 'sub_msg', '') or ''
  340. err_detail = f"{result.msg}" + (f" - {sub_msg}" if sub_msg else "")
  341. log.error(f"支付宝接口调用失败: {result.code} - {err_detail}")
  342. raise CustomException(msg=f"手工批量发放额度失败: {sub_msg or result.msg}")
  343. # 保存批次记录到本地
  344. try:
  345. issue_batch_crud = IssueBatchCRUD(auth)
  346. total_amount = Decimal("0")
  347. if data.issue_target_info_list:
  348. for item in data.issue_target_info_list:
  349. try:
  350. total_amount += Decimal(item.issue_quota)
  351. except Exception:
  352. pass
  353. batch_data = {
  354. "issue_batch_id": result.issue_batch_id,
  355. "batch_no": data.batch_no,
  356. "institution_id": data.institution_id,
  357. "issue_name": data.issue_name,
  358. "quota_type": data.quota_type,
  359. "share_mode": data.share_mode,
  360. "total_count": len(data.issue_target_info_list or []),
  361. "total_amount": total_amount,
  362. "status": "ACTIVE",
  363. "effective_start_date": datetime.strptime(data.effective_start_date, "%Y-%m-%d %H:%M:%S"),
  364. "effective_end_date": datetime.strptime(data.effective_end_date, "%Y-%m-%d %H:%M:%S"),
  365. "issue_desc": data.issue_desc,
  366. }
  367. await issue_batch_crud.create(batch_data)
  368. except Exception as e:
  369. log.warning(f"保存发放批次记录失败(不影响发放): {e}")
  370. # 组装校验失败列表(在更新本地记录前获取,用于过滤)
  371. failed_owner_ids: set[str] = set()
  372. if hasattr(result, 'issue_quota_check_failed_list') and result.issue_quota_check_failed_list:
  373. failed_owner_ids = {
  374. str(getattr(f, 'owner_id', ''))
  375. for f in result.issue_quota_check_failed_list
  376. if getattr(f, 'owner_id', None)
  377. }
  378. # 为每个员工插入新的额度记录(每次发放独立记录,支持同一员工多次发放)
  379. try:
  380. from app.plugin.module_payment.expense.quota.model import QuotaModel
  381. from app.plugin.module_payment.expense.quota.enums import QuotaStatusEnum
  382. from sqlalchemy import insert
  383. tenant_id = auth.user.tenant_id if auth.user else 1
  384. effective_start = datetime.strptime(data.effective_start_date, "%Y-%m-%d %H:%M:%S") if data.effective_start_date else None
  385. effective_end = datetime.strptime(data.effective_end_date, "%Y-%m-%d %H:%M:%S") if data.effective_end_date else None
  386. inserted_count = 0
  387. if data.issue_target_info_list:
  388. for item in data.issue_target_info_list:
  389. emp_id = (item.owner_id or "")
  390. # 跳过支付宝校验失败的员工
  391. if emp_id in failed_owner_ids:
  392. log.warning(f"手工发放 - 跳过校验失败员工: owner_id={emp_id}")
  393. continue
  394. try:
  395. quota_amount = Decimal(item.issue_quota)
  396. except Exception:
  397. quota_amount = Decimal("0")
  398. # 每次都新增独立记录,不覆盖已有记录
  399. stmt = insert(QuotaModel).values(
  400. employee_id=emp_id,
  401. institution_id=data.institution_id,
  402. quota_type=data.quota_type,
  403. target_type="INSTITUTION",
  404. target_id=data.institution_id,
  405. out_biz_no=f"batch_{data.batch_no}_{emp_id}",
  406. total_amount=quota_amount,
  407. available_amount=quota_amount,
  408. status=QuotaStatusEnum.QUOTA_ACTIVE.value,
  409. valid_from=effective_start,
  410. valid_to=effective_end,
  411. enterprise_id=data.enterprise_id,
  412. tenant_id=tenant_id,
  413. )
  414. await auth.db.execute(stmt)
  415. inserted_count += 1
  416. await auth.db.flush()
  417. log.info(f"手工发放 - 新增 {inserted_count} 条额度记录(跳过 {len(failed_owner_ids)} 条校验失败)")
  418. # 查询支付宝端的发放记录,获取每个员工的 real quota_id 并更新本地记录
  419. try:
  420. from alipay.aop.api.request.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest import (
  421. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest,
  422. )
  423. from alipay.aop.api.domain.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel import (
  424. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel,
  425. )
  426. from alipay.aop.api.response.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse import (
  427. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse,
  428. )
  429. records_model = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel()
  430. records_model.enterprise_id = data.enterprise_id
  431. records_model.institution_id = data.institution_id
  432. records_model.issue_batch_id = result.issue_batch_id
  433. records_model.page_size = len(data.issue_target_info_list)
  434. records_model.page_num = 1
  435. records_request = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest()
  436. records_request.biz_model = records_model
  437. records_response = client.execute(records_request)
  438. if records_response:
  439. records_result = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse()
  440. records_result.parse_response_content(records_response)
  441. if records_result.is_success() and hasattr(records_result, 'issue_record_info_list') and records_result.issue_record_info_list:
  442. updated_quota_id_count = 0
  443. for r in records_result.issue_record_info_list:
  444. quota_id = getattr(r, 'quota_id', None)
  445. owner_id = getattr(r, 'owner_id', None)
  446. if quota_id and owner_id:
  447. # 按 out_biz_no 模式匹配,只更新本批次插入的记录
  448. q_upd = sa_update(QuotaModel).where(
  449. QuotaModel.out_biz_no == f"batch_{data.batch_no}_{owner_id}",
  450. QuotaModel.institution_id == data.institution_id,
  451. ).values(quota_id=quota_id)
  452. await auth.db.execute(q_upd)
  453. updated_quota_id_count += 1
  454. if updated_quota_id_count:
  455. await auth.db.flush()
  456. log.info(f"手工发放 - 已从支付宝查询并更新 {updated_quota_id_count} 条记录的 quota_id")
  457. except Exception as e:
  458. log.warning(f"查询支付宝发放记录获取 quota_id 失败(不影响发放): {e}")
  459. except Exception as e:
  460. log.warning(f"保存额度记录到本地失败(不影响发放): {e}")
  461. # 组装校验失败列表
  462. failed_list = None
  463. if hasattr(result, 'issue_quota_check_failed_list') and result.issue_quota_check_failed_list:
  464. failed_list = []
  465. for f in result.issue_quota_check_failed_list:
  466. failed_list.append(IssueQuotaCheckFailedItem(
  467. user_name=getattr(f, 'user_name', None),
  468. owner_type=getattr(f, 'owner_type', None),
  469. owner_id=getattr(f, 'owner_id', None),
  470. owner_open_id=getattr(f, 'owner_open_id', None),
  471. issue_quota=getattr(f, 'issue_quota', None),
  472. message=getattr(f, 'message', None),
  473. result=getattr(f, 'result', None),
  474. ))
  475. return IssueBatchCreateOutSchema(
  476. issue_batch_id=result.issue_batch_id,
  477. issue_quota_check_failed_list=failed_list,
  478. )
  479. @classmethod
  480. async def issue_batch_cancel_service(
  481. cls, auth: AuthSchema, data: IssueBatchCancelSchema
  482. ) -> IssueBatchCancelOutSchema:
  483. """
  484. 作废手工发放批次
  485. 调用: alipay.ebpp.invoice.expensecontrol.issuebatch.cancel
  486. """
  487. try:
  488. from alipay.aop.api.request.AlipayEbppInvoiceExpensecontrolIssuebatchCancelRequest import (
  489. AlipayEbppInvoiceExpensecontrolIssuebatchCancelRequest,
  490. )
  491. from alipay.aop.api.domain.AlipayEbppInvoiceExpensecontrolIssuebatchCancelModel import (
  492. AlipayEbppInvoiceExpensecontrolIssuebatchCancelModel,
  493. )
  494. from alipay.aop.api.response.AlipayEbppInvoiceExpensecontrolIssuebatchCancelResponse import (
  495. AlipayEbppInvoiceExpensecontrolIssuebatchCancelResponse,
  496. )
  497. except ImportError:
  498. raise CustomException(msg="支付宝SDK未正确安装,请检查alipay-sdk-python依赖")
  499. model = AlipayEbppInvoiceExpensecontrolIssuebatchCancelModel()
  500. model.enterprise_id = data.enterprise_id
  501. model.institution_id = data.institution_id
  502. model.issue_batch_id = data.issue_batch_id
  503. request = AlipayEbppInvoiceExpensecontrolIssuebatchCancelRequest()
  504. request.biz_model = model
  505. client = AlipayClient.get_client()
  506. response = client.execute(request)
  507. if not response:
  508. raise CustomException(msg="作废手工发放批次失败: 无响应")
  509. result = AlipayEbppInvoiceExpensecontrolIssuebatchCancelResponse()
  510. result.parse_response_content(response)
  511. if not result.is_success():
  512. sub_msg = getattr(result, 'sub_msg', '') or ''
  513. err_detail = f"{result.msg}" + (f" - {sub_msg}" if sub_msg else "")
  514. log.error(f"支付宝接口调用失败: {result.code} - {err_detail}")
  515. raise CustomException(msg=f"作废手工发放批次失败: {sub_msg or result.msg}")
  516. # 更新本地批次记录状态
  517. try:
  518. issue_batch_crud = IssueBatchCRUD(auth)
  519. batch = await issue_batch_crud.get_by_issue_batch_id(data.issue_batch_id)
  520. if batch:
  521. setattr(batch, 'status', 'CANCELLED')
  522. await issue_batch_crud.update(batch.id, {"status": "CANCELLED"})
  523. except Exception as e:
  524. log.warning(f"更新批次本地状态失败(不影响作废): {e}")
  525. # 作废批次后,删除该批次创建的额度记录
  526. try:
  527. from app.plugin.module_payment.expense.quota.model import QuotaModel
  528. from sqlalchemy import delete as sa_delete
  529. # 从本地批次记录获取 batch_no
  530. issue_batch_crud = IssueBatchCRUD(auth)
  531. batch = await issue_batch_crud.get_by_issue_batch_id(data.issue_batch_id)
  532. batch_no = batch.batch_no if batch else None
  533. if batch_no:
  534. # 按 out_biz_no 模式匹配: batch_{batch_no}_%
  535. pattern = f"batch_{batch_no}_%"
  536. del_stmt = sa_delete(QuotaModel).where(
  537. QuotaModel.out_biz_no.like(pattern),
  538. QuotaModel.institution_id == data.institution_id,
  539. )
  540. await auth.db.execute(del_stmt)
  541. await auth.db.flush()
  542. log.info(f"批次作废 - 已删除该批次创建的额度记录: batch_no={batch_no}")
  543. except Exception as e:
  544. log.warning(f"删除额度记录失败(不影响作废): {e}")
  545. return IssueBatchCancelOutSchema(
  546. result=getattr(result, 'result', False),
  547. )
  548. return IssueBatchCancelOutSchema(
  549. result=getattr(result, 'result', False),
  550. )
  551. @classmethod
  552. async def issue_batch_records_query_service(
  553. cls, auth: AuthSchema, data: IssueBatchRecordsQuerySchema
  554. ) -> IssueBatchRecordsQueryOutSchema:
  555. """
  556. 查询手工发放发放明细
  557. 调用: alipay.ebpp.invoice.issuebatch.issuerecords.batchquery
  558. """
  559. try:
  560. from alipay.aop.api.request.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest import (
  561. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest,
  562. )
  563. from alipay.aop.api.domain.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel import (
  564. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel,
  565. )
  566. from alipay.aop.api.response.AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse import (
  567. AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse,
  568. )
  569. except ImportError:
  570. raise CustomException(msg="支付宝SDK未正确安装,请检查alipay-sdk-python依赖")
  571. model = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryModel()
  572. model.enterprise_id = data.enterprise_id
  573. model.institution_id = data.institution_id
  574. model.issue_batch_id = data.issue_batch_id
  575. model.page_size = data.page_size
  576. model.page_num = data.page_num
  577. request = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryRequest()
  578. request.biz_model = model
  579. client = AlipayClient.get_client()
  580. response = client.execute(request)
  581. if not response:
  582. raise CustomException(msg="查询手工发放明细失败: 无响应")
  583. result = AlipayEbppInvoiceIssuebatchIssuerecordsBatchqueryResponse()
  584. result.parse_response_content(response)
  585. if not result.is_success():
  586. log.error(f"支付宝接口调用失败: {result.code} - {result.msg}")
  587. raise CustomException(msg=f"查询手工发放明细失败: {result.msg}")
  588. record_list = None
  589. if hasattr(result, 'issue_record_info_list') and result.issue_record_info_list:
  590. record_list = []
  591. for r in result.issue_record_info_list:
  592. record_list.append(IssueRecordInfoItem(
  593. quota_id=getattr(r, 'quota_id', None),
  594. issue_quota=getattr(r, 'issue_quota', None),
  595. issue_status=getattr(r, 'issue_status', None),
  596. owner_type=getattr(r, 'owner_type', None),
  597. owner_id=getattr(r, 'owner_id', None),
  598. owner_open_id=getattr(r, 'owner_open_id', None),
  599. user_name=getattr(r, 'user_name', None),
  600. currency=getattr(r, 'currency', None),
  601. ))
  602. return IssueBatchRecordsQueryOutSchema(
  603. page_num=getattr(result, 'page_num', data.page_num),
  604. page_size=getattr(result, 'page_size', data.page_size),
  605. total_page_count=getattr(result, 'total_page_count', 0),
  606. issue_record_info_list=record_list,
  607. )
  608. @classmethod
  609. async def list_batch_service(
  610. cls,
  611. auth: AuthSchema,
  612. page_no: int = 1,
  613. page_size: int = 20,
  614. search: dict | None = None,
  615. ) -> dict:
  616. """分页查询手工发放批次列表(本地DB)"""
  617. crud = IssueBatchCRUD(auth)
  618. offset = (page_no - 1) * page_size
  619. return await crud.page(
  620. offset=offset,
  621. limit=page_size,
  622. order_by=[{"id": "desc"}],
  623. search=search or {},
  624. out_schema=IssueBatchListOutSchema,
  625. )
  626. @classmethod
  627. async def list_employee_quota_records_service(
  628. cls,
  629. auth: AuthSchema,
  630. employee_id: str,
  631. institution_id: str | None = None,
  632. ) -> list[dict]:
  633. """查询员工的额度记录列表"""
  634. from app.plugin.module_payment.expense.quota.model import QuotaModel
  635. from sqlalchemy import select
  636. where = [QuotaModel.employee_id == employee_id]
  637. if institution_id:
  638. where.append(QuotaModel.institution_id == institution_id)
  639. stmt = select(QuotaModel).where(*where).order_by(QuotaModel.created_time.desc())
  640. result = await auth.db.execute(stmt)
  641. quotas = result.scalars().all()
  642. return [
  643. {
  644. "quota_id": q.quota_id,
  645. "out_biz_no": q.out_biz_no,
  646. "total_amount": float(q.total_amount) if q.total_amount else 0,
  647. "available_amount": float(q.available_amount) if q.available_amount else 0,
  648. "quota_type": q.quota_type,
  649. "status": q.status,
  650. "valid_from": q.valid_from,
  651. "valid_to": q.valid_to,
  652. "created_time": q.created_time,
  653. "institution_id": q.institution_id,
  654. }
  655. for q in quotas
  656. ]
  657. @classmethod
  658. async def adjust_quota_service(
  659. cls, auth: AuthSchema, data: AdjustQuotaSchema
  660. ) -> dict:
  661. """调整额度金额 (调Alipay quota.modify + 记录变更日志)"""
  662. from .crud import QuotaChangeLogCRUD
  663. from .model import QuotaModel
  664. from sqlalchemy import select, update as sa_update
  665. # 查询当前额度记录
  666. stmt = select(QuotaModel).where(QuotaModel.quota_id == data.quota_id)
  667. result = await auth.db.execute(stmt)
  668. quota = result.scalar_one_or_none()
  669. if not quota:
  670. raise CustomException(msg="额度记录不存在")
  671. current_available = float(quota.available_amount) if quota.available_amount else 0
  672. diff = round(data.amount - current_available, 2)
  673. outer_source_id = str(get_snowflake_id())
  674. # 调Alipay quota.modify
  675. try:
  676. from alipay.aop.api.request.AlipayEbppInvoiceExpensecontrolQuotaModifyRequest import (
  677. AlipayEbppInvoiceExpensecontrolQuotaModifyRequest,
  678. )
  679. from alipay.aop.api.domain.AlipayEbppInvoiceExpensecontrolQuotaModifyModel import (
  680. AlipayEbppInvoiceExpensecontrolQuotaModifyModel,
  681. )
  682. from alipay.aop.api.response.AlipayEbppInvoiceExpensecontrolQuotaModifyResponse import (
  683. AlipayEbppInvoiceExpensecontrolQuotaModifyResponse,
  684. )
  685. except ImportError:
  686. raise CustomException(msg="支付宝SDK未正确安装")
  687. model = AlipayEbppInvoiceExpensecontrolQuotaModifyModel()
  688. model.quota_id = data.quota_id
  689. model.action = "ADD" if diff >= 0 else "DEDUCT"
  690. model.outer_source_id = outer_source_id
  691. model.enterprise_id = data.enterprise_id
  692. model.amount = str(abs(diff))
  693. request = AlipayEbppInvoiceExpensecontrolQuotaModifyRequest()
  694. request.biz_model = model
  695. client = AlipayClient.get_client()
  696. response = client.execute(request)
  697. if not response:
  698. raise CustomException(msg="调整额度失败: 无响应")
  699. mod_result = AlipayEbppInvoiceExpensecontrolQuotaModifyResponse()
  700. mod_result.parse_response_content(response)
  701. if not mod_result.is_success():
  702. sub_msg = getattr(mod_result, 'sub_msg', '') or ''
  703. log.error(f"支付宝接口调用失败: {mod_result.code} - {mod_result.msg}")
  704. raise CustomException(msg=f"调整额度失败: {sub_msg or mod_result.msg}")
  705. # 更新本地额度记录
  706. new_available = data.amount
  707. new_total = float(quota.total_amount) if quota.total_amount else 0
  708. if diff > 0:
  709. new_total += diff
  710. else:
  711. new_total = max(0, new_total + diff)
  712. upd = sa_update(QuotaModel).where(
  713. QuotaModel.quota_id == data.quota_id
  714. ).values(
  715. total_amount=new_total,
  716. available_amount=new_available,
  717. )
  718. await auth.db.execute(upd)
  719. await auth.db.flush()
  720. # 记录变更日志
  721. try:
  722. log_crud = QuotaChangeLogCRUD(auth)
  723. await log_crud.create({
  724. "quota_id": data.quota_id,
  725. "employee_id": quota.employee_id or "",
  726. "institution_id": quota.institution_id or "",
  727. "change_type": "ADJUST",
  728. "coupon_name": quota.out_biz_no or "额度调整",
  729. "change_amount": diff,
  730. "before_amount": current_available,
  731. "after_amount": new_available,
  732. "change_desc": data.change_desc or "",
  733. "enterprise_id": data.enterprise_id,
  734. "tenant_id": auth.user.tenant_id if auth.user else 1,
  735. })
  736. except Exception as e:
  737. log.warning(f"记录变更日志失败(不影响调整): {e}")
  738. return {
  739. "quota_id": data.quota_id,
  740. "before_amount": current_available,
  741. "after_amount": new_available,
  742. "diff": diff,
  743. }
  744. @classmethod
  745. async def list_quota_changes_service(
  746. cls, auth: AuthSchema, quota_id: str
  747. ) -> list[dict]:
  748. """查询额度的变更记录"""
  749. from .crud import QuotaChangeLogCRUD
  750. crud = QuotaChangeLogCRUD(auth)
  751. logs = await crud.list(search={"quota_id": quota_id}, order_by=[{"id": "desc"}])
  752. return [
  753. {
  754. "coupon_name": log.coupon_name,
  755. "change_time": log.created_time,
  756. "change_amount": float(log.change_amount) if log.change_amount else 0,
  757. "change_desc": log.change_desc,
  758. "change_type": log.change_type,
  759. }
  760. for log in (logs or [])
  761. ]
  762. @classmethod
  763. async def list_service(
  764. cls,
  765. auth: AuthSchema,
  766. page_no: int = 1,
  767. page_size: int = 20,
  768. search: dict | None = None,
  769. ) -> dict:
  770. crud = QuotaCRUD(auth)
  771. offset = (page_no - 1) * page_size
  772. result = await crud.page(
  773. offset=offset,
  774. limit=page_size,
  775. order_by=[{"id": "desc"}],
  776. search=search or {},
  777. out_schema=QuotaListOutSchema,
  778. )
  779. # 如果指定了 institution_id,同时查询支付宝端自动发放的额度
  780. if search and search.get("institution_id"):
  781. try:
  782. from alipay.aop.api.request.AlipayEbppInvoiceExpensecontrolQuotaQueryRequest import (
  783. AlipayEbppInvoiceExpensecontrolQuotaQueryRequest,
  784. )
  785. from alipay.aop.api.domain.AlipayEbppInvoiceExpensecontrolQuotaQueryModel import (
  786. AlipayEbppInvoiceExpensecontrolQuotaQueryModel,
  787. )
  788. from alipay.aop.api.response.AlipayEbppInvoiceExpensecontrolQuotaQueryResponse import (
  789. AlipayEbppInvoiceExpensecontrolQuotaQueryResponse,
  790. )
  791. alipay_model = AlipayEbppInvoiceExpensecontrolQuotaQueryModel()
  792. # owner_type 为必填字段,查询制度下所有额度时使用通用类型
  793. alipay_model.owner_type = "ENTERPRISE_PAY_UID"
  794. alipay_model.target_type = "INSTITUTION"
  795. alipay_model.target_id = search["institution_id"]
  796. alipay_model.page_size = 100
  797. alipay_model.page_num = 1
  798. request = AlipayEbppInvoiceExpensecontrolQuotaQueryRequest()
  799. request.biz_model = alipay_model
  800. client = AlipayClient.get_client()
  801. response = client.execute(request)
  802. if response:
  803. alipay_result = AlipayEbppInvoiceExpensecontrolQuotaQueryResponse()
  804. alipay_result.parse_response_content(response)
  805. if alipay_result.is_success() and hasattr(alipay_result, 'quota_detail_info_list') and alipay_result.quota_detail_info_list:
  806. # 将支付宝端额度合并到结果中(去重)
  807. existing_ids = {item.get("quota_id") for item in result.get("items", []) if item.get("quota_id")}
  808. for q in alipay_result.quota_detail_info_list:
  809. qid = getattr(q, 'quota_id', None)
  810. if qid and qid not in existing_ids:
  811. result["items"].append({
  812. "quota_id": qid,
  813. "target_type": getattr(q, 'target_type', None),
  814. "target_id": getattr(q, 'target_id', None),
  815. "quota_type": getattr(q, 'quota_type', None),
  816. "employee_id": getattr(q, 'owner_id', None),
  817. "total_amount": getattr(q, 'total_amount', None),
  818. "available_amount": getattr(q, 'available_amount', None),
  819. "status": getattr(q, 'status', "QUOTA_ACTIVE"),
  820. "created_time": getattr(q, 'effective_start_date', None),
  821. })
  822. existing_ids.add(qid)
  823. result["total"] = len(result["items"])
  824. except Exception as e:
  825. log.warning(f"查询支付宝端额度失败(不影响本地数据): {e}")
  826. return result