pilot_service.py 62 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705
  1. """中试服务管理 API(细胞产品类型/规格/价格体系/项目价格、渠道信息、体检机构)"""
  2. from typing import Any, List, Optional
  3. from fastapi import APIRouter, Depends, Query
  4. from pydantic import BaseModel, ConfigDict, Field
  5. from sqlalchemy.orm import Session
  6. from app.core.database import get_db
  7. from app.core.menu_guard import require_admin_or_super_admin
  8. from app.core.response import error_response, success_response
  9. from app.crud import pilot_service as ps_svc
  10. from app.models.models import SysUser
  11. router = APIRouter(prefix="/web/admin", tags=["中试服务管理"])
  12. def _get_tenant_no(db: Session, user: dict) -> str:
  13. user_id = user.get("user_id")
  14. if not user_id:
  15. raise PermissionError("用户身份信息缺失,无法确定租户")
  16. sys_user = db.query(SysUser).filter(SysUser.id == user_id).first()
  17. if not sys_user or not sys_user.tenant_no:
  18. raise PermissionError("用户租户信息缺失")
  19. return sys_user.tenant_no
  20. # ---------------------------------------------------------------------------
  21. # 1. 细胞产品类型
  22. # ---------------------------------------------------------------------------
  23. class CellProductTypeCreateBody(BaseModel):
  24. model_config = ConfigDict(populate_by_name=True)
  25. product_type_code: str = Field(..., alias="productTypeCode")
  26. product_type_name: str = Field(..., alias="productTypeName")
  27. order_num: Optional[int] = Field(0, alias="orderNum")
  28. remark: Optional[str] = None
  29. class CellProductTypeUpdateBody(BaseModel):
  30. model_config = ConfigDict(populate_by_name=True)
  31. product_type_code: Optional[str] = Field(None, alias="productTypeCode")
  32. product_type_name: Optional[str] = Field(None, alias="productTypeName")
  33. order_num: Optional[int] = Field(None, alias="orderNum")
  34. remark: Optional[str] = None
  35. @router.get("/cell-product-types")
  36. async def list_cell_product_types(
  37. page: int = Query(1, ge=1),
  38. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  39. keyword: Optional[str] = None,
  40. user: dict = Depends(require_admin_or_super_admin),
  41. db: Session = Depends(get_db),
  42. ):
  43. tenant_no = _get_tenant_no(db, user)
  44. items, total = ps_svc.list_cell_product_types(
  45. db, page=page, page_size=pageSize, keyword=keyword, tenant_no=tenant_no, user_info=user
  46. )
  47. tp = ps_svc.total_pages(total, pageSize)
  48. return success_response(
  49. {"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp}
  50. )
  51. @router.post("/cell-product-types")
  52. async def create_cell_product_type(
  53. body: CellProductTypeCreateBody,
  54. user: dict = Depends(require_admin_or_super_admin),
  55. db: Session = Depends(get_db),
  56. ):
  57. try:
  58. tenant_no = _get_tenant_no(db, user)
  59. obj = ps_svc.create_cell_product_type(
  60. db,
  61. product_type_code=body.product_type_code,
  62. product_type_name=body.product_type_name,
  63. tenant_no=tenant_no,
  64. order_num=body.order_num,
  65. remark=body.remark,
  66. user_id=user.get("user_id"),
  67. )
  68. return success_response(ps_svc._row_to_dict_cell_product_type(obj), message="创建成功")
  69. except ValueError as e:
  70. return error_response(400, str(e))
  71. @router.put("/cell-product-types/{row_id}")
  72. async def update_cell_product_type(
  73. row_id: str,
  74. body: CellProductTypeUpdateBody,
  75. user: dict = Depends(require_admin_or_super_admin),
  76. db: Session = Depends(get_db),
  77. ):
  78. try:
  79. tenant_no = _get_tenant_no(db, user)
  80. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  81. obj = ps_svc.update_cell_product_type(
  82. db,
  83. row_id,
  84. product_type_code=payload.get("product_type_code"),
  85. product_type_name=payload.get("product_type_name"),
  86. order_num=payload.get("order_num"),
  87. remark=payload.get("remark"),
  88. user_id=user.get("user_id"),
  89. tenant_no=tenant_no,
  90. )
  91. return success_response(ps_svc._row_to_dict_cell_product_type(obj), message="更新成功")
  92. except ValueError as e:
  93. return error_response(400, str(e))
  94. @router.delete("/cell-product-types/{row_id}")
  95. async def delete_cell_product_type(
  96. row_id: str,
  97. user: dict = Depends(require_admin_or_super_admin),
  98. db: Session = Depends(get_db),
  99. ):
  100. try:
  101. tenant_no = _get_tenant_no(db, user)
  102. ps_svc.soft_delete_cell_product_type(db, row_id, tenant_no=tenant_no)
  103. return success_response(None, message="删除成功")
  104. except ValueError as e:
  105. return error_response(400, str(e))
  106. # ---------------------------------------------------------------------------
  107. # 2. 细胞产品规格
  108. # ---------------------------------------------------------------------------
  109. class CellProductSpecCreateBody(BaseModel):
  110. model_config = ConfigDict(populate_by_name=True)
  111. spec_code: str = Field(..., alias="specCode")
  112. spec_name: str = Field(..., alias="specName")
  113. order_num: Optional[int] = Field(0, alias="orderNum")
  114. remark: Optional[str] = None
  115. class CellProductSpecUpdateBody(BaseModel):
  116. model_config = ConfigDict(populate_by_name=True)
  117. spec_code: Optional[str] = Field(None, alias="specCode")
  118. spec_name: Optional[str] = Field(None, alias="specName")
  119. order_num: Optional[int] = Field(None, alias="orderNum")
  120. remark: Optional[str] = None
  121. @router.get("/cell-product-specs")
  122. async def list_cell_product_specs(
  123. page: int = Query(1, ge=1),
  124. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  125. keyword: Optional[str] = None,
  126. user: dict = Depends(require_admin_or_super_admin),
  127. db: Session = Depends(get_db),
  128. ):
  129. tenant_no = _get_tenant_no(db, user)
  130. items, total = ps_svc.list_cell_product_specs(
  131. db, page=page, page_size=pageSize, keyword=keyword, tenant_no=tenant_no, user_info=user
  132. )
  133. tp = ps_svc.total_pages(total, pageSize)
  134. return success_response(
  135. {"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp}
  136. )
  137. @router.post("/cell-product-specs")
  138. async def create_cell_product_spec(
  139. body: CellProductSpecCreateBody,
  140. user: dict = Depends(require_admin_or_super_admin),
  141. db: Session = Depends(get_db),
  142. ):
  143. try:
  144. tenant_no = _get_tenant_no(db, user)
  145. obj = ps_svc.create_cell_product_spec(
  146. db,
  147. spec_code=body.spec_code,
  148. spec_name=body.spec_name,
  149. tenant_no=tenant_no,
  150. order_num=body.order_num,
  151. remark=body.remark,
  152. user_id=user.get("user_id"),
  153. )
  154. return success_response(ps_svc._row_to_dict_cell_product_spec(obj), message="创建成功")
  155. except ValueError as e:
  156. return error_response(400, str(e))
  157. @router.put("/cell-product-specs/{row_id}")
  158. async def update_cell_product_spec(
  159. row_id: str,
  160. body: CellProductSpecUpdateBody,
  161. user: dict = Depends(require_admin_or_super_admin),
  162. db: Session = Depends(get_db),
  163. ):
  164. try:
  165. tenant_no = _get_tenant_no(db, user)
  166. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  167. obj = ps_svc.update_cell_product_spec(
  168. db,
  169. row_id,
  170. spec_code=payload.get("spec_code"),
  171. spec_name=payload.get("spec_name"),
  172. order_num=payload.get("order_num"),
  173. remark=payload.get("remark"),
  174. user_id=user.get("user_id"),
  175. tenant_no=tenant_no,
  176. )
  177. return success_response(ps_svc._row_to_dict_cell_product_spec(obj), message="更新成功")
  178. except ValueError as e:
  179. return error_response(400, str(e))
  180. @router.delete("/cell-product-specs/{row_id}")
  181. async def delete_cell_product_spec(
  182. row_id: str,
  183. user: dict = Depends(require_admin_or_super_admin),
  184. db: Session = Depends(get_db),
  185. ):
  186. try:
  187. tenant_no = _get_tenant_no(db, user)
  188. ps_svc.soft_delete_cell_product_spec(db, row_id, tenant_no=tenant_no)
  189. return success_response(None, message="删除成功")
  190. except ValueError as e:
  191. return error_response(400, str(e))
  192. # ---------------------------------------------------------------------------
  193. # 3. 细胞产品价格
  194. # ---------------------------------------------------------------------------
  195. class CellProductPriceCreateBody(BaseModel):
  196. model_config = ConfigDict(populate_by_name=True)
  197. project_name: str = Field(..., alias="projectName")
  198. project_code: str = Field(..., alias="projectCode")
  199. cell_type_code: Optional[str] = Field(None, alias="cellTypeCode")
  200. cell_type_name: Optional[str] = Field(None, alias="cellTypeName")
  201. spec_text: Optional[str] = Field(None, alias="specText")
  202. spec_id: Optional[str] = Field(None, alias="specId")
  203. cell_product_type_id: Optional[str] = Field(None, alias="cellProductTypeId")
  204. unit_price: Optional[float] = Field(None, alias="unitPrice")
  205. order_num: Optional[int] = Field(0, alias="orderNum")
  206. remark: Optional[str] = None
  207. class CellProductPriceUpdateBody(BaseModel):
  208. model_config = ConfigDict(populate_by_name=True)
  209. project_name: Optional[str] = Field(None, alias="projectName")
  210. project_code: Optional[str] = Field(None, alias="projectCode")
  211. cell_type_code: Optional[str] = Field(None, alias="cellTypeCode")
  212. cell_type_name: Optional[str] = Field(None, alias="cellTypeName")
  213. spec_text: Optional[str] = Field(None, alias="specText")
  214. spec_id: Optional[str] = Field(None, alias="specId")
  215. cell_product_type_id: Optional[str] = Field(None, alias="cellProductTypeId")
  216. unit_price: Optional[float] = Field(None, alias="unitPrice")
  217. order_num: Optional[int] = Field(None, alias="orderNum")
  218. remark: Optional[str] = None
  219. @router.get("/cell-product-prices")
  220. async def list_cell_product_prices(
  221. page: int = Query(1, ge=1),
  222. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  223. keyword: Optional[str] = None,
  224. cell_product_type_id: Optional[str] = Query(None, alias="cellProductTypeId"),
  225. user: dict = Depends(require_admin_or_super_admin),
  226. db: Session = Depends(get_db),
  227. ):
  228. tenant_no = _get_tenant_no(db, user)
  229. items, total = ps_svc.list_cell_product_prices(
  230. db,
  231. page=page,
  232. page_size=pageSize,
  233. keyword=keyword,
  234. tenant_no=tenant_no,
  235. cell_product_type_id=cell_product_type_id,
  236. user_info=user,
  237. )
  238. tp = ps_svc.total_pages(total, pageSize)
  239. return success_response(
  240. {"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp}
  241. )
  242. @router.post("/cell-product-prices")
  243. async def create_cell_product_price(
  244. body: CellProductPriceCreateBody,
  245. user: dict = Depends(require_admin_or_super_admin),
  246. db: Session = Depends(get_db),
  247. ):
  248. try:
  249. tenant_no = _get_tenant_no(db, user)
  250. obj = ps_svc.create_cell_product_price(
  251. db,
  252. project_name=body.project_name,
  253. project_code=body.project_code,
  254. cell_type_code=body.cell_type_code or "",
  255. cell_type_name=body.cell_type_name or "",
  256. spec_text=body.spec_text or "",
  257. spec_id=body.spec_id or "",
  258. cell_product_type_id=body.cell_product_type_id or "",
  259. unit_price=body.unit_price,
  260. tenant_no=tenant_no,
  261. order_num=body.order_num,
  262. remark=body.remark,
  263. user_id=user.get("user_id"),
  264. )
  265. return success_response(ps_svc._row_to_dict_cell_product_price(obj), message="创建成功")
  266. except ValueError as e:
  267. return error_response(400, str(e))
  268. @router.put("/cell-product-prices/{row_id}")
  269. async def update_cell_product_price(
  270. row_id: str,
  271. body: CellProductPriceUpdateBody,
  272. user: dict = Depends(require_admin_or_super_admin),
  273. db: Session = Depends(get_db),
  274. ):
  275. try:
  276. tenant_no = _get_tenant_no(db, user)
  277. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  278. obj = ps_svc.update_cell_product_price(
  279. db,
  280. row_id,
  281. project_name=payload.get("project_name"),
  282. project_code=payload.get("project_code"),
  283. cell_type_code=payload.get("cell_type_code"),
  284. cell_type_name=payload.get("cell_type_name"),
  285. spec_text=payload.get("spec_text"),
  286. spec_id=payload.get("spec_id"),
  287. cell_product_type_id=payload.get("cell_product_type_id"),
  288. unit_price=payload.get("unit_price"),
  289. order_num=payload.get("order_num"),
  290. remark=payload.get("remark"),
  291. user_id=user.get("user_id"),
  292. tenant_no=tenant_no,
  293. )
  294. return success_response(ps_svc._row_to_dict_cell_product_price(obj), message="更新成功")
  295. except ValueError as e:
  296. return error_response(400, str(e))
  297. @router.delete("/cell-product-prices/{row_id}")
  298. async def delete_cell_product_price(
  299. row_id: str,
  300. user: dict = Depends(require_admin_or_super_admin),
  301. db: Session = Depends(get_db),
  302. ):
  303. try:
  304. tenant_no = _get_tenant_no(db, user)
  305. ps_svc.soft_delete_cell_product_price(db, row_id, tenant_no=tenant_no)
  306. return success_response(None, message="删除成功")
  307. except ValueError as e:
  308. return error_response(400, str(e))
  309. # ---------------------------------------------------------------------------
  310. # 4. 项目价格体系(主表 + 明细子表)
  311. # ---------------------------------------------------------------------------
  312. class ProjectSystemCreateBody(BaseModel):
  313. model_config = ConfigDict(populate_by_name=True)
  314. system_name: str = Field(..., alias="systemName")
  315. system_code: str = Field(..., alias="systemCode")
  316. status: Optional[str] = Field("1", alias="status")
  317. order_num: Optional[int] = Field(0, alias="orderNum")
  318. remark: Optional[str] = None
  319. class ProjectSystemUpdateBody(BaseModel):
  320. model_config = ConfigDict(populate_by_name=True)
  321. system_name: Optional[str] = Field(None, alias="systemName")
  322. system_code: Optional[str] = Field(None, alias="systemCode")
  323. status: Optional[str] = Field(None, alias="status")
  324. order_num: Optional[int] = Field(None, alias="orderNum")
  325. remark: Optional[str] = None
  326. @router.get("/project-price-systems")
  327. async def list_project_systems(
  328. page: int = Query(1, ge=1),
  329. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  330. keyword: Optional[str] = None,
  331. user: dict = Depends(require_admin_or_super_admin),
  332. db: Session = Depends(get_db),
  333. ):
  334. tenant_no = _get_tenant_no(db, user)
  335. items, total = ps_svc.list_project_systems(
  336. db, page=page, page_size=pageSize, keyword=keyword, tenant_no=tenant_no, user_info=user
  337. )
  338. tp = ps_svc.total_pages(total, pageSize)
  339. return success_response(
  340. {"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp}
  341. )
  342. @router.post("/project-price-systems")
  343. async def create_project_system(
  344. body: ProjectSystemCreateBody,
  345. user: dict = Depends(require_admin_or_super_admin),
  346. db: Session = Depends(get_db),
  347. ):
  348. try:
  349. tenant_no = _get_tenant_no(db, user)
  350. obj = ps_svc.create_project_system(
  351. db,
  352. system_name=body.system_name,
  353. system_code=body.system_code,
  354. status=body.status,
  355. tenant_no=tenant_no,
  356. order_num=body.order_num,
  357. remark=body.remark,
  358. user_id=user.get("user_id"),
  359. )
  360. return success_response(ps_svc._row_to_dict_project_system(obj), message="创建成功")
  361. except ValueError as e:
  362. return error_response(400, str(e))
  363. @router.put("/project-price-systems/{row_id}")
  364. async def update_project_system(
  365. row_id: str,
  366. body: ProjectSystemUpdateBody,
  367. user: dict = Depends(require_admin_or_super_admin),
  368. db: Session = Depends(get_db),
  369. ):
  370. try:
  371. tenant_no = _get_tenant_no(db, user)
  372. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  373. obj = ps_svc.update_project_system(
  374. db,
  375. row_id,
  376. system_name=payload.get("system_name"),
  377. system_code=payload.get("system_code"),
  378. status=payload.get("status"),
  379. order_num=payload.get("order_num"),
  380. remark=payload.get("remark"),
  381. user_id=user.get("user_id"),
  382. tenant_no=tenant_no,
  383. )
  384. return success_response(ps_svc._row_to_dict_project_system(obj), message="更新成功")
  385. except ValueError as e:
  386. return error_response(400, str(e))
  387. @router.delete("/project-price-systems/{row_id}")
  388. async def delete_project_system(
  389. row_id: str,
  390. user: dict = Depends(require_admin_or_super_admin),
  391. db: Session = Depends(get_db),
  392. ):
  393. try:
  394. tenant_no = _get_tenant_no(db, user)
  395. ps_svc.soft_delete_project_system(db, row_id, tenant_no=tenant_no)
  396. return success_response(None, message="删除成功")
  397. except ValueError as e:
  398. return error_response(400, str(e))
  399. # --- 项目价格体系明细 ---
  400. class ProjectSystemItemCreateBody(BaseModel):
  401. model_config = ConfigDict(populate_by_name=True)
  402. source_price_id: Optional[str] = Field(None, alias="sourcePriceId")
  403. project_name: str = Field(..., alias="projectName")
  404. project_code: str = Field(..., alias="projectCode")
  405. cell_type_code: Optional[str] = Field(None, alias="cellTypeCode")
  406. cell_type_name: Optional[str] = Field(None, alias="cellTypeName")
  407. spec_text: Optional[str] = Field(None, alias="specText")
  408. spec_id: Optional[str] = Field(None, alias="specId")
  409. cell_product_type_id: Optional[str] = Field(None, alias="cellProductTypeId")
  410. unit_price: Optional[float] = Field(None, alias="unitPrice")
  411. order_num: Optional[int] = Field(0, alias="orderNum")
  412. remark: Optional[str] = None
  413. class ProjectSystemItemUpdateBody(BaseModel):
  414. model_config = ConfigDict(populate_by_name=True)
  415. source_price_id: Optional[str] = Field(None, alias="sourcePriceId")
  416. project_name: Optional[str] = Field(None, alias="projectName")
  417. project_code: Optional[str] = Field(None, alias="projectCode")
  418. cell_type_code: Optional[str] = Field(None, alias="cellTypeCode")
  419. cell_type_name: Optional[str] = Field(None, alias="cellTypeName")
  420. spec_text: Optional[str] = Field(None, alias="specText")
  421. spec_id: Optional[str] = Field(None, alias="specId")
  422. cell_product_type_id: Optional[str] = Field(None, alias="cellProductTypeId")
  423. unit_price: Optional[float] = Field(None, alias="unitPrice")
  424. order_num: Optional[int] = Field(None, alias="orderNum")
  425. remark: Optional[str] = None
  426. @router.get("/project-price-systems/{system_id}/items")
  427. async def list_project_system_items(
  428. system_id: str,
  429. user: dict = Depends(require_admin_or_super_admin),
  430. db: Session = Depends(get_db),
  431. ):
  432. try:
  433. tenant_no = _get_tenant_no(db, user)
  434. items = ps_svc.list_project_system_items(db, system_id, tenant_no=tenant_no, user_info=user)
  435. return success_response(items)
  436. except ValueError as e:
  437. return error_response(400, str(e))
  438. @router.post("/project-price-systems/{system_id}/items")
  439. async def create_project_system_item(
  440. system_id: str,
  441. body: ProjectSystemItemCreateBody,
  442. user: dict = Depends(require_admin_or_super_admin),
  443. db: Session = Depends(get_db),
  444. ):
  445. try:
  446. tenant_no = _get_tenant_no(db, user)
  447. obj = ps_svc.create_project_system_item(
  448. db,
  449. project_system_id=system_id,
  450. source_price_id=body.source_price_id or "",
  451. project_name=body.project_name,
  452. project_code=body.project_code,
  453. cell_type_code=body.cell_type_code or "",
  454. cell_type_name=body.cell_type_name or "",
  455. spec_text=body.spec_text or "",
  456. spec_id=body.spec_id or "",
  457. cell_product_type_id=body.cell_product_type_id or "",
  458. unit_price=body.unit_price,
  459. tenant_no=tenant_no,
  460. order_num=body.order_num,
  461. remark=body.remark,
  462. user_id=user.get("user_id"),
  463. )
  464. return success_response(ps_svc._row_to_dict_project_system_item(obj), message="创建成功")
  465. except ValueError as e:
  466. return error_response(400, str(e))
  467. @router.put("/project-system-items/{item_id}")
  468. async def update_project_system_item(
  469. item_id: str,
  470. body: ProjectSystemItemUpdateBody,
  471. user: dict = Depends(require_admin_or_super_admin),
  472. db: Session = Depends(get_db),
  473. ):
  474. try:
  475. tenant_no = _get_tenant_no(db, user)
  476. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  477. obj = ps_svc.update_project_system_item(
  478. db,
  479. item_id,
  480. source_price_id=payload.get("source_price_id"),
  481. project_name=payload.get("project_name"),
  482. project_code=payload.get("project_code"),
  483. cell_type_code=payload.get("cell_type_code"),
  484. cell_type_name=payload.get("cell_type_name"),
  485. spec_text=payload.get("spec_text"),
  486. spec_id=payload.get("spec_id"),
  487. cell_product_type_id=payload.get("cell_product_type_id"),
  488. unit_price=payload.get("unit_price"),
  489. order_num=payload.get("order_num"),
  490. remark=payload.get("remark"),
  491. user_id=user.get("user_id"),
  492. tenant_no=tenant_no,
  493. )
  494. return success_response(ps_svc._row_to_dict_project_system_item(obj), message="更新成功")
  495. except ValueError as e:
  496. return error_response(400, str(e))
  497. @router.delete("/project-system-items/{item_id}")
  498. async def delete_project_system_item(
  499. item_id: str,
  500. user: dict = Depends(require_admin_or_super_admin),
  501. db: Session = Depends(get_db),
  502. ):
  503. try:
  504. tenant_no = _get_tenant_no(db, user)
  505. ps_svc.soft_delete_project_system_item(db, item_id, tenant_no=tenant_no)
  506. return success_response(None, message="删除成功")
  507. except ValueError as e:
  508. return error_response(400, str(e))
  509. # ---------------------------------------------------------------------------
  510. # 5. 渠道信息(主表 + 开票/邮寄地址/协议/协议产品价格子表)
  511. # ---------------------------------------------------------------------------
  512. class ChannelCreateBody(BaseModel):
  513. model_config = ConfigDict(populate_by_name=True)
  514. channel_name: str = Field(..., alias="channelName")
  515. credit_code: Optional[str] = Field(None, alias="creditCode")
  516. channel_type: Optional[str] = Field(None, alias="channelType")
  517. contact_name: Optional[str] = Field(None, alias="contactName")
  518. contact_phone: Optional[str] = Field(None, alias="contactPhone")
  519. contact_email: Optional[str] = Field(None, alias="contactEmail")
  520. protocol_attachments: Optional[str] = Field(None, alias="protocolAttachments")
  521. order_num: Optional[int] = Field(0, alias="orderNum")
  522. remark: Optional[str] = None
  523. class ChannelUpdateBody(BaseModel):
  524. model_config = ConfigDict(populate_by_name=True)
  525. channel_name: Optional[str] = Field(None, alias="channelName")
  526. credit_code: Optional[str] = Field(None, alias="creditCode")
  527. channel_type: Optional[str] = Field(None, alias="channelType")
  528. contact_name: Optional[str] = Field(None, alias="contactName")
  529. contact_phone: Optional[str] = Field(None, alias="contactPhone")
  530. contact_email: Optional[str] = Field(None, alias="contactEmail")
  531. protocol_attachments: Optional[str] = Field(None, alias="protocolAttachments")
  532. order_num: Optional[int] = Field(None, alias="orderNum")
  533. remark: Optional[str] = None
  534. @router.get("/channels")
  535. async def list_channels(
  536. page: int = Query(1, ge=1),
  537. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  538. keyword: Optional[str] = None,
  539. user: dict = Depends(require_admin_or_super_admin),
  540. db: Session = Depends(get_db),
  541. ):
  542. tenant_no = _get_tenant_no(db, user)
  543. items, total = ps_svc.list_channels(
  544. db, page=page, page_size=pageSize, keyword=keyword, tenant_no=tenant_no, user_info=user
  545. )
  546. tp = ps_svc.total_pages(total, pageSize)
  547. return success_response(
  548. {"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp}
  549. )
  550. @router.post("/channels")
  551. async def create_channel(
  552. body: ChannelCreateBody,
  553. user: dict = Depends(require_admin_or_super_admin),
  554. db: Session = Depends(get_db),
  555. ):
  556. try:
  557. tenant_no = _get_tenant_no(db, user)
  558. obj = ps_svc.create_channel(
  559. db,
  560. channel_name=body.channel_name,
  561. credit_code=body.credit_code,
  562. channel_type=body.channel_type,
  563. contact_name=body.contact_name,
  564. contact_phone=body.contact_phone,
  565. contact_email=body.contact_email,
  566. protocol_attachments=body.protocol_attachments,
  567. tenant_no=tenant_no,
  568. order_num=body.order_num,
  569. remark=body.remark,
  570. user_id=user.get("user_id"),
  571. )
  572. return success_response(
  573. ps_svc._row_to_dict_channel(obj, db, tenant_no),
  574. message="创建成功",
  575. )
  576. except ValueError as e:
  577. return error_response(400, str(e))
  578. @router.put("/channels/{row_id}")
  579. async def update_channel(
  580. row_id: str,
  581. body: ChannelUpdateBody,
  582. user: dict = Depends(require_admin_or_super_admin),
  583. db: Session = Depends(get_db),
  584. ):
  585. try:
  586. tenant_no = _get_tenant_no(db, user)
  587. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  588. obj = ps_svc.update_channel(
  589. db,
  590. row_id,
  591. channel_name=payload.get("channel_name"),
  592. credit_code=payload.get("credit_code"),
  593. channel_type=payload.get("channel_type"),
  594. contact_name=payload.get("contact_name"),
  595. contact_phone=payload.get("contact_phone"),
  596. contact_email=payload.get("contact_email"),
  597. protocol_attachments=payload.get("protocol_attachments"),
  598. order_num=payload.get("order_num"),
  599. remark=payload.get("remark"),
  600. user_id=user.get("user_id"),
  601. tenant_no=tenant_no,
  602. )
  603. return success_response(
  604. ps_svc._row_to_dict_channel(obj, db, tenant_no),
  605. message="更新成功",
  606. )
  607. except ValueError as e:
  608. return error_response(400, str(e))
  609. @router.delete("/channels/{row_id}")
  610. async def delete_channel(
  611. row_id: str,
  612. user: dict = Depends(require_admin_or_super_admin),
  613. db: Session = Depends(get_db),
  614. ):
  615. try:
  616. tenant_no = _get_tenant_no(db, user)
  617. ps_svc.soft_delete_channel(db, row_id, tenant_no=tenant_no)
  618. return success_response(None, message="删除成功")
  619. except ValueError as e:
  620. return error_response(400, str(e))
  621. # --- 渠道开票信息 ---
  622. class ChannelInvoiceCreateBody(BaseModel):
  623. model_config = ConfigDict(populate_by_name=True)
  624. company_name: str = Field(..., alias="companyName")
  625. tax_no: Optional[str] = Field(None, alias="taxNo")
  626. company_address: Optional[str] = Field(None, alias="companyAddress")
  627. company_phone: Optional[str] = Field(None, alias="companyPhone")
  628. bank_name: Optional[str] = Field(None, alias="bankName")
  629. bank_account: Optional[str] = Field(None, alias="bankAccount")
  630. order_num: Optional[int] = Field(0, alias="orderNum")
  631. remark: Optional[str] = None
  632. class ChannelInvoiceUpdateBody(BaseModel):
  633. model_config = ConfigDict(populate_by_name=True)
  634. company_name: Optional[str] = Field(None, alias="companyName")
  635. tax_no: Optional[str] = Field(None, alias="taxNo")
  636. company_address: Optional[str] = Field(None, alias="companyAddress")
  637. company_phone: Optional[str] = Field(None, alias="companyPhone")
  638. bank_name: Optional[str] = Field(None, alias="bankName")
  639. bank_account: Optional[str] = Field(None, alias="bankAccount")
  640. order_num: Optional[int] = Field(None, alias="orderNum")
  641. remark: Optional[str] = None
  642. @router.get("/channels/{channel_id}/invoices")
  643. async def list_channel_invoices(
  644. channel_id: str,
  645. user: dict = Depends(require_admin_or_super_admin),
  646. db: Session = Depends(get_db),
  647. ):
  648. try:
  649. tenant_no = _get_tenant_no(db, user)
  650. items = ps_svc.list_channel_invoices(db, channel_id, tenant_no=tenant_no, user_info=user)
  651. return success_response(items)
  652. except ValueError as e:
  653. return error_response(400, str(e))
  654. @router.post("/channels/{channel_id}/invoices")
  655. async def create_channel_invoice(
  656. channel_id: str,
  657. body: ChannelInvoiceCreateBody,
  658. user: dict = Depends(require_admin_or_super_admin),
  659. db: Session = Depends(get_db),
  660. ):
  661. try:
  662. tenant_no = _get_tenant_no(db, user)
  663. obj = ps_svc.create_channel_invoice(
  664. db,
  665. channel_id=channel_id,
  666. company_name=body.company_name,
  667. tax_no=body.tax_no,
  668. company_address=body.company_address,
  669. company_phone=body.company_phone,
  670. bank_name=body.bank_name,
  671. bank_account=body.bank_account,
  672. tenant_no=tenant_no,
  673. order_num=body.order_num,
  674. remark=body.remark,
  675. user_id=user.get("user_id"),
  676. )
  677. return success_response(ps_svc._row_to_dict_channel_invoice(obj), message="创建成功")
  678. except ValueError as e:
  679. return error_response(400, str(e))
  680. @router.put("/channel-invoices/{invoice_id}")
  681. async def update_channel_invoice(
  682. invoice_id: str,
  683. body: ChannelInvoiceUpdateBody,
  684. user: dict = Depends(require_admin_or_super_admin),
  685. db: Session = Depends(get_db),
  686. ):
  687. try:
  688. tenant_no = _get_tenant_no(db, user)
  689. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  690. obj = ps_svc.update_channel_invoice(
  691. db,
  692. invoice_id,
  693. company_name=payload.get("company_name"),
  694. tax_no=payload.get("tax_no"),
  695. company_address=payload.get("company_address"),
  696. company_phone=payload.get("company_phone"),
  697. bank_name=payload.get("bank_name"),
  698. bank_account=payload.get("bank_account"),
  699. order_num=payload.get("order_num"),
  700. remark=payload.get("remark"),
  701. user_id=user.get("user_id"),
  702. tenant_no=tenant_no,
  703. )
  704. return success_response(ps_svc._row_to_dict_channel_invoice(obj), message="更新成功")
  705. except ValueError as e:
  706. return error_response(400, str(e))
  707. @router.delete("/channel-invoices/{invoice_id}")
  708. async def delete_channel_invoice(
  709. invoice_id: str,
  710. user: dict = Depends(require_admin_or_super_admin),
  711. db: Session = Depends(get_db),
  712. ):
  713. try:
  714. tenant_no = _get_tenant_no(db, user)
  715. ps_svc.soft_delete_channel_invoice(db, invoice_id, tenant_no=tenant_no)
  716. return success_response(None, message="删除成功")
  717. except ValueError as e:
  718. return error_response(400, str(e))
  719. # --- 渠道邮寄地址 ---
  720. class ChannelMailAddressCreateBody(BaseModel):
  721. model_config = ConfigDict(populate_by_name=True)
  722. mail_address: str = Field(..., alias="mailAddress")
  723. mail_contact: Optional[str] = Field(None, alias="mailContact")
  724. mail_phone: Optional[str] = Field(None, alias="mailPhone")
  725. order_num: Optional[int] = Field(0, alias="orderNum")
  726. remark: Optional[str] = None
  727. class ChannelMailAddressUpdateBody(BaseModel):
  728. model_config = ConfigDict(populate_by_name=True)
  729. mail_address: Optional[str] = Field(None, alias="mailAddress")
  730. mail_contact: Optional[str] = Field(None, alias="mailContact")
  731. mail_phone: Optional[str] = Field(None, alias="mailPhone")
  732. order_num: Optional[int] = Field(None, alias="orderNum")
  733. remark: Optional[str] = None
  734. @router.get("/channels/{channel_id}/mail-addresses")
  735. async def list_channel_mail_addresses(
  736. channel_id: str,
  737. user: dict = Depends(require_admin_or_super_admin),
  738. db: Session = Depends(get_db),
  739. ):
  740. try:
  741. tenant_no = _get_tenant_no(db, user)
  742. items = ps_svc.list_channel_mail_addresses(db, channel_id, tenant_no=tenant_no, user_info=user)
  743. return success_response(items)
  744. except ValueError as e:
  745. return error_response(400, str(e))
  746. @router.post("/channels/{channel_id}/mail-addresses")
  747. async def create_channel_mail_address(
  748. channel_id: str,
  749. body: ChannelMailAddressCreateBody,
  750. user: dict = Depends(require_admin_or_super_admin),
  751. db: Session = Depends(get_db),
  752. ):
  753. try:
  754. tenant_no = _get_tenant_no(db, user)
  755. obj = ps_svc.create_channel_mail_address(
  756. db,
  757. channel_id=channel_id,
  758. mail_address=body.mail_address,
  759. mail_contact=body.mail_contact,
  760. mail_phone=body.mail_phone,
  761. tenant_no=tenant_no,
  762. order_num=body.order_num,
  763. remark=body.remark,
  764. user_id=user.get("user_id"),
  765. )
  766. return success_response(ps_svc._row_to_dict_channel_mail_address(obj), message="创建成功")
  767. except ValueError as e:
  768. return error_response(400, str(e))
  769. @router.put("/channel-mail-addresses/{address_id}")
  770. async def update_channel_mail_address(
  771. address_id: str,
  772. body: ChannelMailAddressUpdateBody,
  773. user: dict = Depends(require_admin_or_super_admin),
  774. db: Session = Depends(get_db),
  775. ):
  776. try:
  777. tenant_no = _get_tenant_no(db, user)
  778. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  779. obj = ps_svc.update_channel_mail_address(
  780. db,
  781. address_id,
  782. mail_address=payload.get("mail_address"),
  783. mail_contact=payload.get("mail_contact"),
  784. mail_phone=payload.get("mail_phone"),
  785. order_num=payload.get("order_num"),
  786. remark=payload.get("remark"),
  787. user_id=user.get("user_id"),
  788. tenant_no=tenant_no,
  789. )
  790. return success_response(ps_svc._row_to_dict_channel_mail_address(obj), message="更新成功")
  791. except ValueError as e:
  792. return error_response(400, str(e))
  793. @router.delete("/channel-mail-addresses/{address_id}")
  794. async def delete_channel_mail_address(
  795. address_id: str,
  796. user: dict = Depends(require_admin_or_super_admin),
  797. db: Session = Depends(get_db),
  798. ):
  799. try:
  800. tenant_no = _get_tenant_no(db, user)
  801. ps_svc.soft_delete_channel_mail_address(db, address_id, tenant_no=tenant_no)
  802. return success_response(None, message="删除成功")
  803. except ValueError as e:
  804. return error_response(400, str(e))
  805. # --- 渠道协议 ---
  806. class ChannelAgreementCreateBody(BaseModel):
  807. model_config = ConfigDict(populate_by_name=True)
  808. agreement_type: str = Field(..., alias="agreementType")
  809. business_type: Optional[str] = Field(None, alias="businessType")
  810. settle_mode: Optional[str] = Field(None, alias="settleMode")
  811. credit_limit_amount: Optional[float] = Field(None, alias="creditLimitAmount")
  812. effective_date: Optional[str] = Field(None, alias="effectiveDate")
  813. expire_date: Optional[str] = Field(None, alias="expireDate")
  814. order_num: Optional[int] = Field(0, alias="orderNum")
  815. remark: Optional[str] = None
  816. class ChannelAgreementUpdateBody(BaseModel):
  817. model_config = ConfigDict(populate_by_name=True)
  818. agreement_type: Optional[str] = Field(None, alias="agreementType")
  819. business_type: Optional[str] = Field(None, alias="businessType")
  820. settle_mode: Optional[str] = Field(None, alias="settleMode")
  821. credit_limit_amount: Optional[float] = Field(None, alias="creditLimitAmount")
  822. effective_date: Optional[str] = Field(None, alias="effectiveDate")
  823. expire_date: Optional[str] = Field(None, alias="expireDate")
  824. order_num: Optional[int] = Field(None, alias="orderNum")
  825. remark: Optional[str] = None
  826. @router.get("/channels/{channel_id}/agreements")
  827. async def list_channel_agreements(
  828. channel_id: str,
  829. user: dict = Depends(require_admin_or_super_admin),
  830. db: Session = Depends(get_db),
  831. ):
  832. try:
  833. tenant_no = _get_tenant_no(db, user)
  834. items = ps_svc.list_channel_agreements(db, channel_id, tenant_no=tenant_no, user_info=user)
  835. return success_response(items)
  836. except ValueError as e:
  837. return error_response(400, str(e))
  838. @router.post("/channels/{channel_id}/agreements")
  839. async def create_channel_agreement(
  840. channel_id: str,
  841. body: ChannelAgreementCreateBody,
  842. user: dict = Depends(require_admin_or_super_admin),
  843. db: Session = Depends(get_db),
  844. ):
  845. try:
  846. tenant_no = _get_tenant_no(db, user)
  847. obj = ps_svc.create_channel_agreement(
  848. db,
  849. channel_id=channel_id,
  850. agreement_type=body.agreement_type,
  851. business_type=body.business_type,
  852. settle_mode=body.settle_mode,
  853. credit_limit_amount=body.credit_limit_amount,
  854. effective_date=body.effective_date,
  855. expire_date=body.expire_date,
  856. tenant_no=tenant_no,
  857. order_num=body.order_num,
  858. remark=body.remark,
  859. user_id=user.get("user_id"),
  860. )
  861. return success_response(ps_svc._row_to_dict_channel_agreement(obj), message="创建成功")
  862. except ValueError as e:
  863. return error_response(400, str(e))
  864. @router.put("/channel-agreements/{agreement_id}")
  865. async def update_channel_agreement(
  866. agreement_id: str,
  867. body: ChannelAgreementUpdateBody,
  868. user: dict = Depends(require_admin_or_super_admin),
  869. db: Session = Depends(get_db),
  870. ):
  871. try:
  872. tenant_no = _get_tenant_no(db, user)
  873. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  874. obj = ps_svc.update_channel_agreement(
  875. db,
  876. agreement_id,
  877. agreement_type=payload.get("agreement_type"),
  878. business_type=payload.get("business_type"),
  879. settle_mode=payload.get("settle_mode"),
  880. credit_limit_amount=payload.get("credit_limit_amount"),
  881. effective_date=payload.get("effective_date"),
  882. expire_date=payload.get("expire_date"),
  883. order_num=payload.get("order_num"),
  884. remark=payload.get("remark"),
  885. user_id=user.get("user_id"),
  886. tenant_no=tenant_no,
  887. )
  888. return success_response(ps_svc._row_to_dict_channel_agreement(obj), message="更新成功")
  889. except ValueError as e:
  890. return error_response(400, str(e))
  891. @router.delete("/channel-agreements/{agreement_id}")
  892. async def delete_channel_agreement(
  893. agreement_id: str,
  894. user: dict = Depends(require_admin_or_super_admin),
  895. db: Session = Depends(get_db),
  896. ):
  897. try:
  898. tenant_no = _get_tenant_no(db, user)
  899. ps_svc.soft_delete_channel_agreement(db, agreement_id, tenant_no=tenant_no)
  900. return success_response(None, message="删除成功")
  901. except ValueError as e:
  902. return error_response(400, str(e))
  903. @router.put("/channel-agreements/{agreement_id}/activate")
  904. async def activate_channel_agreement(
  905. agreement_id: str,
  906. user: dict = Depends(require_admin_or_super_admin),
  907. db: Session = Depends(get_db),
  908. ):
  909. try:
  910. tenant_no = _get_tenant_no(db, user)
  911. obj = ps_svc.activate_channel_agreement(
  912. db,
  913. agreement_id,
  914. tenant_no=tenant_no,
  915. user_id=user.get("user_id"),
  916. )
  917. return success_response(ps_svc._row_to_dict_channel_agreement(obj), message="生效成功")
  918. except ValueError as e:
  919. return error_response(400, str(e))
  920. @router.put("/channel-agreements/{agreement_id}/void")
  921. async def void_channel_agreement(
  922. agreement_id: str,
  923. user: dict = Depends(require_admin_or_super_admin),
  924. db: Session = Depends(get_db),
  925. ):
  926. try:
  927. tenant_no = _get_tenant_no(db, user)
  928. obj = ps_svc.void_channel_agreement(
  929. db,
  930. agreement_id,
  931. tenant_no=tenant_no,
  932. user_id=user.get("user_id"),
  933. )
  934. return success_response(ps_svc._row_to_dict_channel_agreement(obj), message="作废成功")
  935. except ValueError as e:
  936. return error_response(400, str(e))
  937. # --- 渠道协议产品价格 ---
  938. class ChannelAgreementPriceCreateBody(BaseModel):
  939. model_config = ConfigDict(populate_by_name=True)
  940. project_name: Optional[str] = Field(None, alias="projectName")
  941. project_code: Optional[str] = Field(None, alias="projectCode")
  942. cell_type_code: Optional[str] = Field(None, alias="cellTypeCode")
  943. cell_type_name: Optional[str] = Field(None, alias="cellTypeName")
  944. spec_text: Optional[str] = Field(None, alias="specText")
  945. original_price: Optional[float] = Field(None, alias="originalPrice")
  946. agreement_price: Optional[float] = Field(None, alias="agreementPrice")
  947. order_num: Optional[int] = Field(0, alias="orderNum")
  948. remark: Optional[str] = None
  949. class ChannelAgreementPriceUpdateBody(BaseModel):
  950. model_config = ConfigDict(populate_by_name=True)
  951. project_name: Optional[str] = Field(None, alias="projectName")
  952. project_code: Optional[str] = Field(None, alias="projectCode")
  953. cell_type_code: Optional[str] = Field(None, alias="cellTypeCode")
  954. cell_type_name: Optional[str] = Field(None, alias="cellTypeName")
  955. spec_text: Optional[str] = Field(None, alias="specText")
  956. original_price: Optional[float] = Field(None, alias="originalPrice")
  957. agreement_price: Optional[float] = Field(None, alias="agreementPrice")
  958. order_num: Optional[int] = Field(None, alias="orderNum")
  959. remark: Optional[str] = None
  960. @router.get("/channel-agreements/{agreement_id}/product-prices")
  961. async def list_channel_agreement_prices(
  962. agreement_id: str,
  963. user: dict = Depends(require_admin_or_super_admin),
  964. db: Session = Depends(get_db),
  965. ):
  966. try:
  967. tenant_no = _get_tenant_no(db, user)
  968. items = ps_svc.list_channel_agreement_prices(db, agreement_id, tenant_no=tenant_no, user_info=user)
  969. return success_response(items)
  970. except ValueError as e:
  971. return error_response(400, str(e))
  972. @router.post("/channel-agreements/{agreement_id}/product-prices")
  973. async def create_channel_agreement_price(
  974. agreement_id: str,
  975. body: ChannelAgreementPriceCreateBody,
  976. user: dict = Depends(require_admin_or_super_admin),
  977. db: Session = Depends(get_db),
  978. ):
  979. try:
  980. tenant_no = _get_tenant_no(db, user)
  981. obj = ps_svc.create_channel_agreement_price(
  982. db,
  983. channel_agreement_id=agreement_id,
  984. project_name=body.project_name,
  985. project_code=body.project_code,
  986. cell_type_code=body.cell_type_code,
  987. cell_type_name=body.cell_type_name,
  988. spec_text=body.spec_text,
  989. original_price=body.original_price,
  990. agreement_price=body.agreement_price,
  991. tenant_no=tenant_no,
  992. order_num=body.order_num,
  993. remark=body.remark,
  994. user_id=user.get("user_id"),
  995. )
  996. return success_response(ps_svc._row_to_dict_channel_agreement_price(obj), message="创建成功")
  997. except ValueError as e:
  998. return error_response(400, str(e))
  999. @router.put("/channel-agreement-product-prices/{price_id}")
  1000. async def update_channel_agreement_price(
  1001. price_id: str,
  1002. body: ChannelAgreementPriceUpdateBody,
  1003. user: dict = Depends(require_admin_or_super_admin),
  1004. db: Session = Depends(get_db),
  1005. ):
  1006. try:
  1007. tenant_no = _get_tenant_no(db, user)
  1008. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  1009. obj = ps_svc.update_channel_agreement_price(
  1010. db,
  1011. price_id,
  1012. project_name=payload.get("project_name"),
  1013. project_code=payload.get("project_code"),
  1014. cell_type_code=payload.get("cell_type_code"),
  1015. cell_type_name=payload.get("cell_type_name"),
  1016. spec_text=payload.get("spec_text"),
  1017. original_price=payload.get("original_price"),
  1018. agreement_price=payload.get("agreement_price"),
  1019. order_num=payload.get("order_num"),
  1020. remark=payload.get("remark"),
  1021. user_id=user.get("user_id"),
  1022. tenant_no=tenant_no,
  1023. )
  1024. return success_response(ps_svc._row_to_dict_channel_agreement_price(obj), message="更新成功")
  1025. except ValueError as e:
  1026. return error_response(400, str(e))
  1027. @router.delete("/channel-agreement-product-prices/{price_id}")
  1028. async def delete_channel_agreement_price(
  1029. price_id: str,
  1030. user: dict = Depends(require_admin_or_super_admin),
  1031. db: Session = Depends(get_db),
  1032. ):
  1033. try:
  1034. tenant_no = _get_tenant_no(db, user)
  1035. ps_svc.soft_delete_channel_agreement_price(db, price_id, tenant_no=tenant_no)
  1036. return success_response(None, message="删除成功")
  1037. except ValueError as e:
  1038. return error_response(400, str(e))
  1039. # ---------------------------------------------------------------------------
  1040. # 6. 体检机构(主表 + 开票信息/项目及价格子表)
  1041. # ---------------------------------------------------------------------------
  1042. class CheckupInstitutionCreateBody(BaseModel):
  1043. model_config = ConfigDict(populate_by_name=True)
  1044. institution_name: str = Field(..., alias="institutionName")
  1045. institution_address: Optional[str] = Field(None, alias="institutionAddress")
  1046. contact_name: Optional[str] = Field(None, alias="contactName")
  1047. contact_phone: Optional[str] = Field(None, alias="contactPhone")
  1048. status: Optional[str] = Field("1", alias="status")
  1049. order_num: Optional[int] = Field(0, alias="orderNum")
  1050. remark: Optional[str] = None
  1051. class CheckupInstitutionUpdateBody(BaseModel):
  1052. model_config = ConfigDict(populate_by_name=True)
  1053. institution_name: Optional[str] = Field(None, alias="institutionName")
  1054. institution_address: Optional[str] = Field(None, alias="institutionAddress")
  1055. contact_name: Optional[str] = Field(None, alias="contactName")
  1056. contact_phone: Optional[str] = Field(None, alias="contactPhone")
  1057. status: Optional[str] = Field(None, alias="status")
  1058. order_num: Optional[int] = Field(None, alias="orderNum")
  1059. remark: Optional[str] = None
  1060. @router.get("/checkup-institutions")
  1061. async def list_checkup_institutions(
  1062. page: int = Query(1, ge=1),
  1063. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  1064. keyword: Optional[str] = None,
  1065. status: Optional[str] = None,
  1066. user: dict = Depends(require_admin_or_super_admin),
  1067. db: Session = Depends(get_db),
  1068. ):
  1069. tenant_no = _get_tenant_no(db, user)
  1070. items, total = ps_svc.list_checkup_institutions(
  1071. db,
  1072. page=page,
  1073. page_size=pageSize,
  1074. keyword=keyword,
  1075. status=status,
  1076. tenant_no=tenant_no,
  1077. user_info=user,
  1078. )
  1079. tp = ps_svc.total_pages(total, pageSize)
  1080. return success_response(
  1081. {"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp}
  1082. )
  1083. @router.post("/checkup-institutions")
  1084. async def create_checkup_institution(
  1085. body: CheckupInstitutionCreateBody,
  1086. user: dict = Depends(require_admin_or_super_admin),
  1087. db: Session = Depends(get_db),
  1088. ):
  1089. try:
  1090. tenant_no = _get_tenant_no(db, user)
  1091. obj = ps_svc.create_checkup_institution(
  1092. db,
  1093. institution_name=body.institution_name,
  1094. institution_address=body.institution_address,
  1095. contact_name=body.contact_name,
  1096. contact_phone=body.contact_phone,
  1097. status=body.status,
  1098. tenant_no=tenant_no,
  1099. order_num=body.order_num,
  1100. remark=body.remark,
  1101. user_id=user.get("user_id"),
  1102. )
  1103. return success_response(ps_svc._row_to_dict_checkup_institution(obj), message="创建成功")
  1104. except ValueError as e:
  1105. return error_response(400, str(e))
  1106. @router.put("/checkup-institutions/{row_id}")
  1107. async def update_checkup_institution(
  1108. row_id: str,
  1109. body: CheckupInstitutionUpdateBody,
  1110. user: dict = Depends(require_admin_or_super_admin),
  1111. db: Session = Depends(get_db),
  1112. ):
  1113. try:
  1114. tenant_no = _get_tenant_no(db, user)
  1115. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  1116. obj = ps_svc.update_checkup_institution(
  1117. db,
  1118. row_id,
  1119. institution_name=payload.get("institution_name"),
  1120. institution_address=payload.get("institution_address"),
  1121. contact_name=payload.get("contact_name"),
  1122. contact_phone=payload.get("contact_phone"),
  1123. status=payload.get("status"),
  1124. order_num=payload.get("order_num"),
  1125. remark=payload.get("remark"),
  1126. user_id=user.get("user_id"),
  1127. tenant_no=tenant_no,
  1128. )
  1129. return success_response(ps_svc._row_to_dict_checkup_institution(obj), message="更新成功")
  1130. except ValueError as e:
  1131. return error_response(400, str(e))
  1132. @router.delete("/checkup-institutions/{row_id}")
  1133. async def delete_checkup_institution(
  1134. row_id: str,
  1135. user: dict = Depends(require_admin_or_super_admin),
  1136. db: Session = Depends(get_db),
  1137. ):
  1138. try:
  1139. tenant_no = _get_tenant_no(db, user)
  1140. ps_svc.soft_delete_checkup_institution(db, row_id, tenant_no=tenant_no)
  1141. return success_response(None, message="删除成功")
  1142. except ValueError as e:
  1143. return error_response(400, str(e))
  1144. # --- 体检机构开票信息 ---
  1145. class CheckupInstitutionInvoiceCreateBody(BaseModel):
  1146. model_config = ConfigDict(populate_by_name=True)
  1147. company_name: str = Field(..., alias="companyName")
  1148. tax_no: Optional[str] = Field(None, alias="taxNo")
  1149. company_address: Optional[str] = Field(None, alias="companyAddress")
  1150. company_phone: Optional[str] = Field(None, alias="companyPhone")
  1151. bank_name: Optional[str] = Field(None, alias="bankName")
  1152. bank_account: Optional[str] = Field(None, alias="bankAccount")
  1153. order_num: Optional[int] = Field(0, alias="orderNum")
  1154. remark: Optional[str] = None
  1155. class CheckupInstitutionInvoiceUpdateBody(BaseModel):
  1156. model_config = ConfigDict(populate_by_name=True)
  1157. company_name: Optional[str] = Field(None, alias="companyName")
  1158. tax_no: Optional[str] = Field(None, alias="taxNo")
  1159. company_address: Optional[str] = Field(None, alias="companyAddress")
  1160. company_phone: Optional[str] = Field(None, alias="companyPhone")
  1161. bank_name: Optional[str] = Field(None, alias="bankName")
  1162. bank_account: Optional[str] = Field(None, alias="bankAccount")
  1163. order_num: Optional[int] = Field(None, alias="orderNum")
  1164. remark: Optional[str] = None
  1165. @router.get("/checkup-institutions/{institution_id}/invoices")
  1166. async def list_checkup_institution_invoices(
  1167. institution_id: str,
  1168. user: dict = Depends(require_admin_or_super_admin),
  1169. db: Session = Depends(get_db),
  1170. ):
  1171. try:
  1172. tenant_no = _get_tenant_no(db, user)
  1173. items = ps_svc.list_checkup_institution_invoices(db, institution_id, tenant_no=tenant_no, user_info=user)
  1174. return success_response(items)
  1175. except ValueError as e:
  1176. return error_response(400, str(e))
  1177. @router.post("/checkup-institutions/{institution_id}/invoices")
  1178. async def create_checkup_institution_invoice(
  1179. institution_id: str,
  1180. body: CheckupInstitutionInvoiceCreateBody,
  1181. user: dict = Depends(require_admin_or_super_admin),
  1182. db: Session = Depends(get_db),
  1183. ):
  1184. try:
  1185. tenant_no = _get_tenant_no(db, user)
  1186. obj = ps_svc.create_checkup_institution_invoice(
  1187. db,
  1188. checkup_institution_id=institution_id,
  1189. company_name=body.company_name,
  1190. tax_no=body.tax_no,
  1191. company_address=body.company_address,
  1192. company_phone=body.company_phone,
  1193. bank_name=body.bank_name,
  1194. bank_account=body.bank_account,
  1195. tenant_no=tenant_no,
  1196. order_num=body.order_num,
  1197. remark=body.remark,
  1198. user_id=user.get("user_id"),
  1199. )
  1200. return success_response(ps_svc._row_to_dict_checkup_institution_invoice(obj), message="创建成功")
  1201. except ValueError as e:
  1202. return error_response(400, str(e))
  1203. @router.put("/checkup-institution-invoices/{invoice_id}")
  1204. async def update_checkup_institution_invoice(
  1205. invoice_id: str,
  1206. body: CheckupInstitutionInvoiceUpdateBody,
  1207. user: dict = Depends(require_admin_or_super_admin),
  1208. db: Session = Depends(get_db),
  1209. ):
  1210. try:
  1211. tenant_no = _get_tenant_no(db, user)
  1212. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  1213. obj = ps_svc.update_checkup_institution_invoice(
  1214. db,
  1215. invoice_id,
  1216. company_name=payload.get("company_name"),
  1217. tax_no=payload.get("tax_no"),
  1218. company_address=payload.get("company_address"),
  1219. company_phone=payload.get("company_phone"),
  1220. bank_name=payload.get("bank_name"),
  1221. bank_account=payload.get("bank_account"),
  1222. order_num=payload.get("order_num"),
  1223. remark=payload.get("remark"),
  1224. user_id=user.get("user_id"),
  1225. tenant_no=tenant_no,
  1226. )
  1227. return success_response(ps_svc._row_to_dict_checkup_institution_invoice(obj), message="更新成功")
  1228. except ValueError as e:
  1229. return error_response(400, str(e))
  1230. @router.delete("/checkup-institution-invoices/{invoice_id}")
  1231. async def delete_checkup_institution_invoice(
  1232. invoice_id: str,
  1233. user: dict = Depends(require_admin_or_super_admin),
  1234. db: Session = Depends(get_db),
  1235. ):
  1236. try:
  1237. tenant_no = _get_tenant_no(db, user)
  1238. ps_svc.soft_delete_checkup_institution_invoice(db, invoice_id, tenant_no=tenant_no)
  1239. return success_response(None, message="删除成功")
  1240. except ValueError as e:
  1241. return error_response(400, str(e))
  1242. # --- 体检机构项目及价格 ---
  1243. class CheckupInstitutionProjectCreateBody(BaseModel):
  1244. model_config = ConfigDict(populate_by_name=True)
  1245. project_name: str = Field(..., alias="projectName")
  1246. project_fee: Optional[float] = Field(None, alias="projectFee")
  1247. status: Optional[str] = Field("1", alias="status")
  1248. order_num: Optional[int] = Field(0, alias="orderNum")
  1249. remark: Optional[str] = None
  1250. class CheckupInstitutionProjectUpdateBody(BaseModel):
  1251. model_config = ConfigDict(populate_by_name=True)
  1252. project_name: Optional[str] = Field(None, alias="projectName")
  1253. project_fee: Optional[float] = Field(None, alias="projectFee")
  1254. status: Optional[str] = Field(None, alias="status")
  1255. order_num: Optional[int] = Field(None, alias="orderNum")
  1256. remark: Optional[str] = None
  1257. @router.get("/checkup-institutions/{institution_id}/projects")
  1258. async def list_checkup_institution_projects(
  1259. institution_id: str,
  1260. user: dict = Depends(require_admin_or_super_admin),
  1261. db: Session = Depends(get_db),
  1262. ):
  1263. try:
  1264. tenant_no = _get_tenant_no(db, user)
  1265. items = ps_svc.list_checkup_institution_projects(db, institution_id, tenant_no=tenant_no, user_info=user)
  1266. return success_response(items)
  1267. except ValueError as e:
  1268. return error_response(400, str(e))
  1269. @router.post("/checkup-institutions/{institution_id}/projects")
  1270. async def create_checkup_institution_project(
  1271. institution_id: str,
  1272. body: CheckupInstitutionProjectCreateBody,
  1273. user: dict = Depends(require_admin_or_super_admin),
  1274. db: Session = Depends(get_db),
  1275. ):
  1276. try:
  1277. tenant_no = _get_tenant_no(db, user)
  1278. obj = ps_svc.create_checkup_institution_project(
  1279. db,
  1280. checkup_institution_id=institution_id,
  1281. project_name=body.project_name,
  1282. project_fee=body.project_fee,
  1283. status=body.status,
  1284. tenant_no=tenant_no,
  1285. order_num=body.order_num,
  1286. remark=body.remark,
  1287. user_id=user.get("user_id"),
  1288. )
  1289. return success_response(ps_svc._row_to_dict_checkup_institution_project(obj), message="创建成功")
  1290. except ValueError as e:
  1291. return error_response(400, str(e))
  1292. @router.put("/checkup-institution-projects/{project_id}")
  1293. async def update_checkup_institution_project(
  1294. project_id: str,
  1295. body: CheckupInstitutionProjectUpdateBody,
  1296. user: dict = Depends(require_admin_or_super_admin),
  1297. db: Session = Depends(get_db),
  1298. ):
  1299. try:
  1300. tenant_no = _get_tenant_no(db, user)
  1301. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  1302. obj = ps_svc.update_checkup_institution_project(
  1303. db,
  1304. project_id,
  1305. project_name=payload.get("project_name"),
  1306. project_fee=payload.get("project_fee"),
  1307. status=payload.get("status"),
  1308. order_num=payload.get("order_num"),
  1309. remark=payload.get("remark"),
  1310. user_id=user.get("user_id"),
  1311. tenant_no=tenant_no,
  1312. )
  1313. return success_response(ps_svc._row_to_dict_checkup_institution_project(obj), message="更新成功")
  1314. except ValueError as e:
  1315. return error_response(400, str(e))
  1316. @router.delete("/checkup-institution-projects/{project_id}")
  1317. async def delete_checkup_institution_project(
  1318. project_id: str,
  1319. user: dict = Depends(require_admin_or_super_admin),
  1320. db: Session = Depends(get_db),
  1321. ):
  1322. try:
  1323. tenant_no = _get_tenant_no(db, user)
  1324. ps_svc.soft_delete_checkup_institution_project(db, project_id, tenant_no=tenant_no)
  1325. return success_response(None, message="删除成功")
  1326. except ValueError as e:
  1327. return error_response(400, str(e))
  1328. # --- 体检机构项目套餐 ---
  1329. class CheckupPackageItemBody(BaseModel):
  1330. model_config = ConfigDict(populate_by_name=True)
  1331. checkup_institution_project_id: str = Field(..., alias="checkupInstitutionProjectId")
  1332. project_name: Optional[str] = Field(None, alias="projectName")
  1333. project_fee: Optional[float] = Field(None, alias="projectFee")
  1334. order_num: Optional[int] = Field(None, alias="orderNum")
  1335. remark: Optional[str] = None
  1336. class CheckupInstitutionPackageCreateBody(BaseModel):
  1337. model_config = ConfigDict(populate_by_name=True)
  1338. package_name: str = Field(..., alias="packageName")
  1339. package_code: str = Field(..., alias="packageCode")
  1340. status: Optional[str] = Field("1", alias="status")
  1341. order_num: Optional[int] = Field(0, alias="orderNum")
  1342. remark: Optional[str] = None
  1343. items: List[CheckupPackageItemBody] = Field(default_factory=list)
  1344. class CheckupInstitutionPackageUpdateBody(BaseModel):
  1345. model_config = ConfigDict(populate_by_name=True)
  1346. package_name: Optional[str] = Field(None, alias="packageName")
  1347. package_code: Optional[str] = Field(None, alias="packageCode")
  1348. status: Optional[str] = Field(None, alias="status")
  1349. order_num: Optional[int] = Field(None, alias="orderNum")
  1350. remark: Optional[str] = None
  1351. items: Optional[List[CheckupPackageItemBody]] = None
  1352. def _package_items_to_crud(items) -> List[dict]:
  1353. out = []
  1354. for x in items:
  1355. if isinstance(x, dict):
  1356. out.append(
  1357. {
  1358. "checkup_institution_project_id": x.get("checkup_institution_project_id")
  1359. or x.get("checkupInstitutionProjectId"),
  1360. "project_name": x.get("project_name") or x.get("projectName"),
  1361. "project_fee": x.get("project_fee") if x.get("project_fee") is not None else x.get("projectFee"),
  1362. "order_num": x.get("order_num") if x.get("order_num") is not None else x.get("orderNum"),
  1363. "remark": x.get("remark"),
  1364. }
  1365. )
  1366. else:
  1367. out.append(
  1368. {
  1369. "checkup_institution_project_id": x.checkup_institution_project_id,
  1370. "project_name": x.project_name,
  1371. "project_fee": x.project_fee,
  1372. "order_num": x.order_num,
  1373. "remark": x.remark,
  1374. }
  1375. )
  1376. return out
  1377. @router.get("/checkup-institutions/{institution_id}/packages")
  1378. async def list_checkup_institution_packages(
  1379. institution_id: str,
  1380. user: dict = Depends(require_admin_or_super_admin),
  1381. db: Session = Depends(get_db),
  1382. ):
  1383. try:
  1384. tenant_no = _get_tenant_no(db, user)
  1385. items = ps_svc.list_checkup_institution_packages(db, institution_id, tenant_no=tenant_no, user_info=user)
  1386. return success_response(items)
  1387. except ValueError as e:
  1388. return error_response(400, str(e))
  1389. @router.post("/checkup-institutions/{institution_id}/packages")
  1390. async def create_checkup_institution_package(
  1391. institution_id: str,
  1392. body: CheckupInstitutionPackageCreateBody,
  1393. user: dict = Depends(require_admin_or_super_admin),
  1394. db: Session = Depends(get_db),
  1395. ):
  1396. try:
  1397. tenant_no = _get_tenant_no(db, user)
  1398. detail = ps_svc.create_checkup_institution_package(
  1399. db,
  1400. checkup_institution_id=institution_id,
  1401. package_name=body.package_name,
  1402. package_code=body.package_code,
  1403. status=body.status,
  1404. items=_package_items_to_crud(body.items),
  1405. tenant_no=tenant_no,
  1406. order_num=body.order_num,
  1407. remark=body.remark,
  1408. user_id=user.get("user_id"),
  1409. )
  1410. return success_response(detail, message="创建成功")
  1411. except ValueError as e:
  1412. return error_response(400, str(e))
  1413. @router.get("/checkup-institution-packages/{package_id}")
  1414. async def get_checkup_institution_package_detail(
  1415. package_id: str,
  1416. user: dict = Depends(require_admin_or_super_admin),
  1417. db: Session = Depends(get_db),
  1418. ):
  1419. try:
  1420. tenant_no = _get_tenant_no(db, user)
  1421. detail = ps_svc.get_checkup_institution_package_detail(db, package_id, tenant_no=tenant_no)
  1422. if not detail:
  1423. return error_response(404, "记录不存在")
  1424. return success_response(detail)
  1425. except ValueError as e:
  1426. return error_response(400, str(e))
  1427. @router.put("/checkup-institution-packages/{package_id}")
  1428. async def update_checkup_institution_package(
  1429. package_id: str,
  1430. body: CheckupInstitutionPackageUpdateBody,
  1431. user: dict = Depends(require_admin_or_super_admin),
  1432. db: Session = Depends(get_db),
  1433. ):
  1434. try:
  1435. tenant_no = _get_tenant_no(db, user)
  1436. payload = body.model_dump(exclude_unset=True, by_alias=False)
  1437. items_raw = payload.pop("items", None)
  1438. items = _package_items_to_crud(items_raw) if items_raw is not None else None
  1439. detail = ps_svc.update_checkup_institution_package(
  1440. db,
  1441. package_id,
  1442. package_name=payload.get("package_name"),
  1443. package_code=payload.get("package_code"),
  1444. status=payload.get("status"),
  1445. items=items,
  1446. order_num=payload.get("order_num"),
  1447. remark=payload.get("remark"),
  1448. user_id=user.get("user_id"),
  1449. tenant_no=tenant_no,
  1450. )
  1451. return success_response(detail, message="更新成功")
  1452. except ValueError as e:
  1453. return error_response(400, str(e))
  1454. @router.delete("/checkup-institution-packages/{package_id}")
  1455. async def delete_checkup_institution_package(
  1456. package_id: str,
  1457. user: dict = Depends(require_admin_or_super_admin),
  1458. db: Session = Depends(get_db),
  1459. ):
  1460. try:
  1461. tenant_no = _get_tenant_no(db, user)
  1462. ps_svc.soft_delete_checkup_institution_package(db, package_id, tenant_no=tenant_no)
  1463. return success_response(None, message="删除成功")
  1464. except ValueError as e:
  1465. return error_response(400, str(e))