pilot_service.py 139 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901
  1. """中试服务管理 CRUD(细胞产品类型/规格/价格体系/项目价格、渠道信息、体检机构)"""
  2. from __future__ import annotations
  3. import json
  4. import math
  5. from datetime import datetime
  6. from typing import Any, Dict, List, Optional, Tuple
  7. from sqlalchemy import and_, func, or_
  8. from sqlalchemy.orm import Session
  9. from app.core.utils import generate_uuid
  10. from app.crud.collection_protocol import urls_to_attachment_list
  11. from app.models.models import (
  12. BizCellProductType,
  13. BizCellProductSpec,
  14. BizCellProductPrice,
  15. BizCellProductProjectSystem,
  16. BizCellProductProjectSystemItem,
  17. BizChannel,
  18. BizChannelAccount,
  19. BizChannelAgreement,
  20. BizChannelAgreementProductPrice,
  21. BizChannelInvoice,
  22. BizChannelMailAddress,
  23. BizChannelProductQuota,
  24. BizChannelServiceOrder,
  25. BizChannelServiceOrderProduct,
  26. BizChannelServiceOrderTxn,
  27. BizCheckupInstitution,
  28. BizCheckupInstitutionInvoice,
  29. BizCheckupInstitutionProject,
  30. BizCheckupInstitutionPackage,
  31. BizCheckupInstitutionPackageItem,
  32. SysUser,
  33. )
  34. def _normalize_date_bound(value: str, *, end_of_day: bool = False) -> str:
  35. val = (value or "").strip()
  36. if len(val) == 10:
  37. return f"{val} 23:59:59" if end_of_day else f"{val} 00:00:00"
  38. return val
  39. def _user_nick_name_map(db: Session, user_ids: List[str]) -> Dict[str, str]:
  40. ids = list({x for x in user_ids if x})
  41. if not ids:
  42. return {}
  43. rows = db.query(SysUser.id, SysUser.nick_name).filter(SysUser.id.in_(ids)).all()
  44. return {u.id: (u.nick_name or "").strip() for u in rows}
  45. # ────────────────────────────────────────────
  46. # Entity 1: BizCellProductType 细胞产品类型
  47. # ────────────────────────────────────────────
  48. def _row_to_dict_cell_product_type(row) -> Dict[str, Any]:
  49. return {
  50. "id": row.id,
  51. "productTypeCode": row.product_type_code or "",
  52. "productTypeName": row.product_type_name or "",
  53. "tenantNo": row.tenant_no or "",
  54. "orderNum": row.order_num if row.order_num is not None else 0,
  55. "remark": row.remark or "",
  56. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  57. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  58. }
  59. def list_cell_product_types(db, *, page=1, page_size=10, keyword=None, tenant_no=None, user_info=None):
  60. def _filtered():
  61. q0 = db.query(BizCellProductType).filter(BizCellProductType.del_flag == 0)
  62. if tenant_no:
  63. q0 = q0.filter(BizCellProductType.tenant_no == tenant_no)
  64. kw = (keyword or "").strip()
  65. if kw:
  66. like = f"%{kw}%"
  67. q0 = q0.filter(
  68. or_(
  69. BizCellProductType.product_type_code.like(like),
  70. BizCellProductType.product_type_name.like(like),
  71. )
  72. )
  73. if user_info:
  74. from app.core.data_scope import apply_data_scope
  75. q0 = apply_data_scope(q0, BizCellProductType, user_info, db, tenant_no)
  76. return q0
  77. total = _filtered().order_by(None).with_entities(func.count(BizCellProductType.id)).scalar() or 0
  78. page = max(1, page)
  79. page_size = min(max(1, page_size), 100)
  80. rows = _filtered().order_by(BizCellProductType.create_time.desc()).offset((page - 1) * page_size).limit(page_size).all()
  81. return [_row_to_dict_cell_product_type(x) for x in rows], int(total)
  82. def get_cell_product_type(db, type_id, tenant_no=None):
  83. q = db.query(BizCellProductType).filter(and_(BizCellProductType.id == type_id, BizCellProductType.del_flag == 0))
  84. if tenant_no:
  85. q = q.filter(BizCellProductType.tenant_no == tenant_no)
  86. return q.first()
  87. def cell_product_type_code_exists(db, code, tenant_no, exclude_id=None):
  88. q = db.query(BizCellProductType).filter(
  89. and_(BizCellProductType.product_type_code == code, BizCellProductType.del_flag == 0)
  90. )
  91. if tenant_no:
  92. q = q.filter(BizCellProductType.tenant_no == tenant_no)
  93. if exclude_id:
  94. q = q.filter(BizCellProductType.id != exclude_id)
  95. return db.query(q.exists()).scalar()
  96. def create_cell_product_type(db, *, product_type_code, product_type_name, tenant_no="zkxw", order_num=None, remark=None, user_id=None):
  97. if cell_product_type_code_exists(db, product_type_code, tenant_no):
  98. raise ValueError(f"产品类型编码 {product_type_code} 已存在")
  99. obj = BizCellProductType(
  100. id=generate_uuid(),
  101. product_type_code=product_type_code,
  102. product_type_name=product_type_name,
  103. tenant_no=tenant_no,
  104. order_num=order_num or 0,
  105. remark=remark or "",
  106. create_by=user_id,
  107. create_time=datetime.now(),
  108. update_by=user_id,
  109. update_time=datetime.now(),
  110. del_flag=0,
  111. )
  112. db.add(obj)
  113. db.commit()
  114. db.refresh(obj)
  115. return obj
  116. def update_cell_product_type(db, type_id, *, product_type_code=None, product_type_name=None, order_num=None, remark=None, user_id=None, tenant_no=None):
  117. obj = get_cell_product_type(db, type_id, tenant_no)
  118. if not obj:
  119. raise ValueError("记录不存在")
  120. if product_type_code is not None and product_type_code != obj.product_type_code:
  121. if cell_product_type_code_exists(db, product_type_code, tenant_no, exclude_id=type_id):
  122. raise ValueError(f"产品类型编码 {product_type_code} 已存在")
  123. obj.product_type_code = product_type_code
  124. if product_type_name is not None:
  125. obj.product_type_name = product_type_name
  126. if order_num is not None:
  127. obj.order_num = order_num
  128. if remark is not None:
  129. obj.remark = remark
  130. obj.update_by = user_id
  131. obj.update_time = datetime.now()
  132. db.add(obj)
  133. db.commit()
  134. db.refresh(obj)
  135. return obj
  136. def soft_delete_cell_product_type(db, type_id, tenant_no=None):
  137. obj = get_cell_product_type(db, type_id, tenant_no)
  138. if not obj:
  139. raise ValueError("记录不存在")
  140. obj.del_flag = 1
  141. obj.update_time = datetime.now()
  142. db.add(obj)
  143. db.commit()
  144. # ────────────────────────────────────────────
  145. # Entity 2: BizCellProductSpec 细胞产品规格
  146. # ────────────────────────────────────────────
  147. def _row_to_dict_cell_product_spec(row) -> Dict[str, Any]:
  148. return {
  149. "id": row.id,
  150. "specCode": row.spec_code or "",
  151. "specName": row.spec_name or "",
  152. "tenantNo": row.tenant_no or "",
  153. "orderNum": row.order_num if row.order_num is not None else 0,
  154. "remark": row.remark or "",
  155. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  156. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  157. }
  158. def list_cell_product_specs(db, *, page=1, page_size=10, keyword=None, tenant_no=None, user_info=None):
  159. def _filtered():
  160. q0 = db.query(BizCellProductSpec).filter(BizCellProductSpec.del_flag == 0)
  161. if tenant_no:
  162. q0 = q0.filter(BizCellProductSpec.tenant_no == tenant_no)
  163. kw = (keyword or "").strip()
  164. if kw:
  165. like = f"%{kw}%"
  166. q0 = q0.filter(
  167. or_(
  168. BizCellProductSpec.spec_code.like(like),
  169. BizCellProductSpec.spec_name.like(like),
  170. )
  171. )
  172. if user_info:
  173. from app.core.data_scope import apply_data_scope
  174. q0 = apply_data_scope(q0, BizCellProductSpec, user_info, db, tenant_no)
  175. return q0
  176. total = _filtered().order_by(None).with_entities(func.count(BizCellProductSpec.id)).scalar() or 0
  177. page = max(1, page)
  178. page_size = min(max(1, page_size), 100)
  179. rows = _filtered().order_by(BizCellProductSpec.create_time.desc()).offset((page - 1) * page_size).limit(page_size).all()
  180. return [_row_to_dict_cell_product_spec(x) for x in rows], int(total)
  181. def get_cell_product_spec(db, spec_id, tenant_no=None):
  182. q = db.query(BizCellProductSpec).filter(and_(BizCellProductSpec.id == spec_id, BizCellProductSpec.del_flag == 0))
  183. if tenant_no:
  184. q = q.filter(BizCellProductSpec.tenant_no == tenant_no)
  185. return q.first()
  186. def cell_product_spec_code_exists(db, code, tenant_no, exclude_id=None):
  187. q = db.query(BizCellProductSpec).filter(
  188. and_(BizCellProductSpec.spec_code == code, BizCellProductSpec.del_flag == 0)
  189. )
  190. if tenant_no:
  191. q = q.filter(BizCellProductSpec.tenant_no == tenant_no)
  192. if exclude_id:
  193. q = q.filter(BizCellProductSpec.id != exclude_id)
  194. return db.query(q.exists()).scalar()
  195. def create_cell_product_spec(db, *, spec_code, spec_name, tenant_no="zkxw", order_num=None, remark=None, user_id=None):
  196. if cell_product_spec_code_exists(db, spec_code, tenant_no):
  197. raise ValueError(f"规格编码 {spec_code} 已存在")
  198. obj = BizCellProductSpec(
  199. id=generate_uuid(),
  200. spec_code=spec_code,
  201. spec_name=spec_name,
  202. tenant_no=tenant_no,
  203. order_num=order_num or 0,
  204. remark=remark or "",
  205. create_by=user_id,
  206. create_time=datetime.now(),
  207. update_by=user_id,
  208. update_time=datetime.now(),
  209. del_flag=0,
  210. )
  211. db.add(obj)
  212. db.commit()
  213. db.refresh(obj)
  214. return obj
  215. def update_cell_product_spec(db, spec_id, *, spec_code=None, spec_name=None, order_num=None, remark=None, user_id=None, tenant_no=None):
  216. obj = get_cell_product_spec(db, spec_id, tenant_no)
  217. if not obj:
  218. raise ValueError("记录不存在")
  219. if spec_code is not None and spec_code != obj.spec_code:
  220. if cell_product_spec_code_exists(db, spec_code, tenant_no, exclude_id=spec_id):
  221. raise ValueError(f"规格编码 {spec_code} 已存在")
  222. obj.spec_code = spec_code
  223. if spec_name is not None:
  224. obj.spec_name = spec_name
  225. if order_num is not None:
  226. obj.order_num = order_num
  227. if remark is not None:
  228. obj.remark = remark
  229. obj.update_by = user_id
  230. obj.update_time = datetime.now()
  231. db.add(obj)
  232. db.commit()
  233. db.refresh(obj)
  234. return obj
  235. def soft_delete_cell_product_spec(db, spec_id, tenant_no=None):
  236. obj = get_cell_product_spec(db, spec_id, tenant_no)
  237. if not obj:
  238. raise ValueError("记录不存在")
  239. obj.del_flag = 1
  240. obj.update_time = datetime.now()
  241. db.add(obj)
  242. db.commit()
  243. # ────────────────────────────────────────────
  244. # Entity 3: BizCellProductPrice 细胞产品价格
  245. # ────────────────────────────────────────────
  246. def _row_to_dict_cell_product_price(row) -> Dict[str, Any]:
  247. return {
  248. "id": row.id,
  249. "projectName": row.project_name or "",
  250. "projectCode": row.project_code or "",
  251. "cellTypeCode": row.cell_type_code or "",
  252. "cellTypeName": row.cell_type_name or "",
  253. "specText": row.spec_text or "",
  254. "specId": row.spec_id or "",
  255. "cellProductTypeId": row.cell_product_type_id or "",
  256. "unitPrice": float(row.unit_price) if row.unit_price is not None else None,
  257. "tenantNo": row.tenant_no or "",
  258. "orderNum": row.order_num if row.order_num is not None else 0,
  259. "remark": row.remark or "",
  260. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  261. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  262. }
  263. def list_cell_product_prices(db, *, page=1, page_size=10, keyword=None, tenant_no=None, cell_product_type_id=None, user_info=None):
  264. def _filtered():
  265. q0 = db.query(BizCellProductPrice).filter(BizCellProductPrice.del_flag == 0)
  266. if tenant_no:
  267. q0 = q0.filter(BizCellProductPrice.tenant_no == tenant_no)
  268. if cell_product_type_id:
  269. q0 = q0.filter(BizCellProductPrice.cell_product_type_id == cell_product_type_id)
  270. kw = (keyword or "").strip()
  271. if kw:
  272. like = f"%{kw}%"
  273. q0 = q0.filter(
  274. or_(
  275. BizCellProductPrice.project_name.like(like),
  276. BizCellProductPrice.project_code.like(like),
  277. BizCellProductPrice.cell_type_name.like(like),
  278. )
  279. )
  280. if user_info:
  281. from app.core.data_scope import apply_data_scope
  282. q0 = apply_data_scope(q0, BizCellProductPrice, user_info, db, tenant_no)
  283. return q0
  284. total = _filtered().order_by(None).with_entities(func.count(BizCellProductPrice.id)).scalar() or 0
  285. page = max(1, page)
  286. page_size = min(max(1, page_size), 100)
  287. rows = _filtered().order_by(BizCellProductPrice.create_time.desc()).offset((page - 1) * page_size).limit(page_size).all()
  288. return [_row_to_dict_cell_product_price(x) for x in rows], int(total)
  289. def get_cell_product_price(db, price_id, tenant_no=None):
  290. q = db.query(BizCellProductPrice).filter(and_(BizCellProductPrice.id == price_id, BizCellProductPrice.del_flag == 0))
  291. if tenant_no:
  292. q = q.filter(BizCellProductPrice.tenant_no == tenant_no)
  293. return q.first()
  294. def cell_product_price_code_exists(db, code, tenant_no, exclude_id=None):
  295. q = db.query(BizCellProductPrice).filter(
  296. and_(BizCellProductPrice.project_code == code, BizCellProductPrice.del_flag == 0)
  297. )
  298. if tenant_no:
  299. q = q.filter(BizCellProductPrice.tenant_no == tenant_no)
  300. if exclude_id:
  301. q = q.filter(BizCellProductPrice.id != exclude_id)
  302. return db.query(q.exists()).scalar()
  303. def create_cell_product_price(
  304. db,
  305. *,
  306. project_name,
  307. project_code,
  308. cell_type_code,
  309. cell_type_name,
  310. spec_text,
  311. spec_id,
  312. cell_product_type_id,
  313. unit_price,
  314. tenant_no="zkxw",
  315. order_num=None,
  316. remark=None,
  317. user_id=None,
  318. ):
  319. if cell_product_price_code_exists(db, project_code, tenant_no):
  320. raise ValueError(f"项目编码 {project_code} 已存在")
  321. obj = BizCellProductPrice(
  322. id=generate_uuid(),
  323. project_name=project_name,
  324. project_code=project_code,
  325. cell_type_code=cell_type_code,
  326. cell_type_name=cell_type_name,
  327. spec_text=spec_text or "",
  328. spec_id=spec_id or "",
  329. cell_product_type_id=cell_product_type_id or "",
  330. unit_price=unit_price,
  331. tenant_no=tenant_no,
  332. order_num=order_num or 0,
  333. remark=remark or "",
  334. create_by=user_id,
  335. create_time=datetime.now(),
  336. update_by=user_id,
  337. update_time=datetime.now(),
  338. del_flag=0,
  339. )
  340. db.add(obj)
  341. db.commit()
  342. db.refresh(obj)
  343. return obj
  344. def update_cell_product_price(
  345. db,
  346. price_id,
  347. *,
  348. project_name=None,
  349. project_code=None,
  350. cell_type_code=None,
  351. cell_type_name=None,
  352. spec_text=None,
  353. spec_id=None,
  354. cell_product_type_id=None,
  355. unit_price=None,
  356. order_num=None,
  357. remark=None,
  358. user_id=None,
  359. tenant_no=None,
  360. ):
  361. obj = get_cell_product_price(db, price_id, tenant_no)
  362. if not obj:
  363. raise ValueError("记录不存在")
  364. if project_code is not None and project_code != obj.project_code:
  365. if cell_product_price_code_exists(db, project_code, tenant_no, exclude_id=price_id):
  366. raise ValueError(f"项目编码 {project_code} 已存在")
  367. obj.project_code = project_code
  368. if project_name is not None:
  369. obj.project_name = project_name
  370. if cell_type_code is not None:
  371. obj.cell_type_code = cell_type_code
  372. if cell_type_name is not None:
  373. obj.cell_type_name = cell_type_name
  374. if spec_text is not None:
  375. obj.spec_text = spec_text
  376. if spec_id is not None:
  377. obj.spec_id = spec_id
  378. if cell_product_type_id is not None:
  379. obj.cell_product_type_id = cell_product_type_id
  380. if unit_price is not None:
  381. obj.unit_price = unit_price
  382. if order_num is not None:
  383. obj.order_num = order_num
  384. if remark is not None:
  385. obj.remark = remark
  386. obj.update_by = user_id
  387. obj.update_time = datetime.now()
  388. db.add(obj)
  389. db.commit()
  390. db.refresh(obj)
  391. return obj
  392. def soft_delete_cell_product_price(db, price_id, tenant_no=None):
  393. obj = get_cell_product_price(db, price_id, tenant_no)
  394. if not obj:
  395. raise ValueError("记录不存在")
  396. obj.del_flag = 1
  397. obj.update_time = datetime.now()
  398. db.add(obj)
  399. db.commit()
  400. # ────────────────────────────────────────────
  401. # Entity 4: BizCellProductProjectSystem 项目价格体系(主)
  402. # ────────────────────────────────────────────
  403. def _row_to_dict_project_system(row) -> Dict[str, Any]:
  404. return {
  405. "id": row.id,
  406. "systemName": row.system_name or "",
  407. "systemCode": row.system_code or "",
  408. "status": row.status or "1",
  409. "tenantNo": row.tenant_no or "",
  410. "orderNum": row.order_num if row.order_num is not None else 0,
  411. "remark": row.remark or "",
  412. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  413. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  414. }
  415. def list_project_systems(db, *, page=1, page_size=10, keyword=None, tenant_no=None, user_info=None):
  416. def _filtered():
  417. q0 = db.query(BizCellProductProjectSystem).filter(BizCellProductProjectSystem.del_flag == 0)
  418. if tenant_no:
  419. q0 = q0.filter(BizCellProductProjectSystem.tenant_no == tenant_no)
  420. kw = (keyword or "").strip()
  421. if kw:
  422. like = f"%{kw}%"
  423. q0 = q0.filter(
  424. or_(
  425. BizCellProductProjectSystem.system_code.like(like),
  426. BizCellProductProjectSystem.system_name.like(like),
  427. )
  428. )
  429. if user_info:
  430. from app.core.data_scope import apply_data_scope
  431. q0 = apply_data_scope(q0, BizCellProductProjectSystem, user_info, db, tenant_no)
  432. return q0
  433. total = _filtered().order_by(None).with_entities(func.count(BizCellProductProjectSystem.id)).scalar() or 0
  434. page = max(1, page)
  435. page_size = min(max(1, page_size), 100)
  436. rows = _filtered().order_by(BizCellProductProjectSystem.create_time.desc()).offset((page - 1) * page_size).limit(page_size).all()
  437. return [_row_to_dict_project_system(x) for x in rows], int(total)
  438. def get_project_system(db, system_id, tenant_no=None):
  439. q = db.query(BizCellProductProjectSystem).filter(
  440. and_(BizCellProductProjectSystem.id == system_id, BizCellProductProjectSystem.del_flag == 0)
  441. )
  442. if tenant_no:
  443. q = q.filter(BizCellProductProjectSystem.tenant_no == tenant_no)
  444. return q.first()
  445. def project_system_code_exists(db, code, tenant_no, exclude_id=None):
  446. q = db.query(BizCellProductProjectSystem).filter(
  447. and_(BizCellProductProjectSystem.system_code == code, BizCellProductProjectSystem.del_flag == 0)
  448. )
  449. if tenant_no:
  450. q = q.filter(BizCellProductProjectSystem.tenant_no == tenant_no)
  451. if exclude_id:
  452. q = q.filter(BizCellProductProjectSystem.id != exclude_id)
  453. return db.query(q.exists()).scalar()
  454. def create_project_system(db, *, system_name, system_code, status="1", tenant_no="zkxw", order_num=None, remark=None, user_id=None):
  455. if project_system_code_exists(db, system_code, tenant_no):
  456. raise ValueError(f"体系编码 {system_code} 已存在")
  457. obj = BizCellProductProjectSystem(
  458. id=generate_uuid(),
  459. system_name=system_name,
  460. system_code=system_code,
  461. status=status or "1",
  462. tenant_no=tenant_no,
  463. order_num=order_num or 0,
  464. remark=remark or "",
  465. create_by=user_id,
  466. create_time=datetime.now(),
  467. update_by=user_id,
  468. update_time=datetime.now(),
  469. del_flag=0,
  470. )
  471. db.add(obj)
  472. db.commit()
  473. db.refresh(obj)
  474. return obj
  475. def update_project_system(db, system_id, *, system_name=None, system_code=None, status=None, order_num=None, remark=None, user_id=None, tenant_no=None):
  476. obj = get_project_system(db, system_id, tenant_no)
  477. if not obj:
  478. raise ValueError("记录不存在")
  479. if system_code is not None and system_code != obj.system_code:
  480. if project_system_code_exists(db, system_code, tenant_no, exclude_id=system_id):
  481. raise ValueError(f"体系编码 {system_code} 已存在")
  482. obj.system_code = system_code
  483. if system_name is not None:
  484. obj.system_name = system_name
  485. if status is not None:
  486. obj.status = status
  487. if order_num is not None:
  488. obj.order_num = order_num
  489. if remark is not None:
  490. obj.remark = remark
  491. obj.update_by = user_id
  492. obj.update_time = datetime.now()
  493. db.add(obj)
  494. db.commit()
  495. db.refresh(obj)
  496. return obj
  497. def soft_delete_project_system(db, system_id, tenant_no=None):
  498. obj = get_project_system(db, system_id, tenant_no)
  499. if not obj:
  500. raise ValueError("记录不存在")
  501. obj.del_flag = 1
  502. obj.update_time = datetime.now()
  503. db.add(obj)
  504. db.commit()
  505. # ────────────────────────────────────────────
  506. # Entity 5: BizCellProductProjectSystemItem 项目价格体系明细
  507. # ────────────────────────────────────────────
  508. def _row_to_dict_project_system_item(row) -> Dict[str, Any]:
  509. return {
  510. "id": row.id,
  511. "projectSystemId": row.project_system_id or "",
  512. "sourcePriceId": row.source_price_id or "",
  513. "projectName": row.project_name or "",
  514. "projectCode": row.project_code or "",
  515. "cellTypeCode": row.cell_type_code or "",
  516. "cellTypeName": row.cell_type_name or "",
  517. "specText": row.spec_text or "",
  518. "specId": row.spec_id or "",
  519. "cellProductTypeId": row.cell_product_type_id or "",
  520. "unitPrice": float(row.unit_price) if row.unit_price is not None else None,
  521. "tenantNo": row.tenant_no or "",
  522. "orderNum": row.order_num if row.order_num is not None else 0,
  523. "remark": row.remark or "",
  524. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  525. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  526. }
  527. def list_project_system_items(db, project_system_id, tenant_no=None, user_info=None):
  528. q = db.query(BizCellProductProjectSystemItem).filter(
  529. and_(
  530. BizCellProductProjectSystemItem.project_system_id == project_system_id,
  531. BizCellProductProjectSystemItem.del_flag == 0,
  532. )
  533. )
  534. if tenant_no:
  535. q = q.filter(BizCellProductProjectSystemItem.tenant_no == tenant_no)
  536. if user_info:
  537. from app.core.data_scope import apply_data_scope
  538. q = apply_data_scope(q, BizCellProductProjectSystemItem, user_info, db, tenant_no)
  539. rows = q.order_by(
  540. BizCellProductProjectSystemItem.order_num.asc(),
  541. BizCellProductProjectSystemItem.create_time.asc(),
  542. ).all()
  543. return [_row_to_dict_project_system_item(x) for x in rows]
  544. def create_project_system_item(
  545. db,
  546. *,
  547. project_system_id,
  548. source_price_id="",
  549. project_name,
  550. project_code,
  551. cell_type_code,
  552. cell_type_name,
  553. spec_text="",
  554. spec_id="",
  555. cell_product_type_id="",
  556. unit_price,
  557. tenant_no="zkxw",
  558. order_num=None,
  559. remark=None,
  560. user_id=None,
  561. ):
  562. obj = BizCellProductProjectSystemItem(
  563. id=generate_uuid(),
  564. project_system_id=project_system_id,
  565. source_price_id=source_price_id,
  566. project_name=project_name,
  567. project_code=project_code,
  568. cell_type_code=cell_type_code,
  569. cell_type_name=cell_type_name,
  570. spec_text=spec_text or "",
  571. spec_id=spec_id or "",
  572. cell_product_type_id=cell_product_type_id or "",
  573. unit_price=unit_price,
  574. tenant_no=tenant_no,
  575. order_num=order_num or 0,
  576. remark=remark or "",
  577. create_by=user_id,
  578. create_time=datetime.now(),
  579. update_by=user_id,
  580. update_time=datetime.now(),
  581. del_flag=0,
  582. )
  583. db.add(obj)
  584. db.commit()
  585. db.refresh(obj)
  586. return obj
  587. def update_project_system_item(
  588. db,
  589. item_id,
  590. *,
  591. source_price_id=None,
  592. project_name=None,
  593. project_code=None,
  594. cell_type_code=None,
  595. cell_type_name=None,
  596. spec_text=None,
  597. spec_id=None,
  598. cell_product_type_id=None,
  599. unit_price=None,
  600. order_num=None,
  601. remark=None,
  602. user_id=None,
  603. tenant_no=None,
  604. ):
  605. q = db.query(BizCellProductProjectSystemItem).filter(
  606. and_(BizCellProductProjectSystemItem.id == item_id, BizCellProductProjectSystemItem.del_flag == 0)
  607. )
  608. if tenant_no:
  609. q = q.filter(BizCellProductProjectSystemItem.tenant_no == tenant_no)
  610. obj = q.first()
  611. if not obj:
  612. raise ValueError("记录不存在")
  613. if source_price_id is not None:
  614. obj.source_price_id = source_price_id
  615. if project_name is not None:
  616. obj.project_name = project_name
  617. if project_code is not None:
  618. obj.project_code = project_code
  619. if cell_type_code is not None:
  620. obj.cell_type_code = cell_type_code
  621. if cell_type_name is not None:
  622. obj.cell_type_name = cell_type_name
  623. if spec_text is not None:
  624. obj.spec_text = spec_text
  625. if spec_id is not None:
  626. obj.spec_id = spec_id
  627. if cell_product_type_id is not None:
  628. obj.cell_product_type_id = cell_product_type_id
  629. if unit_price is not None:
  630. obj.unit_price = unit_price
  631. if order_num is not None:
  632. obj.order_num = order_num
  633. if remark is not None:
  634. obj.remark = remark
  635. obj.update_by = user_id
  636. obj.update_time = datetime.now()
  637. db.add(obj)
  638. db.commit()
  639. db.refresh(obj)
  640. return obj
  641. def soft_delete_project_system_item(db, item_id, tenant_no=None):
  642. q = db.query(BizCellProductProjectSystemItem).filter(
  643. and_(BizCellProductProjectSystemItem.id == item_id, BizCellProductProjectSystemItem.del_flag == 0)
  644. )
  645. if tenant_no:
  646. q = q.filter(BizCellProductProjectSystemItem.tenant_no == tenant_no)
  647. obj = q.first()
  648. if not obj:
  649. raise ValueError("记录不存在")
  650. obj.del_flag = 1
  651. obj.update_time = datetime.now()
  652. db.add(obj)
  653. db.commit()
  654. # ────────────────────────────────────────────
  655. # Entity 6: BizChannel 渠道信息
  656. # ────────────────────────────────────────────
  657. def _row_to_dict_channel(
  658. row,
  659. db: Optional[Session] = None,
  660. tenant_no: Optional[str] = None,
  661. ) -> Dict[str, Any]:
  662. tn = tenant_no or row.tenant_no
  663. return {
  664. "id": row.id,
  665. "channelName": row.channel_name or "",
  666. "creditCode": row.credit_code or "",
  667. "channelType": row.channel_type or "",
  668. "contactName": row.contact_name or "",
  669. "contactPhone": row.contact_phone or "",
  670. "contactEmail": row.contact_email or "",
  671. "protocolAttachments": row.protocol_attachments or "",
  672. "attachmentFiles": urls_to_attachment_list(row.protocol_attachments, db, tn) if db else [],
  673. "tenantNo": row.tenant_no or "",
  674. "orderNum": row.order_num if row.order_num is not None else 0,
  675. "remark": row.remark or "",
  676. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  677. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  678. }
  679. def list_channels(db, *, page=1, page_size=10, keyword=None, tenant_no=None, user_info=None):
  680. def _filtered():
  681. q0 = db.query(BizChannel).filter(BizChannel.del_flag == 0)
  682. if tenant_no:
  683. q0 = q0.filter(BizChannel.tenant_no == tenant_no)
  684. kw = (keyword or "").strip()
  685. if kw:
  686. like = f"%{kw}%"
  687. q0 = q0.filter(
  688. or_(
  689. BizChannel.channel_name.like(like),
  690. BizChannel.credit_code.like(like),
  691. BizChannel.contact_name.like(like),
  692. )
  693. )
  694. if user_info:
  695. from app.core.data_scope import apply_data_scope
  696. q0 = apply_data_scope(q0, BizChannel, user_info, db, tenant_no)
  697. return q0
  698. total = _filtered().order_by(None).with_entities(func.count(BizChannel.id)).scalar() or 0
  699. page = max(1, page)
  700. page_size = min(max(1, page_size), 100)
  701. rows = _filtered().order_by(BizChannel.create_time.desc()).offset((page - 1) * page_size).limit(page_size).all()
  702. return [_row_to_dict_channel(x, db, tenant_no) for x in rows], int(total)
  703. def get_channel(db, channel_id, tenant_no=None):
  704. q = db.query(BizChannel).filter(and_(BizChannel.id == channel_id, BizChannel.del_flag == 0))
  705. if tenant_no:
  706. q = q.filter(BizChannel.tenant_no == tenant_no)
  707. return q.first()
  708. def create_channel(
  709. db,
  710. *,
  711. channel_name,
  712. credit_code=None,
  713. channel_type=None,
  714. contact_name=None,
  715. contact_phone=None,
  716. contact_email=None,
  717. protocol_attachments=None,
  718. tenant_no="zkxw",
  719. order_num=None,
  720. remark=None,
  721. user_id=None,
  722. ):
  723. obj = BizChannel(
  724. id=generate_uuid(),
  725. channel_name=channel_name,
  726. credit_code=credit_code or "",
  727. channel_type=channel_type or "",
  728. contact_name=contact_name or "",
  729. contact_phone=contact_phone or "",
  730. contact_email=contact_email or "",
  731. protocol_attachments=protocol_attachments or "",
  732. tenant_no=tenant_no,
  733. order_num=order_num or 0,
  734. remark=remark or "",
  735. create_by=user_id,
  736. create_time=datetime.now(),
  737. update_by=user_id,
  738. update_time=datetime.now(),
  739. del_flag=0,
  740. )
  741. db.add(obj)
  742. db.commit()
  743. db.refresh(obj)
  744. return obj
  745. def update_channel(
  746. db,
  747. channel_id,
  748. *,
  749. channel_name=None,
  750. credit_code=None,
  751. channel_type=None,
  752. contact_name=None,
  753. contact_phone=None,
  754. contact_email=None,
  755. protocol_attachments=None,
  756. order_num=None,
  757. remark=None,
  758. user_id=None,
  759. tenant_no=None,
  760. ):
  761. obj = get_channel(db, channel_id, tenant_no)
  762. if not obj:
  763. raise ValueError("记录不存在")
  764. if channel_name is not None:
  765. obj.channel_name = channel_name
  766. if credit_code is not None:
  767. obj.credit_code = credit_code
  768. if channel_type is not None:
  769. obj.channel_type = channel_type
  770. if contact_name is not None:
  771. obj.contact_name = contact_name
  772. if contact_phone is not None:
  773. obj.contact_phone = contact_phone
  774. if contact_email is not None:
  775. obj.contact_email = contact_email
  776. if protocol_attachments is not None:
  777. obj.protocol_attachments = protocol_attachments
  778. if order_num is not None:
  779. obj.order_num = order_num
  780. if remark is not None:
  781. obj.remark = remark
  782. obj.update_by = user_id
  783. obj.update_time = datetime.now()
  784. db.add(obj)
  785. db.commit()
  786. db.refresh(obj)
  787. return obj
  788. def soft_delete_channel(db, channel_id, tenant_no=None):
  789. obj = get_channel(db, channel_id, tenant_no)
  790. if not obj:
  791. raise ValueError("记录不存在")
  792. obj.del_flag = 1
  793. obj.update_time = datetime.now()
  794. db.add(obj)
  795. db.commit()
  796. # ────────────────────────────────────────────
  797. # Entity 7: BizChannelInvoice 渠道开票信息
  798. # ────────────────────────────────────────────
  799. def _row_to_dict_channel_invoice(row) -> Dict[str, Any]:
  800. return {
  801. "id": row.id,
  802. "channelId": row.channel_id or "",
  803. "companyName": row.company_name or "",
  804. "taxNo": row.tax_no or "",
  805. "companyAddress": row.company_address or "",
  806. "companyPhone": row.company_phone or "",
  807. "bankName": row.bank_name or "",
  808. "bankAccount": row.bank_account or "",
  809. "tenantNo": row.tenant_no or "",
  810. "orderNum": row.order_num if row.order_num is not None else 0,
  811. "remark": row.remark or "",
  812. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  813. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  814. }
  815. def list_channel_invoices(db, channel_id, tenant_no=None, user_info=None):
  816. q = db.query(BizChannelInvoice).filter(
  817. and_(
  818. BizChannelInvoice.channel_id == channel_id,
  819. BizChannelInvoice.del_flag == 0,
  820. )
  821. )
  822. if tenant_no:
  823. q = q.filter(BizChannelInvoice.tenant_no == tenant_no)
  824. if user_info:
  825. from app.core.data_scope import apply_data_scope
  826. q = apply_data_scope(q, BizChannelInvoice, user_info, db, tenant_no)
  827. rows = q.order_by(BizChannelInvoice.create_time.desc()).all()
  828. return [_row_to_dict_channel_invoice(x) for x in rows]
  829. def create_channel_invoice(
  830. db,
  831. *,
  832. channel_id,
  833. company_name,
  834. tax_no=None,
  835. company_address=None,
  836. company_phone=None,
  837. bank_name=None,
  838. bank_account=None,
  839. tenant_no="zkxw",
  840. order_num=None,
  841. remark=None,
  842. user_id=None,
  843. ):
  844. obj = BizChannelInvoice(
  845. id=generate_uuid(),
  846. channel_id=channel_id,
  847. company_name=company_name,
  848. tax_no=tax_no or "",
  849. company_address=company_address or "",
  850. company_phone=company_phone or "",
  851. bank_name=bank_name or "",
  852. bank_account=bank_account or "",
  853. tenant_no=tenant_no,
  854. order_num=order_num or 0,
  855. remark=remark or "",
  856. create_by=user_id,
  857. create_time=datetime.now(),
  858. update_by=user_id,
  859. update_time=datetime.now(),
  860. del_flag=0,
  861. )
  862. db.add(obj)
  863. db.commit()
  864. db.refresh(obj)
  865. return obj
  866. def update_channel_invoice(
  867. db,
  868. invoice_id,
  869. *,
  870. company_name=None,
  871. tax_no=None,
  872. company_address=None,
  873. company_phone=None,
  874. bank_name=None,
  875. bank_account=None,
  876. order_num=None,
  877. remark=None,
  878. user_id=None,
  879. tenant_no=None,
  880. ):
  881. q = db.query(BizChannelInvoice).filter(
  882. and_(BizChannelInvoice.id == invoice_id, BizChannelInvoice.del_flag == 0)
  883. )
  884. if tenant_no:
  885. q = q.filter(BizChannelInvoice.tenant_no == tenant_no)
  886. obj = q.first()
  887. if not obj:
  888. raise ValueError("记录不存在")
  889. if company_name is not None:
  890. obj.company_name = company_name
  891. if tax_no is not None:
  892. obj.tax_no = tax_no
  893. if company_address is not None:
  894. obj.company_address = company_address
  895. if company_phone is not None:
  896. obj.company_phone = company_phone
  897. if bank_name is not None:
  898. obj.bank_name = bank_name
  899. if bank_account is not None:
  900. obj.bank_account = bank_account
  901. if order_num is not None:
  902. obj.order_num = order_num
  903. if remark is not None:
  904. obj.remark = remark
  905. obj.update_by = user_id
  906. obj.update_time = datetime.now()
  907. db.add(obj)
  908. db.commit()
  909. db.refresh(obj)
  910. return obj
  911. def soft_delete_channel_invoice(db, invoice_id, tenant_no=None):
  912. q = db.query(BizChannelInvoice).filter(
  913. and_(BizChannelInvoice.id == invoice_id, BizChannelInvoice.del_flag == 0)
  914. )
  915. if tenant_no:
  916. q = q.filter(BizChannelInvoice.tenant_no == tenant_no)
  917. obj = q.first()
  918. if not obj:
  919. raise ValueError("记录不存在")
  920. obj.del_flag = 1
  921. obj.update_time = datetime.now()
  922. db.add(obj)
  923. db.commit()
  924. # ────────────────────────────────────────────
  925. # Entity 8: BizChannelMailAddress 渠道邮寄地址
  926. # ────────────────────────────────────────────
  927. def _row_to_dict_channel_mail_address(row) -> Dict[str, Any]:
  928. return {
  929. "id": row.id,
  930. "channelId": row.channel_id or "",
  931. "mailAddress": row.mail_address or "",
  932. "mailContact": row.mail_contact or "",
  933. "mailPhone": row.mail_phone or "",
  934. "tenantNo": row.tenant_no or "",
  935. "orderNum": row.order_num if row.order_num is not None else 0,
  936. "remark": row.remark or "",
  937. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  938. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  939. }
  940. def list_channel_mail_addresses(db, channel_id, tenant_no=None, user_info=None):
  941. q = db.query(BizChannelMailAddress).filter(
  942. and_(
  943. BizChannelMailAddress.channel_id == channel_id,
  944. BizChannelMailAddress.del_flag == 0,
  945. )
  946. )
  947. if tenant_no:
  948. q = q.filter(BizChannelMailAddress.tenant_no == tenant_no)
  949. if user_info:
  950. from app.core.data_scope import apply_data_scope
  951. q = apply_data_scope(q, BizChannelMailAddress, user_info, db, tenant_no)
  952. rows = q.order_by(BizChannelMailAddress.create_time.desc()).all()
  953. return [_row_to_dict_channel_mail_address(x) for x in rows]
  954. def create_channel_mail_address(
  955. db,
  956. *,
  957. channel_id,
  958. mail_address,
  959. mail_contact=None,
  960. mail_phone=None,
  961. tenant_no="zkxw",
  962. order_num=None,
  963. remark=None,
  964. user_id=None,
  965. ):
  966. obj = BizChannelMailAddress(
  967. id=generate_uuid(),
  968. channel_id=channel_id,
  969. mail_address=mail_address,
  970. mail_contact=mail_contact or "",
  971. mail_phone=mail_phone or "",
  972. tenant_no=tenant_no,
  973. order_num=order_num or 0,
  974. remark=remark or "",
  975. create_by=user_id,
  976. create_time=datetime.now(),
  977. update_by=user_id,
  978. update_time=datetime.now(),
  979. del_flag=0,
  980. )
  981. db.add(obj)
  982. db.commit()
  983. db.refresh(obj)
  984. return obj
  985. def update_channel_mail_address(
  986. db,
  987. address_id,
  988. *,
  989. mail_address=None,
  990. mail_contact=None,
  991. mail_phone=None,
  992. order_num=None,
  993. remark=None,
  994. user_id=None,
  995. tenant_no=None,
  996. ):
  997. q = db.query(BizChannelMailAddress).filter(
  998. and_(BizChannelMailAddress.id == address_id, BizChannelMailAddress.del_flag == 0)
  999. )
  1000. if tenant_no:
  1001. q = q.filter(BizChannelMailAddress.tenant_no == tenant_no)
  1002. obj = q.first()
  1003. if not obj:
  1004. raise ValueError("记录不存在")
  1005. if mail_address is not None:
  1006. obj.mail_address = mail_address
  1007. if mail_contact is not None:
  1008. obj.mail_contact = mail_contact
  1009. if mail_phone is not None:
  1010. obj.mail_phone = mail_phone
  1011. if order_num is not None:
  1012. obj.order_num = order_num
  1013. if remark is not None:
  1014. obj.remark = remark
  1015. obj.update_by = user_id
  1016. obj.update_time = datetime.now()
  1017. db.add(obj)
  1018. db.commit()
  1019. db.refresh(obj)
  1020. return obj
  1021. def soft_delete_channel_mail_address(db, address_id, tenant_no=None):
  1022. q = db.query(BizChannelMailAddress).filter(
  1023. and_(BizChannelMailAddress.id == address_id, BizChannelMailAddress.del_flag == 0)
  1024. )
  1025. if tenant_no:
  1026. q = q.filter(BizChannelMailAddress.tenant_no == tenant_no)
  1027. obj = q.first()
  1028. if not obj:
  1029. raise ValueError("记录不存在")
  1030. obj.del_flag = 1
  1031. obj.update_time = datetime.now()
  1032. db.add(obj)
  1033. db.commit()
  1034. # ────────────────────────────────────────────
  1035. # Entity 9: BizChannelAgreement 渠道协议
  1036. # ────────────────────────────────────────────
  1037. def _row_to_dict_channel_agreement(row) -> Dict[str, Any]:
  1038. def _fmt_date(v):
  1039. """CHAR(8) YYYYMMDD → YYYY-MM-DD"""
  1040. if not v or len(v) != 8:
  1041. return v or ""
  1042. return f"{v[:4]}-{v[4:6]}-{v[6:8]}"
  1043. return {
  1044. "id": row.id,
  1045. "channelId": row.channel_id or "",
  1046. "agreementType": row.agreement_type or "",
  1047. "businessType": row.business_type or "",
  1048. "settleMode": row.settle_mode or "",
  1049. "creditLimitAmount": float(row.credit_limit_amount) if row.credit_limit_amount is not None else None,
  1050. "effectiveDate": _fmt_date(row.effective_date),
  1051. "expireDate": _fmt_date(row.expire_date),
  1052. "agreementStatus": row.agreement_status or "",
  1053. "tenantNo": row.tenant_no or "",
  1054. "orderNum": row.order_num if row.order_num is not None else 0,
  1055. "remark": row.remark or "",
  1056. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  1057. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  1058. }
  1059. def list_channel_agreements(db, channel_id, tenant_no=None, user_info=None):
  1060. q = db.query(BizChannelAgreement).filter(
  1061. and_(
  1062. BizChannelAgreement.channel_id == channel_id,
  1063. BizChannelAgreement.del_flag == 0,
  1064. )
  1065. )
  1066. if tenant_no:
  1067. q = q.filter(BizChannelAgreement.tenant_no == tenant_no)
  1068. if user_info:
  1069. from app.core.data_scope import apply_data_scope
  1070. q = apply_data_scope(q, BizChannelAgreement, user_info, db, tenant_no)
  1071. rows = q.order_by(BizChannelAgreement.create_time.desc()).all()
  1072. return [_row_to_dict_channel_agreement(x) for x in rows]
  1073. def create_channel_agreement(
  1074. db,
  1075. *,
  1076. channel_id,
  1077. agreement_type,
  1078. business_type=None,
  1079. settle_mode=None,
  1080. credit_limit_amount=None,
  1081. effective_date=None,
  1082. expire_date=None,
  1083. agreement_status="dsx",
  1084. tenant_no="zkxw",
  1085. order_num=None,
  1086. remark=None,
  1087. user_id=None,
  1088. ):
  1089. _ed = (effective_date or "").replace("-", "") if effective_date else ""
  1090. _xd = (expire_date or "").replace("-", "") if expire_date else ""
  1091. obj = BizChannelAgreement(
  1092. id=generate_uuid(),
  1093. channel_id=channel_id,
  1094. agreement_type=agreement_type,
  1095. business_type=business_type or "",
  1096. settle_mode=settle_mode or "",
  1097. credit_limit_amount=credit_limit_amount,
  1098. effective_date=_ed,
  1099. expire_date=_xd,
  1100. agreement_status=agreement_status or "dsx",
  1101. tenant_no=tenant_no,
  1102. order_num=order_num or 0,
  1103. remark=remark or "",
  1104. create_by=user_id,
  1105. create_time=datetime.now(),
  1106. update_by=user_id,
  1107. update_time=datetime.now(),
  1108. del_flag=0,
  1109. )
  1110. db.add(obj)
  1111. db.commit()
  1112. db.refresh(obj)
  1113. return obj
  1114. def update_channel_agreement(
  1115. db,
  1116. agreement_id,
  1117. *,
  1118. agreement_type=None,
  1119. business_type=None,
  1120. settle_mode=None,
  1121. credit_limit_amount=None,
  1122. effective_date=None,
  1123. expire_date=None,
  1124. order_num=None,
  1125. remark=None,
  1126. user_id=None,
  1127. tenant_no=None,
  1128. ):
  1129. q = db.query(BizChannelAgreement).filter(
  1130. and_(BizChannelAgreement.id == agreement_id, BizChannelAgreement.del_flag == 0)
  1131. )
  1132. if tenant_no:
  1133. q = q.filter(BizChannelAgreement.tenant_no == tenant_no)
  1134. obj = q.first()
  1135. if not obj:
  1136. raise ValueError("记录不存在")
  1137. if agreement_type is not None:
  1138. obj.agreement_type = agreement_type
  1139. if business_type is not None:
  1140. obj.business_type = business_type
  1141. if settle_mode is not None:
  1142. obj.settle_mode = settle_mode
  1143. if credit_limit_amount is not None:
  1144. obj.credit_limit_amount = credit_limit_amount
  1145. if effective_date is not None:
  1146. obj.effective_date = effective_date.replace("-", "") if effective_date else ""
  1147. if expire_date is not None:
  1148. obj.expire_date = expire_date.replace("-", "") if expire_date else ""
  1149. if order_num is not None:
  1150. obj.order_num = order_num
  1151. if remark is not None:
  1152. obj.remark = remark
  1153. obj.update_by = user_id
  1154. obj.update_time = datetime.now()
  1155. db.add(obj)
  1156. db.commit()
  1157. db.refresh(obj)
  1158. return obj
  1159. def soft_delete_channel_agreement(db, agreement_id, tenant_no=None):
  1160. q = db.query(BizChannelAgreement).filter(
  1161. and_(BizChannelAgreement.id == agreement_id, BizChannelAgreement.del_flag == 0)
  1162. )
  1163. if tenant_no:
  1164. q = q.filter(BizChannelAgreement.tenant_no == tenant_no)
  1165. obj = q.first()
  1166. if not obj:
  1167. raise ValueError("记录不存在")
  1168. obj.del_flag = 1
  1169. obj.update_time = datetime.now()
  1170. db.add(obj)
  1171. db.commit()
  1172. def activate_channel_agreement(db, agreement_id, tenant_no=None, user_id=None):
  1173. """生效协议:仅待生效(dsx)可变为生效中(sxz)"""
  1174. q = db.query(BizChannelAgreement).filter(
  1175. and_(BizChannelAgreement.id == agreement_id, BizChannelAgreement.del_flag == 0)
  1176. )
  1177. if tenant_no:
  1178. q = q.filter(BizChannelAgreement.tenant_no == tenant_no)
  1179. obj = q.first()
  1180. if not obj:
  1181. raise ValueError("记录不存在")
  1182. if (obj.agreement_status or "").strip() != "dsx":
  1183. raise ValueError("仅待生效状态的协议可执行生效操作")
  1184. obj.agreement_status = "sxz"
  1185. obj.update_by = user_id
  1186. obj.update_time = datetime.now()
  1187. db.add(obj)
  1188. db.commit()
  1189. db.refresh(obj)
  1190. return obj
  1191. def void_channel_agreement(db, agreement_id, tenant_no=None, user_id=None):
  1192. """作废协议:设置 agreement_status='yzf'"""
  1193. q = db.query(BizChannelAgreement).filter(
  1194. and_(BizChannelAgreement.id == agreement_id, BizChannelAgreement.del_flag == 0)
  1195. )
  1196. if tenant_no:
  1197. q = q.filter(BizChannelAgreement.tenant_no == tenant_no)
  1198. obj = q.first()
  1199. if not obj:
  1200. raise ValueError("记录不存在")
  1201. obj.agreement_status = "yzf"
  1202. obj.update_by = user_id
  1203. obj.update_time = datetime.now()
  1204. db.add(obj)
  1205. db.commit()
  1206. db.refresh(obj)
  1207. return obj
  1208. # ────────────────────────────────────────────
  1209. # Entity 10: BizChannelAgreementProductPrice 渠道协议产品价格
  1210. # ────────────────────────────────────────────
  1211. def _row_to_dict_channel_agreement_price(row) -> Dict[str, Any]:
  1212. return {
  1213. "id": row.id,
  1214. "channelAgreementId": row.channel_agreement_id or "",
  1215. "projectName": row.project_name or "",
  1216. "projectCode": row.project_code or "",
  1217. "cellTypeCode": row.cell_type_code or "",
  1218. "cellTypeName": row.cell_type_name or "",
  1219. "specText": row.spec_text or "",
  1220. "originalPrice": float(row.original_price) if row.original_price is not None else None,
  1221. "agreementPrice": float(row.agreement_price) if row.agreement_price is not None else None,
  1222. "tenantNo": row.tenant_no or "",
  1223. "orderNum": row.order_num if row.order_num is not None else 0,
  1224. "remark": row.remark or "",
  1225. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  1226. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  1227. }
  1228. def _channel_agreement_price_item_key(
  1229. project_code=None,
  1230. project_name=None,
  1231. cell_type_code=None,
  1232. cell_type_name=None,
  1233. spec_text=None,
  1234. ) -> str:
  1235. code = (project_code or project_name or "").strip().lower()
  1236. cell = (cell_type_code or cell_type_name or "").strip().lower()
  1237. spec = (spec_text or "").strip().lower()
  1238. return f"{code}|{cell}|{spec}"
  1239. def _channel_agreement_price_exists(
  1240. db,
  1241. channel_agreement_id,
  1242. *,
  1243. project_code=None,
  1244. project_name=None,
  1245. cell_type_code=None,
  1246. cell_type_name=None,
  1247. spec_text=None,
  1248. tenant_no=None,
  1249. exclude_price_id=None,
  1250. ):
  1251. target_key = _channel_agreement_price_item_key(
  1252. project_code=project_code,
  1253. project_name=project_name,
  1254. cell_type_code=cell_type_code,
  1255. cell_type_name=cell_type_name,
  1256. spec_text=spec_text,
  1257. )
  1258. q = db.query(BizChannelAgreementProductPrice).filter(
  1259. and_(
  1260. BizChannelAgreementProductPrice.channel_agreement_id == channel_agreement_id,
  1261. BizChannelAgreementProductPrice.del_flag == 0,
  1262. )
  1263. )
  1264. if tenant_no:
  1265. q = q.filter(BizChannelAgreementProductPrice.tenant_no == tenant_no)
  1266. if exclude_price_id:
  1267. q = q.filter(BizChannelAgreementProductPrice.id != exclude_price_id)
  1268. for row in q.all():
  1269. if _channel_agreement_price_item_key(
  1270. project_code=row.project_code,
  1271. project_name=row.project_name,
  1272. cell_type_code=row.cell_type_code,
  1273. cell_type_name=row.cell_type_name,
  1274. spec_text=row.spec_text,
  1275. ) == target_key:
  1276. return True
  1277. return False
  1278. def list_channel_agreement_prices(db, channel_agreement_id, tenant_no=None, user_info=None):
  1279. q = db.query(BizChannelAgreementProductPrice).filter(
  1280. and_(
  1281. BizChannelAgreementProductPrice.channel_agreement_id == channel_agreement_id,
  1282. BizChannelAgreementProductPrice.del_flag == 0,
  1283. )
  1284. )
  1285. if tenant_no:
  1286. q = q.filter(BizChannelAgreementProductPrice.tenant_no == tenant_no)
  1287. if user_info:
  1288. from app.core.data_scope import apply_data_scope
  1289. q = apply_data_scope(q, BizChannelAgreementProductPrice, user_info, db, tenant_no)
  1290. rows = q.order_by(BizChannelAgreementProductPrice.create_time.desc()).all()
  1291. return [_row_to_dict_channel_agreement_price(x) for x in rows]
  1292. def create_channel_agreement_price(
  1293. db,
  1294. *,
  1295. channel_agreement_id,
  1296. project_name=None,
  1297. project_code=None,
  1298. cell_type_code=None,
  1299. cell_type_name=None,
  1300. spec_text=None,
  1301. original_price=None,
  1302. agreement_price=None,
  1303. tenant_no="zkxw",
  1304. order_num=None,
  1305. remark=None,
  1306. user_id=None,
  1307. ):
  1308. if _channel_agreement_price_exists(
  1309. db,
  1310. channel_agreement_id,
  1311. project_code=project_code,
  1312. project_name=project_name,
  1313. cell_type_code=cell_type_code,
  1314. cell_type_name=cell_type_name,
  1315. spec_text=spec_text,
  1316. tenant_no=tenant_no,
  1317. ):
  1318. raise ValueError("该项目信息已存在,不可重复添加")
  1319. obj = BizChannelAgreementProductPrice(
  1320. id=generate_uuid(),
  1321. channel_agreement_id=channel_agreement_id,
  1322. project_name=project_name or "",
  1323. project_code=project_code or "",
  1324. cell_type_code=cell_type_code or "",
  1325. cell_type_name=cell_type_name or "",
  1326. spec_text=spec_text or "",
  1327. original_price=original_price,
  1328. agreement_price=agreement_price,
  1329. tenant_no=tenant_no,
  1330. order_num=order_num or 0,
  1331. remark=remark or "",
  1332. create_by=user_id,
  1333. create_time=datetime.now(),
  1334. update_by=user_id,
  1335. update_time=datetime.now(),
  1336. del_flag=0,
  1337. )
  1338. db.add(obj)
  1339. db.commit()
  1340. db.refresh(obj)
  1341. return obj
  1342. def update_channel_agreement_price(
  1343. db,
  1344. price_id,
  1345. *,
  1346. project_name=None,
  1347. project_code=None,
  1348. cell_type_code=None,
  1349. cell_type_name=None,
  1350. spec_text=None,
  1351. original_price=None,
  1352. agreement_price=None,
  1353. order_num=None,
  1354. remark=None,
  1355. user_id=None,
  1356. tenant_no=None,
  1357. ):
  1358. q = db.query(BizChannelAgreementProductPrice).filter(
  1359. and_(BizChannelAgreementProductPrice.id == price_id, BizChannelAgreementProductPrice.del_flag == 0)
  1360. )
  1361. if tenant_no:
  1362. q = q.filter(BizChannelAgreementProductPrice.tenant_no == tenant_no)
  1363. obj = q.first()
  1364. if not obj:
  1365. raise ValueError("记录不存在")
  1366. next_project_name = project_name if project_name is not None else obj.project_name
  1367. next_project_code = project_code if project_code is not None else obj.project_code
  1368. next_cell_type_code = cell_type_code if cell_type_code is not None else obj.cell_type_code
  1369. next_cell_type_name = cell_type_name if cell_type_name is not None else obj.cell_type_name
  1370. next_spec_text = spec_text if spec_text is not None else obj.spec_text
  1371. if _channel_agreement_price_exists(
  1372. db,
  1373. obj.channel_agreement_id,
  1374. project_code=next_project_code,
  1375. project_name=next_project_name,
  1376. cell_type_code=next_cell_type_code,
  1377. cell_type_name=next_cell_type_name,
  1378. spec_text=next_spec_text,
  1379. tenant_no=tenant_no,
  1380. exclude_price_id=price_id,
  1381. ):
  1382. raise ValueError("该项目信息已存在,不可重复添加")
  1383. if project_name is not None:
  1384. obj.project_name = project_name
  1385. if project_code is not None:
  1386. obj.project_code = project_code
  1387. if cell_type_code is not None:
  1388. obj.cell_type_code = cell_type_code
  1389. if cell_type_name is not None:
  1390. obj.cell_type_name = cell_type_name
  1391. if spec_text is not None:
  1392. obj.spec_text = spec_text
  1393. if original_price is not None:
  1394. obj.original_price = original_price
  1395. if agreement_price is not None:
  1396. obj.agreement_price = agreement_price
  1397. if order_num is not None:
  1398. obj.order_num = order_num
  1399. if remark is not None:
  1400. obj.remark = remark
  1401. obj.update_by = user_id
  1402. obj.update_time = datetime.now()
  1403. db.add(obj)
  1404. db.commit()
  1405. db.refresh(obj)
  1406. return obj
  1407. def soft_delete_channel_agreement_price(db, price_id, tenant_no=None):
  1408. q = db.query(BizChannelAgreementProductPrice).filter(
  1409. and_(BizChannelAgreementProductPrice.id == price_id, BizChannelAgreementProductPrice.del_flag == 0)
  1410. )
  1411. if tenant_no:
  1412. q = q.filter(BizChannelAgreementProductPrice.tenant_no == tenant_no)
  1413. obj = q.first()
  1414. if not obj:
  1415. raise ValueError("记录不存在")
  1416. obj.del_flag = 1
  1417. obj.update_time = datetime.now()
  1418. db.add(obj)
  1419. db.commit()
  1420. # ────────────────────────────────────────────
  1421. # Entity 11: BizCheckupInstitution 体检机构
  1422. # ────────────────────────────────────────────
  1423. def _row_to_dict_checkup_institution(row) -> Dict[str, Any]:
  1424. return {
  1425. "id": row.id,
  1426. "institutionName": row.institution_name or "",
  1427. "institutionAddress": row.institution_address or "",
  1428. "contactName": row.contact_name or "",
  1429. "contactPhone": row.contact_phone or "",
  1430. "status": row.status or "1",
  1431. "tenantNo": row.tenant_no or "",
  1432. "orderNum": row.order_num if row.order_num is not None else 0,
  1433. "remark": row.remark or "",
  1434. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  1435. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  1436. }
  1437. def list_checkup_institutions(db, *, page=1, page_size=10, keyword=None, status=None, tenant_no=None, user_info=None):
  1438. def _filtered():
  1439. q0 = db.query(BizCheckupInstitution).filter(BizCheckupInstitution.del_flag == 0)
  1440. if tenant_no:
  1441. q0 = q0.filter(BizCheckupInstitution.tenant_no == tenant_no)
  1442. kw = (keyword or "").strip()
  1443. if kw:
  1444. like = f"%{kw}%"
  1445. q0 = q0.filter(
  1446. or_(
  1447. BizCheckupInstitution.institution_name.like(like),
  1448. BizCheckupInstitution.contact_name.like(like),
  1449. )
  1450. )
  1451. st = (status or "").strip()
  1452. if st in ("0", "1"):
  1453. q0 = q0.filter(BizCheckupInstitution.status == st)
  1454. if user_info:
  1455. from app.core.data_scope import apply_data_scope
  1456. q0 = apply_data_scope(q0, BizCheckupInstitution, user_info, db, tenant_no)
  1457. return q0
  1458. total = _filtered().order_by(None).with_entities(func.count(BizCheckupInstitution.id)).scalar() or 0
  1459. page = max(1, page)
  1460. page_size = min(max(1, page_size), 100)
  1461. rows = _filtered().order_by(BizCheckupInstitution.create_time.desc()).offset((page - 1) * page_size).limit(page_size).all()
  1462. return [_row_to_dict_checkup_institution(x) for x in rows], int(total)
  1463. def get_checkup_institution(db, institution_id, tenant_no=None):
  1464. q = db.query(BizCheckupInstitution).filter(
  1465. and_(BizCheckupInstitution.id == institution_id, BizCheckupInstitution.del_flag == 0)
  1466. )
  1467. if tenant_no:
  1468. q = q.filter(BizCheckupInstitution.tenant_no == tenant_no)
  1469. return q.first()
  1470. def create_checkup_institution(
  1471. db,
  1472. *,
  1473. institution_name,
  1474. institution_address=None,
  1475. contact_name=None,
  1476. contact_phone=None,
  1477. status="1",
  1478. tenant_no="zkxw",
  1479. order_num=None,
  1480. remark=None,
  1481. user_id=None,
  1482. ):
  1483. obj = BizCheckupInstitution(
  1484. id=generate_uuid(),
  1485. institution_name=institution_name,
  1486. institution_address=institution_address or "",
  1487. contact_name=contact_name or "",
  1488. contact_phone=contact_phone or "",
  1489. status=status or "1",
  1490. tenant_no=tenant_no,
  1491. order_num=order_num or 0,
  1492. remark=remark or "",
  1493. create_by=user_id,
  1494. create_time=datetime.now(),
  1495. update_by=user_id,
  1496. update_time=datetime.now(),
  1497. del_flag=0,
  1498. )
  1499. db.add(obj)
  1500. db.commit()
  1501. db.refresh(obj)
  1502. return obj
  1503. def update_checkup_institution(
  1504. db,
  1505. institution_id,
  1506. *,
  1507. institution_name=None,
  1508. institution_address=None,
  1509. contact_name=None,
  1510. contact_phone=None,
  1511. status=None,
  1512. order_num=None,
  1513. remark=None,
  1514. user_id=None,
  1515. tenant_no=None,
  1516. ):
  1517. obj = get_checkup_institution(db, institution_id, tenant_no)
  1518. if not obj:
  1519. raise ValueError("记录不存在")
  1520. if institution_name is not None:
  1521. obj.institution_name = institution_name
  1522. if institution_address is not None:
  1523. obj.institution_address = institution_address
  1524. if contact_name is not None:
  1525. obj.contact_name = contact_name
  1526. if contact_phone is not None:
  1527. obj.contact_phone = contact_phone
  1528. if status is not None:
  1529. obj.status = status
  1530. if order_num is not None:
  1531. obj.order_num = order_num
  1532. if remark is not None:
  1533. obj.remark = remark
  1534. obj.update_by = user_id
  1535. obj.update_time = datetime.now()
  1536. db.add(obj)
  1537. db.commit()
  1538. db.refresh(obj)
  1539. return obj
  1540. def soft_delete_checkup_institution(db, institution_id, tenant_no=None):
  1541. obj = get_checkup_institution(db, institution_id, tenant_no)
  1542. if not obj:
  1543. raise ValueError("记录不存在")
  1544. obj.del_flag = 1
  1545. obj.update_time = datetime.now()
  1546. db.add(obj)
  1547. db.commit()
  1548. # ────────────────────────────────────────────
  1549. # Entity 12: BizCheckupInstitutionInvoice 体检机构开票信息
  1550. # ────────────────────────────────────────────
  1551. def _row_to_dict_checkup_institution_invoice(row) -> Dict[str, Any]:
  1552. return {
  1553. "id": row.id,
  1554. "checkupInstitutionId": row.checkup_institution_id or "",
  1555. "companyName": row.company_name or "",
  1556. "taxNo": row.tax_no or "",
  1557. "companyAddress": row.company_address or "",
  1558. "companyPhone": row.company_phone or "",
  1559. "bankName": row.bank_name or "",
  1560. "bankAccount": row.bank_account or "",
  1561. "tenantNo": row.tenant_no or "",
  1562. "orderNum": row.order_num if row.order_num is not None else 0,
  1563. "remark": row.remark or "",
  1564. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  1565. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  1566. }
  1567. def list_checkup_institution_invoices(db, checkup_institution_id, tenant_no=None, user_info=None):
  1568. q = db.query(BizCheckupInstitutionInvoice).filter(
  1569. and_(
  1570. BizCheckupInstitutionInvoice.checkup_institution_id == checkup_institution_id,
  1571. BizCheckupInstitutionInvoice.del_flag == 0,
  1572. )
  1573. )
  1574. if tenant_no:
  1575. q = q.filter(BizCheckupInstitutionInvoice.tenant_no == tenant_no)
  1576. if user_info:
  1577. from app.core.data_scope import apply_data_scope
  1578. q = apply_data_scope(q, BizCheckupInstitutionInvoice, user_info, db, tenant_no)
  1579. rows = q.order_by(BizCheckupInstitutionInvoice.create_time.desc()).all()
  1580. return [_row_to_dict_checkup_institution_invoice(x) for x in rows]
  1581. def create_checkup_institution_invoice(
  1582. db,
  1583. *,
  1584. checkup_institution_id,
  1585. company_name,
  1586. tax_no=None,
  1587. company_address=None,
  1588. company_phone=None,
  1589. bank_name=None,
  1590. bank_account=None,
  1591. tenant_no="zkxw",
  1592. order_num=None,
  1593. remark=None,
  1594. user_id=None,
  1595. ):
  1596. obj = BizCheckupInstitutionInvoice(
  1597. id=generate_uuid(),
  1598. checkup_institution_id=checkup_institution_id,
  1599. company_name=company_name,
  1600. tax_no=tax_no or "",
  1601. company_address=company_address or "",
  1602. company_phone=company_phone or "",
  1603. bank_name=bank_name or "",
  1604. bank_account=bank_account or "",
  1605. tenant_no=tenant_no,
  1606. order_num=order_num or 0,
  1607. remark=remark or "",
  1608. create_by=user_id,
  1609. create_time=datetime.now(),
  1610. update_by=user_id,
  1611. update_time=datetime.now(),
  1612. del_flag=0,
  1613. )
  1614. db.add(obj)
  1615. db.commit()
  1616. db.refresh(obj)
  1617. return obj
  1618. def update_checkup_institution_invoice(
  1619. db,
  1620. invoice_id,
  1621. *,
  1622. company_name=None,
  1623. tax_no=None,
  1624. company_address=None,
  1625. company_phone=None,
  1626. bank_name=None,
  1627. bank_account=None,
  1628. order_num=None,
  1629. remark=None,
  1630. user_id=None,
  1631. tenant_no=None,
  1632. ):
  1633. q = db.query(BizCheckupInstitutionInvoice).filter(
  1634. and_(BizCheckupInstitutionInvoice.id == invoice_id, BizCheckupInstitutionInvoice.del_flag == 0)
  1635. )
  1636. if tenant_no:
  1637. q = q.filter(BizCheckupInstitutionInvoice.tenant_no == tenant_no)
  1638. obj = q.first()
  1639. if not obj:
  1640. raise ValueError("记录不存在")
  1641. if company_name is not None:
  1642. obj.company_name = company_name
  1643. if tax_no is not None:
  1644. obj.tax_no = tax_no
  1645. if company_address is not None:
  1646. obj.company_address = company_address
  1647. if company_phone is not None:
  1648. obj.company_phone = company_phone
  1649. if bank_name is not None:
  1650. obj.bank_name = bank_name
  1651. if bank_account is not None:
  1652. obj.bank_account = bank_account
  1653. if order_num is not None:
  1654. obj.order_num = order_num
  1655. if remark is not None:
  1656. obj.remark = remark
  1657. obj.update_by = user_id
  1658. obj.update_time = datetime.now()
  1659. db.add(obj)
  1660. db.commit()
  1661. db.refresh(obj)
  1662. return obj
  1663. def soft_delete_checkup_institution_invoice(db, invoice_id, tenant_no=None):
  1664. q = db.query(BizCheckupInstitutionInvoice).filter(
  1665. and_(BizCheckupInstitutionInvoice.id == invoice_id, BizCheckupInstitutionInvoice.del_flag == 0)
  1666. )
  1667. if tenant_no:
  1668. q = q.filter(BizCheckupInstitutionInvoice.tenant_no == tenant_no)
  1669. obj = q.first()
  1670. if not obj:
  1671. raise ValueError("记录不存在")
  1672. obj.del_flag = 1
  1673. obj.update_time = datetime.now()
  1674. db.add(obj)
  1675. db.commit()
  1676. # ────────────────────────────────────────────
  1677. # Entity 13: BizCheckupInstitutionProject 体检机构项目及价格
  1678. # ────────────────────────────────────────────
  1679. def _row_to_dict_checkup_institution_project(row) -> Dict[str, Any]:
  1680. return {
  1681. "id": row.id,
  1682. "checkupInstitutionId": row.checkup_institution_id or "",
  1683. "projectName": row.project_name or "",
  1684. "projectFee": float(row.project_fee) if row.project_fee is not None else None,
  1685. "status": row.status or "1",
  1686. "tenantNo": row.tenant_no or "",
  1687. "orderNum": row.order_num if row.order_num is not None else 0,
  1688. "remark": row.remark or "",
  1689. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  1690. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  1691. }
  1692. def list_checkup_institution_projects(db, checkup_institution_id, tenant_no=None, user_info=None):
  1693. q = db.query(BizCheckupInstitutionProject).filter(
  1694. and_(
  1695. BizCheckupInstitutionProject.checkup_institution_id == checkup_institution_id,
  1696. BizCheckupInstitutionProject.del_flag == 0,
  1697. )
  1698. )
  1699. if tenant_no:
  1700. q = q.filter(BizCheckupInstitutionProject.tenant_no == tenant_no)
  1701. if user_info:
  1702. from app.core.data_scope import apply_data_scope
  1703. q = apply_data_scope(q, BizCheckupInstitutionProject, user_info, db, tenant_no)
  1704. rows = q.order_by(BizCheckupInstitutionProject.create_time.desc()).all()
  1705. return [_row_to_dict_checkup_institution_project(x) for x in rows]
  1706. def create_checkup_institution_project(
  1707. db,
  1708. *,
  1709. checkup_institution_id,
  1710. project_name,
  1711. project_fee=None,
  1712. status="1",
  1713. tenant_no="zkxw",
  1714. order_num=None,
  1715. remark=None,
  1716. user_id=None,
  1717. ):
  1718. obj = BizCheckupInstitutionProject(
  1719. id=generate_uuid(),
  1720. checkup_institution_id=checkup_institution_id,
  1721. project_name=project_name,
  1722. project_fee=project_fee,
  1723. status=status or "1",
  1724. tenant_no=tenant_no,
  1725. order_num=order_num or 0,
  1726. remark=remark or "",
  1727. create_by=user_id,
  1728. create_time=datetime.now(),
  1729. update_by=user_id,
  1730. update_time=datetime.now(),
  1731. del_flag=0,
  1732. )
  1733. db.add(obj)
  1734. db.commit()
  1735. db.refresh(obj)
  1736. return obj
  1737. def update_checkup_institution_project(
  1738. db,
  1739. project_id,
  1740. *,
  1741. project_name=None,
  1742. project_fee=None,
  1743. status=None,
  1744. order_num=None,
  1745. remark=None,
  1746. user_id=None,
  1747. tenant_no=None,
  1748. ):
  1749. q = db.query(BizCheckupInstitutionProject).filter(
  1750. and_(BizCheckupInstitutionProject.id == project_id, BizCheckupInstitutionProject.del_flag == 0)
  1751. )
  1752. if tenant_no:
  1753. q = q.filter(BizCheckupInstitutionProject.tenant_no == tenant_no)
  1754. obj = q.first()
  1755. if not obj:
  1756. raise ValueError("记录不存在")
  1757. if project_name is not None:
  1758. obj.project_name = project_name
  1759. if project_fee is not None:
  1760. obj.project_fee = project_fee
  1761. if status is not None:
  1762. obj.status = status
  1763. if order_num is not None:
  1764. obj.order_num = order_num
  1765. if remark is not None:
  1766. obj.remark = remark
  1767. obj.update_by = user_id
  1768. obj.update_time = datetime.now()
  1769. db.add(obj)
  1770. db.commit()
  1771. db.refresh(obj)
  1772. return obj
  1773. def soft_delete_checkup_institution_project(db, project_id, tenant_no=None):
  1774. q = db.query(BizCheckupInstitutionProject).filter(
  1775. and_(BizCheckupInstitutionProject.id == project_id, BizCheckupInstitutionProject.del_flag == 0)
  1776. )
  1777. if tenant_no:
  1778. q = q.filter(BizCheckupInstitutionProject.tenant_no == tenant_no)
  1779. obj = q.first()
  1780. if not obj:
  1781. raise ValueError("记录不存在")
  1782. obj.del_flag = 1
  1783. obj.update_time = datetime.now()
  1784. db.add(obj)
  1785. db.commit()
  1786. # ────────────────────────────────────────────
  1787. # Entity 14: BizCheckupInstitutionPackage 体检机构项目套餐
  1788. # ────────────────────────────────────────────
  1789. def _sum_package_item_fees(items: List[Dict[str, Any]]) -> float:
  1790. total = 0.0
  1791. for it in items:
  1792. fee = it.get("project_fee")
  1793. if fee is None and "projectFee" in it:
  1794. fee = it.get("projectFee")
  1795. total += float(fee) if fee is not None else 0.0
  1796. return total
  1797. def _row_to_dict_checkup_institution_package(
  1798. row,
  1799. *,
  1800. total_fee: Optional[float] = None,
  1801. item_count: Optional[int] = None,
  1802. ) -> Dict[str, Any]:
  1803. d = {
  1804. "id": row.id,
  1805. "checkupInstitutionId": row.checkup_institution_id or "",
  1806. "packageName": row.package_name or "",
  1807. "packageCode": row.package_code or "",
  1808. "status": row.status or "1",
  1809. "tenantNo": row.tenant_no or "",
  1810. "orderNum": row.order_num if row.order_num is not None else 0,
  1811. "remark": row.remark or "",
  1812. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  1813. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  1814. }
  1815. if total_fee is not None:
  1816. d["totalFee"] = total_fee
  1817. if item_count is not None:
  1818. d["itemCount"] = item_count
  1819. return d
  1820. def _row_to_dict_checkup_institution_package_item(
  1821. row,
  1822. *,
  1823. source_project_deleted: bool = False,
  1824. source_project_status: Optional[str] = None,
  1825. ) -> Dict[str, Any]:
  1826. return {
  1827. "id": row.id,
  1828. "packageId": row.package_id or "",
  1829. "checkupInstitutionProjectId": row.checkup_institution_project_id or "",
  1830. "projectName": row.project_name or "",
  1831. "projectFee": float(row.project_fee) if row.project_fee is not None else None,
  1832. "status": source_project_status if source_project_status is not None else "1",
  1833. "sourceProjectDeleted": source_project_deleted,
  1834. "tenantNo": row.tenant_no or "",
  1835. "orderNum": row.order_num if row.order_num is not None else 0,
  1836. "remark": row.remark or "",
  1837. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  1838. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  1839. }
  1840. def checkup_package_code_exists(db, institution_id, code, tenant_no, exclude_id=None):
  1841. q = db.query(BizCheckupInstitutionPackage).filter(
  1842. and_(
  1843. BizCheckupInstitutionPackage.checkup_institution_id == institution_id,
  1844. BizCheckupInstitutionPackage.package_code == code,
  1845. BizCheckupInstitutionPackage.del_flag == 0,
  1846. )
  1847. )
  1848. if tenant_no:
  1849. q = q.filter(BizCheckupInstitutionPackage.tenant_no == tenant_no)
  1850. if exclude_id:
  1851. q = q.filter(BizCheckupInstitutionPackage.id != exclude_id)
  1852. return db.query(q.exists()).scalar()
  1853. def _get_checkup_project_for_package(db, project_id, institution_id, tenant_no=None):
  1854. q = db.query(BizCheckupInstitutionProject).filter(
  1855. BizCheckupInstitutionProject.id == project_id,
  1856. BizCheckupInstitutionProject.checkup_institution_id == institution_id,
  1857. )
  1858. if tenant_no:
  1859. q = q.filter(BizCheckupInstitutionProject.tenant_no == tenant_no)
  1860. return q.first()
  1861. def _validate_package_items_for_save(
  1862. db,
  1863. institution_id,
  1864. items,
  1865. tenant_no=None,
  1866. existing_item_project_ids=None,
  1867. ):
  1868. if not items:
  1869. raise ValueError("套餐至少包含一个体检项目")
  1870. existing_ids = set(existing_item_project_ids or [])
  1871. seen = set()
  1872. for it in items:
  1873. pid = it.get("checkup_institution_project_id") or it.get("checkupInstitutionProjectId")
  1874. if not pid:
  1875. raise ValueError("套餐明细缺少机构项目")
  1876. if pid in seen:
  1877. raise ValueError("套餐内不能重复添加同一体检项目")
  1878. seen.add(pid)
  1879. proj = _get_checkup_project_for_package(db, pid, institution_id, tenant_no)
  1880. if not proj:
  1881. raise ValueError("存在无效的体检项目")
  1882. if pid in existing_ids:
  1883. continue
  1884. if proj.del_flag != 0:
  1885. raise ValueError("不能新添加已删除的体检项目")
  1886. if (proj.status or "1") != "1":
  1887. raise ValueError("套餐不能包含已停用的体检项目")
  1888. def _list_package_items_raw(db, package_id, tenant_no=None):
  1889. q = db.query(BizCheckupInstitutionPackageItem).filter(
  1890. and_(
  1891. BizCheckupInstitutionPackageItem.package_id == package_id,
  1892. BizCheckupInstitutionPackageItem.del_flag == 0,
  1893. )
  1894. )
  1895. if tenant_no:
  1896. q = q.filter(BizCheckupInstitutionPackageItem.tenant_no == tenant_no)
  1897. return q.order_by(BizCheckupInstitutionPackageItem.order_num.asc(), BizCheckupInstitutionPackageItem.create_time.asc()).all()
  1898. def _package_items_to_dicts(db, institution_id, item_rows, tenant_no=None):
  1899. result = []
  1900. for row in item_rows:
  1901. proj = _get_checkup_project_for_package(
  1902. db, row.checkup_institution_project_id, institution_id, tenant_no
  1903. )
  1904. deleted = proj is None or proj.del_flag != 0
  1905. proj_status = (proj.status or "1") if proj else "1"
  1906. result.append(
  1907. _row_to_dict_checkup_institution_package_item(
  1908. row,
  1909. source_project_deleted=deleted,
  1910. source_project_status=proj_status,
  1911. )
  1912. )
  1913. return result
  1914. def _insert_package_items(
  1915. db,
  1916. *,
  1917. package_id,
  1918. institution_id,
  1919. items,
  1920. tenant_no,
  1921. user_id=None,
  1922. ):
  1923. for idx, it in enumerate(items):
  1924. pid = it.get("checkup_institution_project_id") or it.get("checkupInstitutionProjectId")
  1925. proj = _get_checkup_project_for_package(db, pid, institution_id, tenant_no)
  1926. name = (it.get("project_name") or it.get("projectName") or "").strip()
  1927. if not name and proj:
  1928. name = proj.project_name or ""
  1929. fee = it.get("project_fee")
  1930. if fee is None:
  1931. fee = it.get("projectFee")
  1932. obj = BizCheckupInstitutionPackageItem(
  1933. id=generate_uuid(),
  1934. package_id=package_id,
  1935. checkup_institution_project_id=pid,
  1936. project_name=name,
  1937. project_fee=fee,
  1938. tenant_no=tenant_no,
  1939. order_num=it.get("order_num") if it.get("order_num") is not None else it.get("orderNum") or (idx + 1),
  1940. remark=it.get("remark") or "",
  1941. create_by=user_id,
  1942. create_time=datetime.now(),
  1943. update_by=user_id,
  1944. update_time=datetime.now(),
  1945. del_flag=0,
  1946. )
  1947. db.add(obj)
  1948. def list_checkup_institution_packages(db, checkup_institution_id, tenant_no=None, user_info=None):
  1949. q = db.query(BizCheckupInstitutionPackage).filter(
  1950. and_(
  1951. BizCheckupInstitutionPackage.checkup_institution_id == checkup_institution_id,
  1952. BizCheckupInstitutionPackage.del_flag == 0,
  1953. )
  1954. )
  1955. if tenant_no:
  1956. q = q.filter(BizCheckupInstitutionPackage.tenant_no == tenant_no)
  1957. if user_info:
  1958. from app.core.data_scope import apply_data_scope
  1959. q = apply_data_scope(q, BizCheckupInstitutionPackage, user_info, db, tenant_no)
  1960. rows = q.order_by(BizCheckupInstitutionPackage.create_time.desc()).all()
  1961. out = []
  1962. for row in rows:
  1963. item_rows = _list_package_items_raw(db, row.id, tenant_no)
  1964. fees = [float(x.project_fee) if x.project_fee is not None else 0.0 for x in item_rows]
  1965. out.append(
  1966. _row_to_dict_checkup_institution_package(
  1967. row,
  1968. total_fee=sum(fees),
  1969. item_count=len(item_rows),
  1970. )
  1971. )
  1972. return out
  1973. def get_checkup_institution_package(db, package_id, tenant_no=None):
  1974. q = db.query(BizCheckupInstitutionPackage).filter(
  1975. and_(BizCheckupInstitutionPackage.id == package_id, BizCheckupInstitutionPackage.del_flag == 0)
  1976. )
  1977. if tenant_no:
  1978. q = q.filter(BizCheckupInstitutionPackage.tenant_no == tenant_no)
  1979. return q.first()
  1980. def get_checkup_institution_package_detail(db, package_id, tenant_no=None):
  1981. pkg = get_checkup_institution_package(db, package_id, tenant_no)
  1982. if not pkg:
  1983. return None
  1984. item_rows = _list_package_items_raw(db, package_id, tenant_no)
  1985. items = _package_items_to_dicts(db, pkg.checkup_institution_id, item_rows, tenant_no)
  1986. fees = [float(x.project_fee) if x.project_fee is not None else 0.0 for x in item_rows]
  1987. d = _row_to_dict_checkup_institution_package(pkg, total_fee=sum(fees), item_count=len(item_rows))
  1988. d["items"] = items
  1989. return d
  1990. def create_checkup_institution_package(
  1991. db,
  1992. *,
  1993. checkup_institution_id,
  1994. package_name,
  1995. package_code,
  1996. status="1",
  1997. items,
  1998. tenant_no="zkxw",
  1999. order_num=None,
  2000. remark=None,
  2001. user_id=None,
  2002. ):
  2003. code = (package_code or "").strip()
  2004. if not code:
  2005. raise ValueError("套餐编码不能为空")
  2006. if checkup_package_code_exists(db, checkup_institution_id, code, tenant_no):
  2007. raise ValueError(f"套餐编码 {code} 已存在")
  2008. _validate_package_items_for_save(db, checkup_institution_id, items, tenant_no)
  2009. pkg = BizCheckupInstitutionPackage(
  2010. id=generate_uuid(),
  2011. checkup_institution_id=checkup_institution_id,
  2012. package_name=package_name,
  2013. package_code=code,
  2014. status=status or "1",
  2015. tenant_no=tenant_no,
  2016. order_num=order_num or 0,
  2017. remark=remark or "",
  2018. create_by=user_id,
  2019. create_time=datetime.now(),
  2020. update_by=user_id,
  2021. update_time=datetime.now(),
  2022. del_flag=0,
  2023. )
  2024. db.add(pkg)
  2025. db.flush()
  2026. _insert_package_items(
  2027. db,
  2028. package_id=pkg.id,
  2029. institution_id=checkup_institution_id,
  2030. items=items,
  2031. tenant_no=tenant_no,
  2032. user_id=user_id,
  2033. )
  2034. db.commit()
  2035. db.refresh(pkg)
  2036. return get_checkup_institution_package_detail(db, pkg.id, tenant_no)
  2037. def update_checkup_institution_package(
  2038. db,
  2039. package_id,
  2040. *,
  2041. package_name=None,
  2042. package_code=None,
  2043. status=None,
  2044. items=None,
  2045. order_num=None,
  2046. remark=None,
  2047. user_id=None,
  2048. tenant_no=None,
  2049. ):
  2050. pkg = get_checkup_institution_package(db, package_id, tenant_no)
  2051. if not pkg:
  2052. raise ValueError("记录不存在")
  2053. institution_id = pkg.checkup_institution_id
  2054. if package_code is not None:
  2055. code = package_code.strip()
  2056. if not code:
  2057. raise ValueError("套餐编码不能为空")
  2058. if code != pkg.package_code and checkup_package_code_exists(
  2059. db, institution_id, code, tenant_no, exclude_id=package_id
  2060. ):
  2061. raise ValueError(f"套餐编码 {code} 已存在")
  2062. pkg.package_code = code
  2063. if package_name is not None:
  2064. pkg.package_name = package_name
  2065. if status is not None:
  2066. pkg.status = status
  2067. if order_num is not None:
  2068. pkg.order_num = order_num
  2069. if remark is not None:
  2070. pkg.remark = remark
  2071. if items is not None:
  2072. old_rows = _list_package_items_raw(db, package_id, tenant_no)
  2073. existing_pids = {r.checkup_institution_project_id for r in old_rows}
  2074. _validate_package_items_for_save(
  2075. db,
  2076. institution_id,
  2077. items,
  2078. tenant_no,
  2079. existing_item_project_ids=existing_pids,
  2080. )
  2081. # 全量替换:物理删除旧明细,避免 uk(package_id, project_id) 与软删行冲突
  2082. q_items = db.query(BizCheckupInstitutionPackageItem).filter(
  2083. BizCheckupInstitutionPackageItem.package_id == package_id
  2084. )
  2085. if tenant_no:
  2086. q_items = q_items.filter(BizCheckupInstitutionPackageItem.tenant_no == tenant_no)
  2087. for row in q_items.all():
  2088. db.delete(row)
  2089. db.flush()
  2090. _insert_package_items(
  2091. db,
  2092. package_id=package_id,
  2093. institution_id=institution_id,
  2094. items=items,
  2095. tenant_no=tenant_no or pkg.tenant_no,
  2096. user_id=user_id,
  2097. )
  2098. pkg.update_by = user_id
  2099. pkg.update_time = datetime.now()
  2100. db.add(pkg)
  2101. db.commit()
  2102. return get_checkup_institution_package_detail(db, package_id, tenant_no)
  2103. def soft_delete_checkup_institution_package(db, package_id, tenant_no=None):
  2104. pkg = get_checkup_institution_package(db, package_id, tenant_no)
  2105. if not pkg:
  2106. raise ValueError("记录不存在")
  2107. now = datetime.now()
  2108. pkg.del_flag = 1
  2109. pkg.update_time = now
  2110. db.add(pkg)
  2111. for row in _list_package_items_raw(db, package_id, tenant_no):
  2112. row.del_flag = 1
  2113. row.update_time = now
  2114. db.add(row)
  2115. db.commit()
  2116. # ────────────────────────────────────────────
  2117. # Shared utility
  2118. # ────────────────────────────────────────────
  2119. def total_pages(total, page_size):
  2120. if total <= 0:
  2121. return 0
  2122. return int(math.ceil(total / float(page_size)))
  2123. # ────────────────────────────────────────────
  2124. # Entity 15: BizChannelAccount 渠道账户
  2125. # ────────────────────────────────────────────
  2126. def _row_to_dict_channel_account(row) -> Dict[str, Any]:
  2127. return {
  2128. "id": row.id,
  2129. "channelId": row.channel_id or "",
  2130. "balanceAmount": float(row.balance_amount) if row.balance_amount else 0,
  2131. "tenantNo": row.tenant_no or "",
  2132. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  2133. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  2134. }
  2135. def get_channel_account(db, channel_id, tenant_no=None):
  2136. """只读查询渠道账户,不加行锁"""
  2137. q = db.query(BizChannelAccount).filter(
  2138. and_(BizChannelAccount.channel_id == channel_id, BizChannelAccount.del_flag == 0)
  2139. )
  2140. if tenant_no:
  2141. q = q.filter(BizChannelAccount.tenant_no == tenant_no)
  2142. return q.first()
  2143. def _get_or_create_channel_account(db, channel_id, tenant_no, user_id):
  2144. """获取或创建渠道账户(带行锁 with_for_update)"""
  2145. q = db.query(BizChannelAccount).filter(
  2146. and_(BizChannelAccount.channel_id == channel_id, BizChannelAccount.del_flag == 0)
  2147. ).with_for_update()
  2148. if tenant_no:
  2149. q = q.filter(BizChannelAccount.tenant_no == tenant_no)
  2150. obj = q.first()
  2151. if obj:
  2152. return obj
  2153. obj = BizChannelAccount(
  2154. id=generate_uuid(),
  2155. channel_id=channel_id,
  2156. balance_amount=0,
  2157. tenant_no=tenant_no or "zkxw",
  2158. order_num=0,
  2159. remark="",
  2160. create_by=user_id,
  2161. create_dept_id="",
  2162. create_time=datetime.now(),
  2163. update_by=user_id,
  2164. update_time=datetime.now(),
  2165. del_flag=0,
  2166. )
  2167. db.add(obj)
  2168. db.flush()
  2169. return obj
  2170. # ────────────────────────────────────────────
  2171. # Entity 16: BizChannelProductQuota 渠道产品份数
  2172. # ────────────────────────────────────────────
  2173. def _row_to_dict_channel_product_quota(row) -> Dict[str, Any]:
  2174. return {
  2175. "id": row.id,
  2176. "channelId": row.channel_id or "",
  2177. "projectCode": row.project_code or "",
  2178. "projectName": row.project_name or "",
  2179. "cellTypeCode": row.cell_type_code or "",
  2180. "cellTypeName": row.cell_type_name or "",
  2181. "specText": row.spec_text or "",
  2182. "totalQty": float(row.total_qty) if row.total_qty else 0,
  2183. "usedQty": float(row.used_qty) if row.used_qty else 0,
  2184. "remainingQty": float(row.remaining_qty) if row.remaining_qty else 0,
  2185. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  2186. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  2187. }
  2188. def _upsert_channel_product_quota(
  2189. db,
  2190. channel_id,
  2191. project_code,
  2192. project_name,
  2193. cell_type_code,
  2194. cell_type_name,
  2195. spec_text,
  2196. add_qty,
  2197. tenant_no,
  2198. user_id,
  2199. ):
  2200. """增加份数:已存在则累加,不存在则新建"""
  2201. q = db.query(BizChannelProductQuota).filter(
  2202. and_(
  2203. BizChannelProductQuota.channel_id == channel_id,
  2204. BizChannelProductQuota.project_code == (project_code or ""),
  2205. BizChannelProductQuota.cell_type_code == (cell_type_code or ""),
  2206. BizChannelProductQuota.spec_text == (spec_text or ""),
  2207. BizChannelProductQuota.del_flag == 0,
  2208. )
  2209. )
  2210. if tenant_no:
  2211. q = q.filter(BizChannelProductQuota.tenant_no == tenant_no)
  2212. obj = q.first()
  2213. if obj:
  2214. obj.total_qty = (float(obj.total_qty) if obj.total_qty else 0) + add_qty
  2215. obj.remaining_qty = (float(obj.remaining_qty) if obj.remaining_qty else 0) + add_qty
  2216. obj.update_by = user_id
  2217. obj.update_time = datetime.now()
  2218. db.add(obj)
  2219. db.flush()
  2220. return obj
  2221. obj = BizChannelProductQuota(
  2222. id=generate_uuid(),
  2223. channel_id=channel_id,
  2224. project_code=project_code or "",
  2225. project_name=project_name or "",
  2226. cell_type_code=cell_type_code or "",
  2227. cell_type_name=cell_type_name or "",
  2228. spec_text=spec_text or "",
  2229. total_qty=add_qty,
  2230. used_qty=0,
  2231. remaining_qty=add_qty,
  2232. tenant_no=tenant_no or "zkxw",
  2233. order_num=0,
  2234. remark="",
  2235. create_by=user_id,
  2236. create_dept_id="",
  2237. create_time=datetime.now(),
  2238. update_by=user_id,
  2239. update_time=datetime.now(),
  2240. del_flag=0,
  2241. )
  2242. db.add(obj)
  2243. db.flush()
  2244. return obj
  2245. def _reduce_channel_product_quota(db, channel_id, project_code, cell_type_code, spec_text, reduce_qty, user_id):
  2246. """扣减份数:验证 remaining_qty >= reduce_qty"""
  2247. q = db.query(BizChannelProductQuota).filter(
  2248. and_(
  2249. BizChannelProductQuota.channel_id == channel_id,
  2250. BizChannelProductQuota.project_code == (project_code or ""),
  2251. BizChannelProductQuota.cell_type_code == (cell_type_code or ""),
  2252. BizChannelProductQuota.spec_text == (spec_text or ""),
  2253. BizChannelProductQuota.del_flag == 0,
  2254. )
  2255. )
  2256. obj = q.first()
  2257. if not obj:
  2258. raise ValueError("份数记录不存在")
  2259. remaining = float(obj.remaining_qty) if obj.remaining_qty else 0
  2260. if remaining < reduce_qty:
  2261. raise ValueError(f"剩余份数不足,当前剩余 {remaining},需扣减 {reduce_qty}")
  2262. obj.total_qty = (float(obj.total_qty) if obj.total_qty else 0) - reduce_qty
  2263. obj.remaining_qty = remaining - reduce_qty
  2264. obj.update_by = user_id
  2265. obj.update_time = datetime.now()
  2266. db.add(obj)
  2267. db.flush()
  2268. return obj
  2269. def list_channel_product_quotas(db, channel_id, tenant_no=None):
  2270. """查询渠道下所有产品份数"""
  2271. q = db.query(BizChannelProductQuota).filter(
  2272. and_(
  2273. BizChannelProductQuota.channel_id == channel_id,
  2274. BizChannelProductQuota.del_flag == 0,
  2275. )
  2276. )
  2277. if tenant_no:
  2278. q = q.filter(BizChannelProductQuota.tenant_no == tenant_no)
  2279. rows = q.order_by(BizChannelProductQuota.create_time.desc()).all()
  2280. return [_row_to_dict_channel_product_quota(x) for x in rows]
  2281. # ────────────────────────────────────────────
  2282. # Entity 17: BizChannelServiceOrder 渠道服务订单
  2283. # ────────────────────────────────────────────
  2284. def _row_to_dict_service_order(row, db=None, tenant_no=None) -> Dict[str, Any]:
  2285. tn = tenant_no or row.tenant_no
  2286. d = {
  2287. "id": row.id,
  2288. "channelId": row.channel_id or "",
  2289. "channelName": row.channel_name or "",
  2290. "deptName": row.dept_name or "",
  2291. "orderType": row.order_type or "",
  2292. "prepayAmount": float(row.prepay_amount) if row.prepay_amount else 0,
  2293. "purchaseTotalAmount": float(row.purchase_total_amount) if row.purchase_total_amount else 0,
  2294. "orderStatus": row.order_status or "",
  2295. "channelAgreementId": row.channel_agreement_id or "",
  2296. "totalRechargeAmount": float(row.total_recharge_amount) if row.total_recharge_amount else 0,
  2297. "totalRefundAmount": float(row.total_refund_amount) if row.total_refund_amount else 0,
  2298. "currentBalance": float(row.current_balance) if row.current_balance else 0,
  2299. "balanceBefore": float(row.balance_before) if row.balance_before else 0,
  2300. "balanceAfter": float(row.balance_after) if row.balance_after else 0,
  2301. "financeSubmitBy": row.finance_submit_by or "",
  2302. "financeSubmitTime": row.finance_submit_time.strftime("%Y-%m-%d %H:%M:%S") if row.finance_submit_time else "",
  2303. "financeSubmitRemark": row.finance_submit_remark or "",
  2304. "financeSubmitAttachments": row.finance_submit_attachments or "",
  2305. "financeSubmitAttachmentFiles": urls_to_attachment_list(row.finance_submit_attachments, db, tn) if db and row.finance_submit_attachments else [],
  2306. "financeConfirmTime": row.finance_confirm_time.strftime("%Y-%m-%d %H:%M:%S") if row.finance_confirm_time else "",
  2307. "financeConfirmAttachments": row.finance_confirm_attachments or "",
  2308. "financeConfirmAttachmentFiles": urls_to_attachment_list(row.finance_confirm_attachments, db, tn) if db and row.finance_confirm_attachments else [],
  2309. "financeRemark": row.finance_remark or "",
  2310. "refundApplyBy": row.refund_apply_by or "",
  2311. "refundApplyTime": row.refund_apply_time.strftime("%Y-%m-%d %H:%M:%S") if row.refund_apply_time else "",
  2312. "refundAmount": float(row.refund_amount) if row.refund_amount else 0,
  2313. "refundRemark": row.refund_remark or "",
  2314. "refundAttachments": row.refund_attachments or "",
  2315. "refundAttachmentFiles": urls_to_attachment_list(row.refund_attachments, db, tn) if db and row.refund_attachments else [],
  2316. "refundFinanceConfirmTime": row.refund_finance_confirm_time.strftime("%Y-%m-%d %H:%M:%S") if row.refund_finance_confirm_time else "",
  2317. "refundFinanceConfirmAttachments": row.refund_finance_confirm_attachments or "",
  2318. "refundFinanceConfirmAttachmentFiles": urls_to_attachment_list(row.refund_finance_confirm_attachments, db, tn) if db and row.refund_finance_confirm_attachments else [],
  2319. "refundFinanceRemark": row.refund_finance_remark or "",
  2320. "tenantNo": row.tenant_no or "",
  2321. "remark": row.remark or "",
  2322. "createBy": row.create_by or "",
  2323. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  2324. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  2325. }
  2326. return d
  2327. def _row_to_dict_service_order_product(row) -> Dict[str, Any]:
  2328. return {
  2329. "id": row.id,
  2330. "serviceOrderId": row.service_order_id or "",
  2331. "projectCode": row.project_code or "",
  2332. "projectName": row.project_name or "",
  2333. "cellType": row.cell_type or "",
  2334. "specText": row.spec_text or "",
  2335. "agreementUnitPrice": float(row.agreement_unit_price) if row.agreement_unit_price else 0,
  2336. "buyQty": float(row.buy_qty) if row.buy_qty else 0,
  2337. "buyAmount": float(row.buy_amount) if row.buy_amount else 0,
  2338. "totalBuyQty": float(row.total_buy_qty) if row.total_buy_qty else 0,
  2339. "totalUnsubscribeQty": float(row.total_unsubscribe_qty) if row.total_unsubscribe_qty else 0,
  2340. "currentQty": float(row.current_qty) if row.current_qty else 0,
  2341. "totalUnsubscribeAmount": float(row.total_unsubscribe_amount) if row.total_unsubscribe_amount else 0,
  2342. }
  2343. def _row_to_dict_service_order_txn(row, db=None, tenant_no=None) -> Dict[str, Any]:
  2344. tn = tenant_no or row.tenant_no
  2345. d = {
  2346. "id": row.id,
  2347. "serviceOrderId": row.service_order_id or "",
  2348. "txnType": row.txn_type or "",
  2349. "amount": float(row.amount) if row.amount else 0,
  2350. "txnStatus": row.txn_status or "",
  2351. "productsJson": row.products_json or "",
  2352. "remark": row.remark or "",
  2353. "financeSubmitBy": row.finance_submit_by or "",
  2354. "financeSubmitTime": row.finance_submit_time.strftime("%Y-%m-%d %H:%M:%S") if row.finance_submit_time else "",
  2355. "financeSubmitRemark": row.finance_submit_remark or "",
  2356. "financeSubmitAttachments": row.finance_submit_attachments or "",
  2357. "financeSubmitAttachmentFiles": urls_to_attachment_list(row.finance_submit_attachments, db, tn) if db and row.finance_submit_attachments else [],
  2358. "financeConfirmTime": row.finance_confirm_time.strftime("%Y-%m-%d %H:%M:%S") if row.finance_confirm_time else "",
  2359. "financeConfirmBy": row.finance_confirm_by or "",
  2360. "financeConfirmAttachments": row.finance_confirm_attachments or "",
  2361. "financeConfirmAttachmentFiles": urls_to_attachment_list(row.finance_confirm_attachments, db, tn) if db and row.finance_confirm_attachments else [],
  2362. "financeRemark": row.finance_remark or "",
  2363. "voidBy": row.void_by or "",
  2364. "voidTime": row.void_time.strftime("%Y-%m-%d %H:%M:%S") if row.void_time else "",
  2365. "balanceBefore": float(row.balance_before) if row.balance_before else 0,
  2366. "balanceAfter": float(row.balance_after) if row.balance_after else 0,
  2367. "tenantNo": row.tenant_no or "",
  2368. "createBy": row.create_by or "",
  2369. "createTime": row.create_time.strftime("%Y-%m-%d %H:%M:%S") if row.create_time else "",
  2370. "updateTime": row.update_time.strftime("%Y-%m-%d %H:%M:%S") if row.update_time else "",
  2371. }
  2372. if row.products_json:
  2373. try:
  2374. d["products"] = json.loads(row.products_json)
  2375. except Exception:
  2376. d["products"] = []
  2377. else:
  2378. d["products"] = []
  2379. return d
  2380. def _service_order_txn_to_response_dict(
  2381. db,
  2382. txn,
  2383. tenant_no=None,
  2384. *,
  2385. order=None,
  2386. name_map=None,
  2387. ) -> Dict[str, Any]:
  2388. """流水响应字典:含订单渠道信息与提交人/创建人昵称。"""
  2389. d = _row_to_dict_service_order_txn(txn, db, tenant_no)
  2390. if order is None and txn.service_order_id:
  2391. order = get_service_order(db, txn.service_order_id, tenant_no)
  2392. if order:
  2393. d["channelName"] = order.channel_name or ""
  2394. d["deptName"] = order.dept_name or ""
  2395. d["orderType"] = order.order_type or ""
  2396. confirm_uid = txn.finance_confirm_by or (
  2397. txn.update_by
  2398. if (txn.txn_status or "") == "yqr" and txn.finance_confirm_time and not txn.finance_confirm_by
  2399. else None
  2400. )
  2401. void_uid = txn.void_by or (
  2402. txn.update_by
  2403. if (txn.txn_status or "") == "yzf"
  2404. and txn.finance_submit_time
  2405. and not txn.void_by
  2406. else None
  2407. )
  2408. if name_map is None:
  2409. user_ids = [x for x in (txn.create_by, txn.finance_submit_by, confirm_uid, void_uid) if x]
  2410. name_map = _user_nick_name_map(db, user_ids) if db and user_ids else {}
  2411. d["createByName"] = name_map.get(txn.create_by or "", "") or (txn.create_by or "")
  2412. d["financeSubmitByName"] = name_map.get(txn.finance_submit_by or "", "") or (
  2413. txn.finance_submit_by or ""
  2414. )
  2415. d["financeConfirmByName"] = name_map.get(confirm_uid or "", "") or (confirm_uid or "")
  2416. if not d.get("voidTime") and (txn.txn_status or "") == "yzf" and txn.update_time:
  2417. d["voidTime"] = txn.update_time.strftime("%Y-%m-%d %H:%M:%S")
  2418. d["voidByName"] = name_map.get(void_uid or "", "") or (void_uid or "")
  2419. if not (d.get("financeSubmitRemark") or "").strip() and (d.get("remark") or "").strip():
  2420. d["financeSubmitRemark"] = (d.get("remark") or "").strip()
  2421. _enrich_finance_operate_fields(d, txn)
  2422. return d
  2423. def _enrich_finance_operate_fields(d: Dict[str, Any], txn) -> None:
  2424. """财务页「已处理」展示:操作类型、操作人、操作时间。"""
  2425. status = txn.txn_status or ""
  2426. if status == "yqr" and txn.finance_confirm_time:
  2427. d["financeOperateType"] = "confirm"
  2428. d["financeOperateTypeLabel"] = "财务确认"
  2429. d["financeOperateByName"] = d.get("financeConfirmByName") or d.get("financeConfirmBy") or ""
  2430. d["financeOperateTime"] = d.get("financeConfirmTime") or ""
  2431. elif status == "yzf" and txn.finance_submit_time:
  2432. d["financeOperateType"] = "void"
  2433. d["financeOperateTypeLabel"] = "财务作废"
  2434. d["financeOperateByName"] = d.get("voidByName") or d.get("voidBy") or ""
  2435. d["financeOperateTime"] = d.get("voidTime") or ""
  2436. else:
  2437. d["financeOperateType"] = ""
  2438. d["financeOperateTypeLabel"] = ""
  2439. d["financeOperateByName"] = ""
  2440. d["financeOperateTime"] = ""
  2441. def list_channel_service_orders(
  2442. db,
  2443. *,
  2444. page=1,
  2445. page_size=10,
  2446. order_type=None,
  2447. order_status=None,
  2448. channel_id=None,
  2449. keyword=None,
  2450. create_time_start=None,
  2451. create_time_end=None,
  2452. tenant_no=None,
  2453. user_info=None,
  2454. ):
  2455. def _filtered():
  2456. q0 = db.query(BizChannelServiceOrder).filter(BizChannelServiceOrder.del_flag == 0)
  2457. if tenant_no:
  2458. q0 = q0.filter(BizChannelServiceOrder.tenant_no == tenant_no)
  2459. ot = (order_type or "").strip()
  2460. if ot:
  2461. q0 = q0.filter(BizChannelServiceOrder.order_type == ot)
  2462. # Support comma-separated order_status
  2463. os_raw = (order_status or "").strip()
  2464. if os_raw:
  2465. statuses = [s.strip() for s in os_raw.split(",") if s.strip()]
  2466. if statuses:
  2467. q0 = q0.filter(BizChannelServiceOrder.order_status.in_(statuses))
  2468. if channel_id:
  2469. q0 = q0.filter(BizChannelServiceOrder.channel_id == channel_id)
  2470. kw = (keyword or "").strip()
  2471. if kw:
  2472. like = f"%{kw}%"
  2473. q0 = q0.filter(
  2474. or_(
  2475. BizChannelServiceOrder.channel_name.like(like),
  2476. BizChannelServiceOrder.dept_name.like(like),
  2477. )
  2478. )
  2479. if create_time_start and create_time_start.strip():
  2480. q0 = q0.filter(
  2481. BizChannelServiceOrder.create_time
  2482. >= _normalize_date_bound(create_time_start.strip())
  2483. )
  2484. if create_time_end and create_time_end.strip():
  2485. q0 = q0.filter(
  2486. BizChannelServiceOrder.create_time
  2487. <= _normalize_date_bound(create_time_end.strip(), end_of_day=True)
  2488. )
  2489. if user_info:
  2490. from app.core.data_scope import apply_data_scope
  2491. q0 = apply_data_scope(q0, BizChannelServiceOrder, user_info, db, tenant_no)
  2492. return q0
  2493. total = _filtered().order_by(None).with_entities(func.count(BizChannelServiceOrder.id)).scalar() or 0
  2494. page = max(1, page)
  2495. page_size = min(max(1, page_size), 100)
  2496. rows = (
  2497. _filtered()
  2498. .order_by(BizChannelServiceOrder.create_time.desc())
  2499. .offset((page - 1) * page_size)
  2500. .limit(page_size)
  2501. .all()
  2502. )
  2503. user_ids: List[str] = []
  2504. for x in rows:
  2505. if x.create_by:
  2506. user_ids.append(x.create_by)
  2507. if x.finance_submit_by:
  2508. user_ids.append(x.finance_submit_by)
  2509. name_map = _user_nick_name_map(db, user_ids) if db else {}
  2510. return [
  2511. {
  2512. **_row_to_dict_service_order(x, db, tenant_no),
  2513. "createByName": name_map.get(x.create_by or "", "") or (x.create_by or ""),
  2514. "financeSubmitByName": name_map.get(x.finance_submit_by or "", "") or (x.finance_submit_by or ""),
  2515. }
  2516. for x in rows
  2517. ], int(total)
  2518. def get_service_order(db, order_id, tenant_no=None):
  2519. q = db.query(BizChannelServiceOrder).filter(
  2520. and_(BizChannelServiceOrder.id == order_id, BizChannelServiceOrder.del_flag == 0)
  2521. )
  2522. if tenant_no:
  2523. q = q.filter(BizChannelServiceOrder.tenant_no == tenant_no)
  2524. return q.first()
  2525. def get_service_order_products(db, service_order_id, tenant_no=None):
  2526. q = db.query(BizChannelServiceOrderProduct).filter(
  2527. and_(
  2528. BizChannelServiceOrderProduct.service_order_id == service_order_id,
  2529. BizChannelServiceOrderProduct.del_flag == 0,
  2530. )
  2531. )
  2532. if tenant_no:
  2533. q = q.filter(BizChannelServiceOrderProduct.tenant_no == tenant_no)
  2534. rows = q.order_by(BizChannelServiceOrderProduct.create_time.asc()).all()
  2535. return [_row_to_dict_service_order_product(x) for x in rows]
  2536. # ────────────────────────────────────────────
  2537. # Service Order: Create / Update / Delete
  2538. # ────────────────────────────────────────────
  2539. def _calc_gmcp_purchase_total(products):
  2540. total = 0.0
  2541. for p in products:
  2542. price = float(p.get("agreementUnitPrice") or p.get("agreement_unit_price") or 0)
  2543. qty = float(p.get("buyQty") or p.get("buy_qty") or 0)
  2544. total += price * qty
  2545. return total
  2546. def _resolve_agreement_price_cell_type(db, channel_agreement_id, project_code, spec_text, cell_type_fallback=""):
  2547. price_row = (
  2548. db.query(BizChannelAgreementProductPrice)
  2549. .filter(
  2550. and_(
  2551. BizChannelAgreementProductPrice.channel_agreement_id == channel_agreement_id,
  2552. BizChannelAgreementProductPrice.project_code == (project_code or ""),
  2553. BizChannelAgreementProductPrice.spec_text == (spec_text or ""),
  2554. BizChannelAgreementProductPrice.del_flag == 0,
  2555. )
  2556. )
  2557. ).first()
  2558. if price_row:
  2559. return price_row.cell_type_code or "", price_row.cell_type_name or cell_type_fallback
  2560. return "", cell_type_fallback
  2561. def _save_gmcp_product_rows(
  2562. db,
  2563. order_id,
  2564. products,
  2565. tenant_no,
  2566. user_id,
  2567. now=None,
  2568. ):
  2569. """仅保存购买订单产品明细,不调整份数库存"""
  2570. now = now or datetime.now()
  2571. for p in products:
  2572. project_code = p.get("projectCode") or p.get("project_code") or ""
  2573. project_name = p.get("projectName") or p.get("project_name") or ""
  2574. cell_type = p.get("cellType") or p.get("cell_type") or ""
  2575. spec_text = p.get("specText") or p.get("spec_text") or ""
  2576. agreement_unit_price = float(p.get("agreementUnitPrice") or p.get("agreement_unit_price") or 0)
  2577. buy_qty = float(p.get("buyQty") or p.get("buy_qty") or 0)
  2578. buy_amount = float(p.get("buyAmount") or p.get("buy_amount") or agreement_unit_price * buy_qty)
  2579. prod_row = BizChannelServiceOrderProduct(
  2580. id=generate_uuid(),
  2581. service_order_id=order_id,
  2582. project_code=project_code,
  2583. project_name=project_name,
  2584. cell_type=cell_type,
  2585. spec_text=spec_text,
  2586. agreement_unit_price=agreement_unit_price,
  2587. buy_qty=buy_qty,
  2588. buy_amount=buy_amount,
  2589. tenant_no=tenant_no or "zkxw",
  2590. order_num=0,
  2591. remark="",
  2592. create_by=user_id,
  2593. create_dept_id="",
  2594. create_time=now,
  2595. update_by=user_id,
  2596. update_time=now,
  2597. del_flag=0,
  2598. )
  2599. db.add(prod_row)
  2600. db.flush()
  2601. def _replace_gmcp_product_rows(db, order_id, products, tenant_no, user_id):
  2602. now = datetime.now()
  2603. db.query(BizChannelServiceOrderProduct).filter(
  2604. and_(
  2605. BizChannelServiceOrderProduct.service_order_id == order_id,
  2606. BizChannelServiceOrderProduct.del_flag == 0,
  2607. )
  2608. ).update({"del_flag": 1, "update_time": now}, synchronize_session=False)
  2609. _save_gmcp_product_rows(db, order_id, products, tenant_no, user_id, now=now)
  2610. def _apply_gmcp_order_quotas(db, order, tenant_no, user_id):
  2611. """财务确认收款后,按订单明细增加渠道产品份数"""
  2612. products = get_service_order_products(db, order.id, tenant_no)
  2613. for prod in products:
  2614. project_code = prod.get("projectCode") or ""
  2615. spec_text = prod.get("specText") or ""
  2616. cell_type_code, cell_type_name = _resolve_agreement_price_cell_type(
  2617. db,
  2618. order.channel_agreement_id,
  2619. project_code,
  2620. spec_text,
  2621. prod.get("cellType") or "",
  2622. )
  2623. _upsert_channel_product_quota(
  2624. db,
  2625. channel_id=order.channel_id,
  2626. project_code=project_code,
  2627. project_name=prod.get("projectName") or "",
  2628. cell_type_code=cell_type_code,
  2629. cell_type_name=cell_type_name,
  2630. spec_text=spec_text,
  2631. add_qty=float(prod.get("buyQty") or 0),
  2632. tenant_no=tenant_no or order.tenant_no,
  2633. user_id=user_id,
  2634. )
  2635. def _get_valid_channel_agreement(db, channel_agreement_id, channel_id, tenant_no=None):
  2636. """校验渠道协议存在、属于当前渠道且已生效"""
  2637. aid = (channel_agreement_id or "").strip()
  2638. if not aid:
  2639. raise ValueError("请选择渠道协议")
  2640. q = db.query(BizChannelAgreement).filter(
  2641. and_(
  2642. BizChannelAgreement.id == aid,
  2643. BizChannelAgreement.channel_id == channel_id,
  2644. BizChannelAgreement.del_flag == 0,
  2645. )
  2646. )
  2647. if tenant_no:
  2648. q = q.filter(BizChannelAgreement.tenant_no == tenant_no)
  2649. agreement = q.first()
  2650. if not agreement:
  2651. raise ValueError("渠道协议不存在或不属于当前渠道")
  2652. if (agreement.agreement_status or "").strip() != "sxz":
  2653. raise ValueError("协议未生效,无法使用")
  2654. return agreement
  2655. def create_channel_service_order(
  2656. db,
  2657. *,
  2658. channel_id,
  2659. order_type,
  2660. channel_name=None,
  2661. dept_name=None,
  2662. prepay_amount=None,
  2663. channel_agreement_id=None,
  2664. products=None,
  2665. tenant_no=None,
  2666. remark=None,
  2667. user_id=None,
  2668. ):
  2669. # Validate channel exists
  2670. channel = get_channel(db, channel_id, tenant_no)
  2671. if not channel:
  2672. raise ValueError("渠道不存在")
  2673. now = datetime.now()
  2674. balance_before = None
  2675. balance_after = None
  2676. purchase_total_amount = None
  2677. order_status = "dsx"
  2678. if order_type == "yfk":
  2679. # 预存款订单:创建后为待生效,充值金额通过「发起充值」流水录入
  2680. _get_valid_channel_agreement(db, channel_agreement_id, channel_id, tenant_no)
  2681. order_status = "dsx"
  2682. prepay_amount = None
  2683. elif order_type == "gmcp":
  2684. # 购买产品订单:创建后为待生效,走财务确认后生效
  2685. if not products:
  2686. raise ValueError("购买订单的产品明细不能为空")
  2687. _get_valid_channel_agreement(db, channel_agreement_id, channel_id, tenant_no)
  2688. purchase_total_amount = _calc_gmcp_purchase_total(products)
  2689. order_status = "dsx"
  2690. else:
  2691. raise ValueError(f"不支持的订单类型: {order_type}")
  2692. order_id = generate_uuid()
  2693. order = BizChannelServiceOrder(
  2694. id=order_id,
  2695. channel_id=channel_id,
  2696. channel_name=channel_name or channel.channel_name or "",
  2697. dept_name=dept_name or "",
  2698. order_type=order_type,
  2699. prepay_amount=None,
  2700. purchase_total_amount=purchase_total_amount,
  2701. order_status=order_status,
  2702. channel_agreement_id=channel_agreement_id,
  2703. balance_before=balance_before,
  2704. balance_after=balance_after,
  2705. tenant_no=tenant_no or "zkxw",
  2706. order_num=0,
  2707. remark=remark or "",
  2708. create_by=user_id,
  2709. create_dept_id="",
  2710. create_time=now,
  2711. update_by=user_id,
  2712. update_time=now,
  2713. del_flag=0,
  2714. )
  2715. db.add(order)
  2716. db.flush()
  2717. # For gmcp: create product rows only (quotas applied after finance confirm)
  2718. if order_type == "gmcp" and products:
  2719. _save_gmcp_product_rows(db, order_id, products, tenant_no or "zkxw", user_id, now=now)
  2720. db.commit()
  2721. db.refresh(order)
  2722. return order
  2723. def update_channel_service_order(db, order_id, *, tenant_no=None, user_id=None, products=None, **kwargs):
  2724. """待生效状态下可更新;预存款可改协议,购买订单可改协议与产品明细"""
  2725. order = get_service_order(db, order_id, tenant_no)
  2726. if not order:
  2727. raise ValueError("订单不存在")
  2728. if order.order_status != "dsx":
  2729. raise ValueError("当前状态不允许编辑")
  2730. if "dept_name" in kwargs and kwargs["dept_name"] is not None:
  2731. order.dept_name = kwargs["dept_name"]
  2732. if "remark" in kwargs and kwargs["remark"] is not None:
  2733. order.remark = kwargs["remark"]
  2734. if order.order_type == "yfk":
  2735. if "channel_agreement_id" in kwargs and kwargs["channel_agreement_id"] is not None:
  2736. _get_valid_channel_agreement(
  2737. db, kwargs["channel_agreement_id"], order.channel_id, tenant_no
  2738. )
  2739. order.channel_agreement_id = kwargs["channel_agreement_id"]
  2740. elif order.order_type == "gmcp":
  2741. if "channel_agreement_id" in kwargs and kwargs["channel_agreement_id"] is not None:
  2742. _get_valid_channel_agreement(
  2743. db, kwargs["channel_agreement_id"], order.channel_id, tenant_no
  2744. )
  2745. order.channel_agreement_id = kwargs["channel_agreement_id"]
  2746. if products is not None:
  2747. if not products:
  2748. raise ValueError("购买订单的产品明细不能为空")
  2749. order.purchase_total_amount = _calc_gmcp_purchase_total(products)
  2750. _replace_gmcp_product_rows(db, order_id, products, tenant_no or order.tenant_no, user_id)
  2751. order.update_by = user_id
  2752. order.update_time = datetime.now()
  2753. db.add(order)
  2754. db.commit()
  2755. db.refresh(order)
  2756. return order
  2757. def soft_delete_service_order(db, order_id, tenant_no=None):
  2758. """仅允许待生效状态下删除订单"""
  2759. order = get_service_order(db, order_id, tenant_no)
  2760. if not order:
  2761. raise ValueError("订单不存在")
  2762. if order.order_status != "dsx":
  2763. raise ValueError("仅待生效订单可删除")
  2764. order.del_flag = 1
  2765. order.update_time = datetime.now()
  2766. db.add(order)
  2767. db.commit()
  2768. # ────────────────────────────────────────────
  2769. # Service Order: Status Transitions
  2770. # ────────────────────────────────────────────
  2771. def submit_finance_confirm(
  2772. db,
  2773. order_id,
  2774. *,
  2775. finance_submit_remark=None,
  2776. finance_submit_attachments=None,
  2777. tenant_no=None,
  2778. user_id=None,
  2779. ):
  2780. """发起财务确认:type=yfk/gmcp + status=dsx → dqr"""
  2781. order = get_service_order(db, order_id, tenant_no)
  2782. if not order:
  2783. raise ValueError("订单不存在")
  2784. if order.order_type not in ("yfk", "gmcp"):
  2785. raise ValueError("仅预存款或购买产品订单可发起财务确认")
  2786. if order.order_status != "dsx":
  2787. raise ValueError("仅待生效状态可发起财务确认")
  2788. now = datetime.now()
  2789. order.order_status = "dqr"
  2790. order.finance_submit_remark = finance_submit_remark or ""
  2791. order.finance_submit_attachments = finance_submit_attachments or ""
  2792. order.finance_submit_time = now
  2793. order.finance_submit_by = user_id
  2794. order.update_by = user_id
  2795. order.update_time = now
  2796. db.add(order)
  2797. db.commit()
  2798. db.refresh(order)
  2799. return order
  2800. def confirm_receipt(db, order_id, *, finance_remark=None, finance_confirm_attachments=None, tenant_no=None, user_id=None):
  2801. """确认到账:yfk dqr→yqr 增加余额;gmcp dqr→sxz 增加产品份数"""
  2802. order = get_service_order(db, order_id, tenant_no)
  2803. if not order:
  2804. raise ValueError("订单不存在")
  2805. if order.order_type not in ("yfk", "gmcp"):
  2806. raise ValueError("仅预存款或购买产品订单可确认到账")
  2807. if order.order_status != "dqr":
  2808. raise ValueError("仅待确认状态可确认到账")
  2809. now = datetime.now()
  2810. if order.order_type == "yfk":
  2811. account = _get_or_create_channel_account(db, order.channel_id, tenant_no or order.tenant_no, user_id)
  2812. balance_before = float(account.balance_amount) if account.balance_amount else 0
  2813. add_amount = float(order.prepay_amount) if order.prepay_amount else 0
  2814. account.balance_amount = balance_before + add_amount
  2815. balance_after = float(account.balance_amount)
  2816. account.update_by = user_id
  2817. account.update_time = now
  2818. db.add(account)
  2819. db.flush()
  2820. order.order_status = "yqr"
  2821. order.balance_before = balance_before
  2822. order.balance_after = balance_after
  2823. else:
  2824. _apply_gmcp_order_quotas(db, order, tenant_no, user_id)
  2825. order.order_status = "sxz"
  2826. order.finance_confirm_time = now
  2827. order.finance_confirm_attachments = finance_confirm_attachments or ""
  2828. order.finance_remark = finance_remark or ""
  2829. order.update_by = user_id
  2830. order.update_time = now
  2831. db.add(order)
  2832. db.commit()
  2833. db.refresh(order)
  2834. return order
  2835. def submit_refund(db, order_id, *, refund_amount=None, refund_remark=None, refund_attachments=None, tenant_no=None, user_id=None):
  2836. """申请退款:type=yfk + status=yqr → tfqrz"""
  2837. order = get_service_order(db, order_id, tenant_no)
  2838. if not order:
  2839. raise ValueError("订单不存在")
  2840. if order.order_type != "yfk":
  2841. raise ValueError("仅预存款订单可申请退款")
  2842. if order.order_status != "yqr":
  2843. raise ValueError("仅已确认状态可申请退款")
  2844. now = datetime.now()
  2845. # Default refund_amount to prepay_amount
  2846. if refund_amount is None:
  2847. refund_amount = float(order.prepay_amount) if order.prepay_amount else 0
  2848. else:
  2849. refund_amount = float(refund_amount)
  2850. prepay = float(order.prepay_amount) if order.prepay_amount else 0
  2851. if refund_amount > prepay:
  2852. raise ValueError(f"退款金额不能大于预存金额 {prepay}")
  2853. order.order_status = "tfqrz"
  2854. order.refund_apply_by = user_id
  2855. order.refund_apply_time = now
  2856. order.refund_amount = refund_amount
  2857. order.refund_remark = refund_remark or ""
  2858. order.refund_attachments = refund_attachments or ""
  2859. order.update_by = user_id
  2860. order.update_time = now
  2861. db.add(order)
  2862. db.commit()
  2863. db.refresh(order)
  2864. return order
  2865. def confirm_refund(db, order_id, *, refund_finance_remark=None, refund_finance_confirm_attachments=None, tenant_no=None, user_id=None):
  2866. """确认退款:status=tfqrz → ytf,扣减余额"""
  2867. order = get_service_order(db, order_id, tenant_no)
  2868. if not order:
  2869. raise ValueError("订单不存在")
  2870. if order.order_status != "tfqrz":
  2871. raise ValueError("仅退款待确认状态可确认退款")
  2872. now = datetime.now()
  2873. account = _get_or_create_channel_account(db, order.channel_id, tenant_no or order.tenant_no, user_id)
  2874. balance_before = float(account.balance_amount) if account.balance_amount else 0
  2875. deduct = float(order.refund_amount) if order.refund_amount else 0
  2876. account.balance_amount = balance_before - deduct
  2877. balance_after = float(account.balance_amount)
  2878. account.update_by = user_id
  2879. account.update_time = now
  2880. db.add(account)
  2881. db.flush()
  2882. order.order_status = "ytf"
  2883. order.refund_finance_confirm_time = now
  2884. order.refund_finance_confirm_attachments = refund_finance_confirm_attachments or ""
  2885. order.refund_finance_remark = refund_finance_remark or ""
  2886. order.balance_before = balance_before
  2887. order.balance_after = balance_after
  2888. order.update_by = user_id
  2889. order.update_time = now
  2890. db.add(order)
  2891. db.commit()
  2892. db.refresh(order)
  2893. return order
  2894. def void_service_order(db, order_id, *, tenant_no=None, user_id=None):
  2895. """作废订单:type=gmcp + status=sxz → yzf,扣减已生效的产品份数"""
  2896. order = get_service_order(db, order_id, tenant_no)
  2897. if not order:
  2898. raise ValueError("订单不存在")
  2899. if order.order_type != "gmcp":
  2900. raise ValueError("仅购买产品订单可作废")
  2901. if order.order_status != "sxz":
  2902. raise ValueError("仅生效中状态可作废")
  2903. now = datetime.now()
  2904. # Reduce product quotas
  2905. products = get_service_order_products(db, order_id, tenant_no)
  2906. for prod in products:
  2907. cell_type_code, _ = _resolve_agreement_price_cell_type(
  2908. db,
  2909. order.channel_agreement_id,
  2910. prod.get("projectCode") or "",
  2911. prod.get("specText") or "",
  2912. prod.get("cellType") or "",
  2913. )
  2914. _reduce_channel_product_quota(
  2915. db,
  2916. channel_id=order.channel_id,
  2917. project_code=prod.get("projectCode") or "",
  2918. cell_type_code=cell_type_code,
  2919. spec_text=prod.get("specText") or "",
  2920. reduce_qty=prod.get("buyQty") or 0,
  2921. user_id=user_id,
  2922. )
  2923. order.order_status = "yzf"
  2924. order.update_by = user_id
  2925. order.update_time = now
  2926. db.add(order)
  2927. db.commit()
  2928. db.refresh(order)
  2929. return order
  2930. # ────────────────────────────────────────────
  2931. # Service Order Txn: 操作流水 CRUD
  2932. # ────────────────────────────────────────────
  2933. def list_service_order_txns(
  2934. db,
  2935. *,
  2936. service_order_id=None,
  2937. txn_status=None,
  2938. txn_type=None,
  2939. keyword=None,
  2940. create_time_start=None,
  2941. create_time_end=None,
  2942. finance_done_only=False,
  2943. page=1,
  2944. page_size=20,
  2945. tenant_no=None,
  2946. ):
  2947. """查询操作流水列表。finance_done_only:财务已处理(已确认或财务侧作废)。"""
  2948. q = db.query(BizChannelServiceOrderTxn).filter(BizChannelServiceOrderTxn.del_flag == 0)
  2949. if tenant_no:
  2950. q = q.filter(BizChannelServiceOrderTxn.tenant_no == tenant_no)
  2951. if service_order_id:
  2952. q = q.filter(BizChannelServiceOrderTxn.service_order_id == service_order_id)
  2953. if finance_done_only:
  2954. q = q.filter(
  2955. or_(
  2956. BizChannelServiceOrderTxn.txn_status == "yqr",
  2957. and_(
  2958. BizChannelServiceOrderTxn.txn_status == "yzf",
  2959. BizChannelServiceOrderTxn.finance_submit_time.isnot(None),
  2960. ),
  2961. )
  2962. )
  2963. else:
  2964. ss = (txn_status or "").strip()
  2965. if ss:
  2966. statuses = [s.strip() for s in ss.split(",") if s.strip()]
  2967. if statuses:
  2968. q = q.filter(BizChannelServiceOrderTxn.txn_status.in_(statuses))
  2969. tt = (txn_type or "").strip()
  2970. if tt:
  2971. types = [t.strip() for t in tt.split(",") if t.strip()]
  2972. if types:
  2973. q = q.filter(BizChannelServiceOrderTxn.txn_type.in_(types))
  2974. kw = (keyword or "").strip()
  2975. if kw:
  2976. like = f"%{kw}%"
  2977. q = q.outerjoin(
  2978. BizChannelServiceOrder,
  2979. and_(
  2980. BizChannelServiceOrder.id == BizChannelServiceOrderTxn.service_order_id,
  2981. BizChannelServiceOrder.del_flag == 0,
  2982. ),
  2983. ).filter(
  2984. or_(
  2985. BizChannelServiceOrder.channel_name.like(like),
  2986. BizChannelServiceOrder.dept_name.like(like),
  2987. )
  2988. )
  2989. if create_time_start and create_time_start.strip():
  2990. q = q.filter(
  2991. BizChannelServiceOrderTxn.create_time
  2992. >= _normalize_date_bound(create_time_start.strip())
  2993. )
  2994. if create_time_end and create_time_end.strip():
  2995. q = q.filter(
  2996. BizChannelServiceOrderTxn.create_time
  2997. <= _normalize_date_bound(create_time_end.strip(), end_of_day=True)
  2998. )
  2999. total = q.order_by(None).with_entities(func.count(BizChannelServiceOrderTxn.id)).scalar() or 0
  3000. page = max(1, page)
  3001. page_size = min(max(1, page_size), 100)
  3002. if finance_done_only:
  3003. operate_time = func.coalesce(
  3004. BizChannelServiceOrderTxn.finance_confirm_time,
  3005. BizChannelServiceOrderTxn.void_time,
  3006. )
  3007. rows = (
  3008. q.order_by(operate_time.desc(), BizChannelServiceOrderTxn.create_time.desc())
  3009. .offset((page - 1) * page_size)
  3010. .limit(page_size)
  3011. .all()
  3012. )
  3013. else:
  3014. rows = (
  3015. q.order_by(BizChannelServiceOrderTxn.create_time.desc())
  3016. .offset((page - 1) * page_size)
  3017. .limit(page_size)
  3018. .all()
  3019. )
  3020. if not rows:
  3021. return [], int(total)
  3022. order_ids = list({r.service_order_id for r in rows if r.service_order_id})
  3023. orders: Dict[str, Any] = {}
  3024. if order_ids:
  3025. oq = db.query(BizChannelServiceOrder).filter(
  3026. BizChannelServiceOrder.id.in_(order_ids),
  3027. BizChannelServiceOrder.del_flag == 0,
  3028. )
  3029. if tenant_no:
  3030. oq = oq.filter(BizChannelServiceOrder.tenant_no == tenant_no)
  3031. for o in oq.all():
  3032. orders[o.id] = o
  3033. user_ids: List[str] = []
  3034. for r in rows:
  3035. if r.create_by:
  3036. user_ids.append(r.create_by)
  3037. if r.finance_submit_by:
  3038. user_ids.append(r.finance_submit_by)
  3039. if r.finance_confirm_by:
  3040. user_ids.append(r.finance_confirm_by)
  3041. elif (r.txn_status or "") == "yqr" and r.update_by:
  3042. user_ids.append(r.update_by)
  3043. if r.void_by:
  3044. user_ids.append(r.void_by)
  3045. elif (r.txn_status or "") == "yzf" and r.finance_submit_time and r.update_by:
  3046. user_ids.append(r.update_by)
  3047. name_map = _user_nick_name_map(db, user_ids) if db and user_ids else {}
  3048. items = [
  3049. _service_order_txn_to_response_dict(
  3050. db,
  3051. r,
  3052. tenant_no,
  3053. order=orders.get(r.service_order_id),
  3054. name_map=name_map,
  3055. )
  3056. for r in rows
  3057. ]
  3058. return items, int(total)
  3059. def get_service_order_txn(db, txn_id, *, tenant_no=None):
  3060. """获取单条流水"""
  3061. q = db.query(BizChannelServiceOrderTxn).filter(
  3062. and_(BizChannelServiceOrderTxn.id == txn_id, BizChannelServiceOrderTxn.del_flag == 0)
  3063. )
  3064. if tenant_no:
  3065. q = q.filter(BizChannelServiceOrderTxn.tenant_no == tenant_no)
  3066. return q.first()
  3067. def create_recharge_txn(
  3068. db,
  3069. order_id,
  3070. *,
  3071. amount,
  3072. remark=None,
  3073. finance_submit_attachments=None,
  3074. tenant_no=None,
  3075. user_id=None,
  3076. ):
  3077. """创建充值流水(yfk订单),状态 dsx;附件预存于 finance_submit_attachments 供后续财务确认"""
  3078. order = get_service_order(db, order_id, tenant_no)
  3079. if not order:
  3080. raise ValueError("订单不存在")
  3081. if order.order_type != "yfk":
  3082. raise ValueError("仅充值型订单可发起充值")
  3083. if order.order_status not in ("dsx", "sxz"):
  3084. raise ValueError("订单当前状态不允许充值")
  3085. amount = float(amount)
  3086. if amount <= 0:
  3087. raise ValueError("充值金额必须大于0")
  3088. now = datetime.now()
  3089. txn = BizChannelServiceOrderTxn(
  3090. id=generate_uuid(),
  3091. service_order_id=order_id,
  3092. txn_type="recharge",
  3093. amount=amount,
  3094. txn_status="dsx",
  3095. remark=remark or "",
  3096. finance_submit_attachments=(finance_submit_attachments or "").strip() or None,
  3097. tenant_no=tenant_no or order.tenant_no,
  3098. order_num=0,
  3099. create_by=user_id,
  3100. create_dept_id="",
  3101. create_time=now,
  3102. update_by=user_id,
  3103. update_time=now,
  3104. del_flag=0,
  3105. )
  3106. db.add(txn)
  3107. db.commit()
  3108. db.refresh(txn)
  3109. return txn
  3110. def create_refund_txn(db, order_id, *, amount, remark=None, tenant_no=None, user_id=None):
  3111. """创建退款流水(yfk订单),amount <= current_balance"""
  3112. order = get_service_order(db, order_id, tenant_no)
  3113. if not order:
  3114. raise ValueError("订单不存在")
  3115. if order.order_type != "yfk":
  3116. raise ValueError("仅充值型订单可申请退款")
  3117. if order.order_status != "sxz":
  3118. raise ValueError("仅生效中订单可申请退款")
  3119. amount = float(amount)
  3120. if amount <= 0:
  3121. raise ValueError("退款金额必须大于0")
  3122. current_balance = float(order.current_balance) if order.current_balance else 0
  3123. if amount > current_balance:
  3124. raise ValueError(f"退款金额不能大于当前余额 {current_balance}")
  3125. now = datetime.now()
  3126. txn = BizChannelServiceOrderTxn(
  3127. id=generate_uuid(),
  3128. service_order_id=order_id,
  3129. txn_type="refund",
  3130. amount=amount,
  3131. txn_status="dsx",
  3132. remark=remark or "",
  3133. tenant_no=tenant_no or order.tenant_no,
  3134. order_num=0,
  3135. create_by=user_id,
  3136. create_dept_id="",
  3137. create_time=now,
  3138. update_by=user_id,
  3139. update_time=now,
  3140. del_flag=0,
  3141. )
  3142. db.add(txn)
  3143. db.commit()
  3144. db.refresh(txn)
  3145. return txn
  3146. def create_purchase_txn(db, order_id, *, products, remark=None, tenant_no=None, user_id=None):
  3147. """创建购买产品流水(gmcp订单),同时创建/更新产品持仓"""
  3148. order = get_service_order(db, order_id, tenant_no)
  3149. if not order:
  3150. raise ValueError("订单不存在")
  3151. if order.order_type != "gmcp":
  3152. raise ValueError("仅购买型订单可购买产品")
  3153. if order.order_status not in ("dsx", "sxz"):
  3154. raise ValueError("订单当前状态不允许购买")
  3155. if not products:
  3156. raise ValueError("产品明细不能为空")
  3157. now = datetime.now()
  3158. tn = tenant_no or order.tenant_no
  3159. total_amount = 0.0
  3160. # Upsert product holding rows
  3161. for p in products:
  3162. project_code = p.get("projectCode") or p.get("project_code") or ""
  3163. project_name = p.get("projectName") or p.get("project_name") or ""
  3164. cell_type = p.get("cellType") or p.get("cell_type") or ""
  3165. spec_text = p.get("specText") or p.get("spec_text") or ""
  3166. agreement_unit_price = float(p.get("agreementUnitPrice") or p.get("agreement_unit_price") or 0)
  3167. qty = float(p.get("qty") or p.get("buyQty") or p.get("buy_qty") or 0)
  3168. amt = float(p.get("amount") or agreement_unit_price * qty)
  3169. total_amount += amt
  3170. # Find or create product holding row
  3171. existing = db.query(BizChannelServiceOrderProduct).filter(
  3172. and_(
  3173. BizChannelServiceOrderProduct.service_order_id == order_id,
  3174. BizChannelServiceOrderProduct.project_code == project_code,
  3175. BizChannelServiceOrderProduct.spec_text == (spec_text or ""),
  3176. BizChannelServiceOrderProduct.del_flag == 0,
  3177. )
  3178. ).first()
  3179. if existing:
  3180. existing.total_buy_qty = float(existing.total_buy_qty or 0) + qty
  3181. existing.current_qty = float(existing.current_qty or 0) + qty
  3182. existing.update_by = user_id
  3183. existing.update_time = now
  3184. db.add(existing)
  3185. else:
  3186. prod_row = BizChannelServiceOrderProduct(
  3187. id=generate_uuid(),
  3188. service_order_id=order_id,
  3189. project_code=project_code,
  3190. project_name=project_name,
  3191. cell_type=cell_type,
  3192. spec_text=spec_text,
  3193. agreement_unit_price=agreement_unit_price,
  3194. buy_qty=qty,
  3195. buy_amount=amt,
  3196. total_buy_qty=qty,
  3197. total_unsubscribe_qty=0,
  3198. current_qty=qty,
  3199. total_unsubscribe_amount=0,
  3200. tenant_no=tn,
  3201. order_num=0,
  3202. remark="",
  3203. create_by=user_id,
  3204. create_dept_id="",
  3205. create_time=now,
  3206. update_by=user_id,
  3207. update_time=now,
  3208. del_flag=0,
  3209. )
  3210. db.add(prod_row)
  3211. db.flush()
  3212. products_json = json.dumps(products, ensure_ascii=False)
  3213. txn = BizChannelServiceOrderTxn(
  3214. id=generate_uuid(),
  3215. service_order_id=order_id,
  3216. txn_type="purchase",
  3217. amount=total_amount,
  3218. txn_status="dsx",
  3219. products_json=products_json,
  3220. remark=remark or "",
  3221. tenant_no=tn,
  3222. order_num=0,
  3223. create_by=user_id,
  3224. create_dept_id="",
  3225. create_time=now,
  3226. update_by=user_id,
  3227. update_time=now,
  3228. del_flag=0,
  3229. )
  3230. db.add(txn)
  3231. db.commit()
  3232. db.refresh(txn)
  3233. return txn
  3234. def create_unsubscribe_txn(db, order_id, *, products, remark=None, tenant_no=None, user_id=None):
  3235. """创建退订产品流水(gmcp订单),校验退订份数 <= current_qty"""
  3236. order = get_service_order(db, order_id, tenant_no)
  3237. if not order:
  3238. raise ValueError("订单不存在")
  3239. if order.order_type != "gmcp":
  3240. raise ValueError("仅购买型订单可退订产品")
  3241. if order.order_status != "sxz":
  3242. raise ValueError("仅生效中订单可退订")
  3243. if not products:
  3244. raise ValueError("退订产品明细不能为空")
  3245. now = datetime.now()
  3246. tn = tenant_no or order.tenant_no
  3247. total_amount = 0.0
  3248. for p in products:
  3249. project_code = p.get("projectCode") or p.get("project_code") or ""
  3250. spec_text = p.get("specText") or p.get("spec_text") or ""
  3251. qty = float(p.get("qty") or 0)
  3252. if qty <= 0:
  3253. raise ValueError("退订数量必须大于0")
  3254. # Find existing product holding
  3255. existing = db.query(BizChannelServiceOrderProduct).filter(
  3256. and_(
  3257. BizChannelServiceOrderProduct.service_order_id == order_id,
  3258. BizChannelServiceOrderProduct.project_code == project_code,
  3259. BizChannelServiceOrderProduct.spec_text == (spec_text or ""),
  3260. BizChannelServiceOrderProduct.del_flag == 0,
  3261. )
  3262. ).first()
  3263. if not existing:
  3264. raise ValueError(f"产品 {project_code} 不存在于持仓中")
  3265. current_qty = float(existing.current_qty or 0)
  3266. if qty > current_qty:
  3267. raise ValueError(f"退订数量 {qty} 超过当前持有数量 {current_qty}")
  3268. unit_price = float(existing.agreement_unit_price or 0)
  3269. amt = qty * unit_price
  3270. total_amount += amt
  3271. existing.total_unsubscribe_qty = float(existing.total_unsubscribe_qty or 0) + qty
  3272. existing.current_qty = current_qty - qty
  3273. existing.total_unsubscribe_amount = float(existing.total_unsubscribe_amount or 0) + amt
  3274. existing.update_by = user_id
  3275. existing.update_time = now
  3276. db.add(existing)
  3277. db.flush()
  3278. products_json = json.dumps(products, ensure_ascii=False)
  3279. txn = BizChannelServiceOrderTxn(
  3280. id=generate_uuid(),
  3281. service_order_id=order_id,
  3282. txn_type="unsubscribe",
  3283. amount=total_amount,
  3284. txn_status="dsx",
  3285. products_json=products_json,
  3286. remark=remark or "",
  3287. tenant_no=tn,
  3288. order_num=0,
  3289. create_by=user_id,
  3290. create_dept_id="",
  3291. create_time=now,
  3292. update_by=user_id,
  3293. update_time=now,
  3294. del_flag=0,
  3295. )
  3296. db.add(txn)
  3297. db.commit()
  3298. db.refresh(txn)
  3299. return txn
  3300. def update_service_order_txn(
  3301. db,
  3302. txn_id,
  3303. *,
  3304. amount=None,
  3305. remark=None,
  3306. finance_submit_attachments=None,
  3307. tenant_no=None,
  3308. user_id=None,
  3309. ):
  3310. """更新待生效流水:仅 dsx;充值/退款可改金额,购买/退订仅可改备注与凭证"""
  3311. txn = get_service_order_txn(db, txn_id, tenant_no=tenant_no)
  3312. if not txn:
  3313. raise ValueError("流水不存在")
  3314. if txn.txn_status != "dsx":
  3315. raise ValueError("仅待生效流水可编辑")
  3316. now = datetime.now()
  3317. txn_type = (txn.txn_type or "").strip()
  3318. if amount is not None:
  3319. amt = float(amount)
  3320. if amt <= 0:
  3321. raise ValueError("金额必须大于0")
  3322. if txn_type in ("recharge", "refund"):
  3323. if txn_type == "refund":
  3324. order = get_service_order(db, txn.service_order_id, tenant_no)
  3325. if not order:
  3326. raise ValueError("关联订单不存在")
  3327. current_balance = float(order.current_balance) if order.current_balance else 0
  3328. if amt > current_balance:
  3329. raise ValueError(f"退款金额不能大于当前余额 {current_balance}")
  3330. txn.amount = amt
  3331. else:
  3332. raise ValueError("购买/退订流水不可修改金额,请作废后重新发起")
  3333. if remark is not None:
  3334. txn.remark = remark or ""
  3335. if finance_submit_attachments is not None:
  3336. txn.finance_submit_attachments = (finance_submit_attachments or "").strip() or None
  3337. txn.update_by = user_id
  3338. txn.update_time = now
  3339. db.add(txn)
  3340. db.commit()
  3341. db.refresh(txn)
  3342. return txn
  3343. def submit_txn_finance(db, txn_id, *, finance_submit_remark=None, finance_submit_attachments=None, tenant_no=None, user_id=None):
  3344. """发起流水财务确认:dsx → dqr"""
  3345. txn = get_service_order_txn(db, txn_id, tenant_no=tenant_no)
  3346. if not txn:
  3347. raise ValueError("流水不存在")
  3348. if txn.txn_status != "dsx":
  3349. raise ValueError("仅待生效流水可发起确认")
  3350. now = datetime.now()
  3351. txn.txn_status = "dqr"
  3352. txn.finance_submit_by = user_id
  3353. txn.finance_submit_time = now
  3354. submit_remark = (finance_submit_remark or "").strip()
  3355. if not submit_remark:
  3356. submit_remark = (txn.remark or "").strip()
  3357. txn.finance_submit_remark = submit_remark
  3358. if finance_submit_attachments:
  3359. txn.finance_submit_attachments = finance_submit_attachments
  3360. elif not (txn.finance_submit_attachments or "").strip():
  3361. txn.finance_submit_attachments = ""
  3362. txn.update_by = user_id
  3363. txn.update_time = now
  3364. db.add(txn)
  3365. db.commit()
  3366. db.refresh(txn)
  3367. return txn
  3368. def confirm_txn(db, txn_id, *, finance_remark=None, finance_confirm_attachments=None, tenant_no=None, user_id=None):
  3369. """财务确认流水:dqr → yqr,更新主表汇总和产品持仓,更新渠道账户余额"""
  3370. txn = get_service_order_txn(db, txn_id, tenant_no=tenant_no)
  3371. if not txn:
  3372. raise ValueError("流水不存在")
  3373. if txn.txn_status != "dqr":
  3374. raise ValueError("仅待确认流水可进行财务确认")
  3375. order = get_service_order(db, txn.service_order_id, tenant_no)
  3376. if not order:
  3377. raise ValueError("关联订单不存在")
  3378. now = datetime.now()
  3379. if txn.txn_type == "recharge":
  3380. # Increase channel account balance
  3381. account = _get_or_create_channel_account(db, order.channel_id, tenant_no or order.tenant_no, user_id)
  3382. balance_before = float(account.balance_amount) if account.balance_amount else 0
  3383. add_amount = float(txn.amount) if txn.amount else 0
  3384. account.balance_amount = balance_before + add_amount
  3385. balance_after = float(account.balance_amount)
  3386. account.update_by = user_id
  3387. account.update_time = now
  3388. db.add(account)
  3389. db.flush()
  3390. # Update order summary
  3391. order.total_recharge_amount = float(order.total_recharge_amount or 0) + add_amount
  3392. order.current_balance = float(order.current_balance or 0) + add_amount
  3393. txn.balance_before = balance_before
  3394. txn.balance_after = balance_after
  3395. elif txn.txn_type == "refund":
  3396. # Deduct channel account balance
  3397. account = _get_or_create_channel_account(db, order.channel_id, tenant_no or order.tenant_no, user_id)
  3398. balance_before = float(account.balance_amount) if account.balance_amount else 0
  3399. deduct_amount = float(txn.amount) if txn.amount else 0
  3400. account.balance_amount = balance_before - deduct_amount
  3401. balance_after = float(account.balance_amount)
  3402. account.update_by = user_id
  3403. account.update_time = now
  3404. db.add(account)
  3405. db.flush()
  3406. # Update order summary
  3407. order.total_refund_amount = float(order.total_refund_amount or 0) + deduct_amount
  3408. order.current_balance = float(order.current_balance or 0) - deduct_amount
  3409. txn.balance_before = balance_before
  3410. txn.balance_after = balance_after
  3411. elif txn.txn_type == "purchase":
  3412. # Products already updated at create time, just update channel quotas
  3413. products = json.loads(txn.products_json) if txn.products_json else []
  3414. for p in products:
  3415. project_code = p.get("projectCode") or p.get("project_code") or ""
  3416. spec_text = p.get("specText") or p.get("spec_text") or ""
  3417. qty = float(p.get("qty") or p.get("buyQty") or p.get("buy_qty") or 0)
  3418. cell_type_code, _ = _resolve_agreement_price_cell_type(
  3419. db, order.channel_agreement_id, project_code, spec_text, p.get("cellType") or p.get("cell_type") or ""
  3420. )
  3421. _upsert_channel_product_quota(
  3422. db,
  3423. channel_id=order.channel_id,
  3424. project_code=project_code,
  3425. project_name=p.get("projectName") or p.get("project_name") or "",
  3426. cell_type_code=cell_type_code,
  3427. cell_type_name=p.get("cellType") or p.get("cell_type") or "",
  3428. spec_text=spec_text,
  3429. add_qty=qty,
  3430. tenant_no=tenant_no or order.tenant_no,
  3431. user_id=user_id,
  3432. )
  3433. elif txn.txn_type == "unsubscribe":
  3434. # Products already updated at create time, just reduce channel quotas
  3435. products = json.loads(txn.products_json) if txn.products_json else []
  3436. for p in products:
  3437. project_code = p.get("projectCode") or p.get("project_code") or ""
  3438. spec_text = p.get("specText") or p.get("spec_text") or ""
  3439. qty = float(p.get("qty") or 0)
  3440. cell_type_code, _ = _resolve_agreement_price_cell_type(
  3441. db, order.channel_agreement_id, project_code, spec_text, p.get("cellType") or p.get("cell_type") or ""
  3442. )
  3443. _reduce_channel_product_quota(
  3444. db,
  3445. channel_id=order.channel_id,
  3446. project_code=project_code,
  3447. cell_type_code=cell_type_code,
  3448. spec_text=spec_text,
  3449. reduce_qty=qty,
  3450. user_id=user_id,
  3451. )
  3452. # Update txn status
  3453. txn.txn_status = "yqr"
  3454. txn.finance_confirm_time = now
  3455. txn.finance_confirm_by = user_id
  3456. txn.finance_confirm_attachments = finance_confirm_attachments or ""
  3457. txn.finance_remark = finance_remark or ""
  3458. txn.update_by = user_id
  3459. txn.update_time = now
  3460. # Transition order to sxz if currently dsx
  3461. if order.order_status == "dsx":
  3462. order.order_status = "sxz"
  3463. order.update_by = user_id
  3464. order.update_time = now
  3465. db.add(txn)
  3466. db.add(order)
  3467. db.commit()
  3468. db.refresh(txn)
  3469. return txn
  3470. def void_txn(db, txn_id, *, tenant_no=None, user_id=None):
  3471. """作废流水:dsx/dqr → yzf。如果是 purchase/unsubscribe 还需要回滚产品持仓"""
  3472. txn = get_service_order_txn(db, txn_id, tenant_no=tenant_no)
  3473. if not txn:
  3474. raise ValueError("流水不存在")
  3475. if txn.txn_status not in ("dsx", "dqr"):
  3476. raise ValueError("仅待生效或待财务确认流水可作废")
  3477. now = datetime.now()
  3478. order = get_service_order(db, txn.service_order_id, tenant_no)
  3479. if txn.txn_type == "purchase" and order:
  3480. # Rollback product holdings
  3481. products = json.loads(txn.products_json) if txn.products_json else []
  3482. for p in products:
  3483. project_code = p.get("projectCode") or p.get("project_code") or ""
  3484. spec_text = p.get("specText") or p.get("spec_text") or ""
  3485. qty = float(p.get("qty") or p.get("buyQty") or p.get("buy_qty") or 0)
  3486. existing = db.query(BizChannelServiceOrderProduct).filter(
  3487. and_(
  3488. BizChannelServiceOrderProduct.service_order_id == order.id,
  3489. BizChannelServiceOrderProduct.project_code == project_code,
  3490. BizChannelServiceOrderProduct.spec_text == (spec_text or ""),
  3491. BizChannelServiceOrderProduct.del_flag == 0,
  3492. )
  3493. ).first()
  3494. if existing:
  3495. existing.total_buy_qty = float(existing.total_buy_qty or 0) - qty
  3496. existing.current_qty = float(existing.current_qty or 0) - qty
  3497. existing.update_by = user_id
  3498. existing.update_time = now
  3499. db.add(existing)
  3500. elif txn.txn_type == "unsubscribe" and order:
  3501. # Rollback unsubscribe: restore product holdings
  3502. products = json.loads(txn.products_json) if txn.products_json else []
  3503. for p in products:
  3504. project_code = p.get("projectCode") or p.get("project_code") or ""
  3505. spec_text = p.get("specText") or p.get("spec_text") or ""
  3506. qty = float(p.get("qty") or 0)
  3507. unit_price = 0
  3508. existing = db.query(BizChannelServiceOrderProduct).filter(
  3509. and_(
  3510. BizChannelServiceOrderProduct.service_order_id == order.id,
  3511. BizChannelServiceOrderProduct.project_code == project_code,
  3512. BizChannelServiceOrderProduct.spec_text == (spec_text or ""),
  3513. BizChannelServiceOrderProduct.del_flag == 0,
  3514. )
  3515. ).first()
  3516. if existing:
  3517. unit_price = float(existing.agreement_unit_price or 0)
  3518. existing.total_unsubscribe_qty = float(existing.total_unsubscribe_qty or 0) - qty
  3519. existing.current_qty = float(existing.current_qty or 0) + qty
  3520. existing.total_unsubscribe_amount = float(existing.total_unsubscribe_amount or 0) - (qty * unit_price)
  3521. existing.update_by = user_id
  3522. existing.update_time = now
  3523. db.add(existing)
  3524. txn.txn_status = "yzf"
  3525. txn.void_by = user_id
  3526. txn.void_time = now
  3527. txn.update_by = user_id
  3528. txn.update_time = now
  3529. db.add(txn)
  3530. db.commit()
  3531. db.refresh(txn)
  3532. return txn