procurement.py 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. from fastapi import APIRouter, Request, Depends, Query, HTTPException, status,WebSocket,WebSocketDisconnect
  4. from common.security import valid_access_token,valid_websocket_token
  5. from fastapi.responses import JSONResponse
  6. from sqlalchemy.orm import Session
  7. from sqlalchemy.sql import func
  8. from common.auth_user import *
  9. from sqlalchemy import text
  10. from pydantic import BaseModel
  11. from common.BigDataCenterAPI import *
  12. from database import get_db
  13. from typing import List
  14. from models import *
  15. from utils import *
  16. from utils.spatial import *
  17. from utils.ry_system_util import *
  18. from utils.resource_provision_util import *
  19. from common.websocketManager import *
  20. import json
  21. import traceback
  22. router = APIRouter()
  23. @router.post("/create")
  24. async def create_pattern(
  25. user_id=Depends(valid_access_token),
  26. body = Depends(remove_xss_json),
  27. db: Session = Depends(get_db)
  28. ):
  29. try:
  30. dept_info = user_id_get_user_info(db,user_id)
  31. new_declaration = ResourceProvisionMaterialType(
  32. id = new_guid(),
  33. declaration_amount=body['declaration_amount'],
  34. declaration_person=user_id,
  35. declaration_unit=dept_info.dept_id,
  36. create_by = user_id
  37. )
  38. db.add(new_declaration)
  39. for info in body['detail']:
  40. new_detail = ResourceProvisionProcurementDeclarationDetail(
  41. id=new_guid(),
  42. serial_number=info['serial_number'],
  43. declaration_id = new_declaration.id,
  44. warehouse_id = info['warehouse_id'],
  45. material_type = info['material_type'],
  46. material_code = info['material_code'],
  47. material_quantity = info['material_quantity'],
  48. material_unit_price = info['material_unit_price'],
  49. material_purpose = info['material_purpose'],
  50. create_by = user_id
  51. )
  52. db.add(new_detail)
  53. db.commit()
  54. return {"code": 200, "msg": "创建成功", "data": None}
  55. except Exception as e:
  56. traceback.print_exc()
  57. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  58. @router.put("/update/{id}")
  59. async def update_pattern(
  60. id :str ,
  61. user_id=Depends(valid_access_token),
  62. body=Depends(remove_xss_json),
  63. db: Session = Depends(get_db)
  64. ):
  65. try:
  66. update_declaration = declaration_id_get_declaration_info(db,id)
  67. if not update_declaration:
  68. return JSONResponse(status_code=404,content={"code":404,"msg":"declaration not found"})
  69. if str(update_declaration.declaration_person) != str(user_id):
  70. return JSONResponse(status_code=404, content={"code": 404, "msg": "declaration not permission"})
  71. update_declaration.declaration_amount = body['declaration_amount']
  72. update_declaration.update_by = user_id
  73. detail_list = []
  74. for info in body['detail']:
  75. if info['id']=='':
  76. new_detail = ResourceProvisionProcurementDeclarationDetail(
  77. id=new_guid(),
  78. serial_number=info['serial_number'],
  79. declaration_id=update_declaration.id,
  80. warehouse_id = info['warehouse_id'],
  81. material_type=info['material_type'],
  82. material_code=info['material_code'],
  83. material_quantity=info['material_quantity'],
  84. material_unit_price=info['material_unit_price'],
  85. material_purpose=info['material_purpose'],
  86. create_by=user_id
  87. )
  88. db.add(new_detail)
  89. detail_list.append(new_detail.id)
  90. else:
  91. detail = detail_id_get_declaration_detail_info(db,info['id'])
  92. if detail is None:
  93. new_detail = ResourceProvisionProcurementDeclarationDetail(
  94. id=new_guid(),
  95. serial_number=info['serial_number'],
  96. declaration_id=update_declaration.id,
  97. warehouse_id = info['warehouse_id'],
  98. material_type=info['material_type'],
  99. material_code=info['material_code'],
  100. material_quantity=info['material_quantity'],
  101. material_unit_price=info['material_unit_price'],
  102. material_purpose=info['material_purpose'],
  103. create_by=user_id
  104. )
  105. db.add(new_detail)
  106. detail_list.append(new_detail.id)
  107. else:
  108. detail.serial_number = info['serial_number']
  109. detail.warehouse_id = info['warehouse_id']
  110. detail.material_type = info['material_type']
  111. detail.material_code = info['material_code']
  112. detail.material_quantity = info['material_quantity']
  113. detail.material_unit_price = info['material_unit_price']
  114. detail.material_purpose = info['material_purpose']
  115. detail.update_by = user_id
  116. detail_list.append(info['id'])
  117. query = db.query(ResourceProvisionProcurementDeclarationDetail)
  118. query = query.filter_by(declaration_id=id, del_flag='0').filter(~ResourceProvisionProcurementDeclarationDetail.id.in_(detail_list))
  119. for i in query.all():
  120. i.del_flag='2'
  121. db.commit()
  122. return {"code": 200, "msg": "更新成功"}
  123. except Exception as e:
  124. traceback.print_exc()
  125. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  126. @router.get("/info/{id}")
  127. async def get_pattern_info(
  128. id: str,
  129. db: Session = Depends(get_db)
  130. ):
  131. try:
  132. info = declaration_id_get_declaration_info(db,id)
  133. if not info:
  134. return JSONResponse(status_code=404,content={"code":404,"msg":"type not found"})
  135. dept_info = dept_id_get_dept_info(db,info.declaration_unit)
  136. user_info = user_id_get_user_info(db,info.declaration_person)
  137. review_info = declaration_id_get_declaration_review_info(db,id)
  138. if review_info is None:
  139. declaration_details = info.declaration_details
  140. else:
  141. declaration_details = review_info.review_result
  142. data = {"id": info.id,
  143. "declaration_date": info.declaration_date,
  144. "declaration_amount": info.declaration_amount,
  145. "declaration_unit": info.declaration_unit,
  146. "declaration_unit_name": dept_info.dept_name,
  147. "declaration_person": info.declaration_person,
  148. "declaration_person_name": user_info.user_name,
  149. "declaration_details": declaration_details,
  150. "create_time": info.create_time}
  151. detail=[]
  152. for detail_info in declaration_id_get_declaration_detail_list(db,id):
  153. material_type_info = type_id_get_material_type_info(db,detail_info.material_type)
  154. material_into = material_id_get_material_info(db,detail_info.material_code)
  155. warehouse_info = warehouse_id_get_warehouse_info(db,detail_info.warehouse_id)
  156. detail.append({"id":detail_info.id,
  157. "serial_number":detail_info.serial_number,
  158. "warehouse_id":detail_info.warehouse_id,
  159. "warehouse_name":warehouse_info.warehouse_name,
  160. "material_type":detail_info.material_type,
  161. "material_category_name":material_type_info.material_category_name,
  162. "material_code":detail_info.material_code,
  163. "material_name":material_into.material_name,
  164. "material_quantity":detail_info.material_quantity,
  165. "material_unit_price":detail_info.material_unit_price,
  166. "material_purpose":detail_info.material_purpose})
  167. return {"code": 200, "msg": "获取成功", "data": data}
  168. except Exception as e:
  169. traceback.print_exc()
  170. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  171. @router.get("/list")
  172. async def get_pattern_list(
  173. # declaration_details: str = Query(None, description='名称'),
  174. page: int = Query(1, gt=0, description='页码'),
  175. pageSize: int = Query(None, gt=0, description='每页条目数量'),
  176. db: Session = Depends(get_db)
  177. ):
  178. try:
  179. query = db.query(ResourceProvisionProcurementDeclaration)
  180. query = query.filter_by(del_flag='0')
  181. # if declaration_details:
  182. # query = query.filter(ResourceProvisionProcurementDeclaration.declaration_details==declaration_details)
  183. total_items = query.count()
  184. # 排序
  185. if pageSize is None:
  186. pageSize=total_items
  187. query = query.order_by(ResourceProvisionProcurementDeclaration.declaration_date.desc())
  188. # 执行分页查询
  189. lists = query.offset((page - 1) * pageSize).limit(pageSize).all()
  190. data = []
  191. for info in lists:
  192. dept_info = dept_id_get_dept_info(db, info.declaration_unit)
  193. user_info = user_id_get_user_info(db, info.declaration_person)
  194. review_info = declaration_id_get_declaration_review_info(db, info.id)
  195. if review_info is None:
  196. declaration_details = info.declaration_details
  197. else:
  198. declaration_details = review_info.review_result
  199. data = {"id": info.id,
  200. "declaration_date": info.declaration_date,
  201. "declaration_amount": info.declaration_amount,
  202. "declaration_unit": info.declaration_unit,
  203. "declaration_unit_name": dept_info.dept_name,
  204. "declaration_person": info.declaration_person,
  205. "declaration_person_name": user_info.user_name,
  206. "declaration_details": declaration_details,
  207. "create_time": info.create_time}
  208. return {"code": 200, "msg": "查询成功", "data": data,
  209. "total": total_items,
  210. "page": page,
  211. "pageSize": pageSize,
  212. "totalPages": (total_items + pageSize - 1) // pageSize
  213. }
  214. except Exception as e:
  215. traceback.print_exc()
  216. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  217. # @router.delete("/delete/{id}")
  218. # async def delete_pattern(
  219. # id: str,
  220. # db: Session = Depends(get_db)
  221. # ):
  222. # try:
  223. # # 检查图案是否存在
  224. # info = declaration_id_get_declaration_info(db, id)
  225. # if not info:
  226. # return JSONResponse(status_code=404, content={"code": 404, "msg": "type not found"})
  227. # info.del_flag='2'
  228. # review_info = declaration_id_get_declaration_review_info(db, id)
  229. # review_info.del_flag ='2'
  230. # for detail_info in declaration_id_get_declaration_detail_list(db, id):
  231. # detail_info.del_flag='2'
  232. # db.commit()
  233. # return {"code": 200, "msg": "删除成功"}
  234. # except Exception as e:
  235. # traceback.print_exc()
  236. # raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  237. @router.post("/review")
  238. async def create_pattern(
  239. user_id=Depends(valid_access_token),
  240. body = Depends(remove_xss_json),
  241. db: Session = Depends(get_db)
  242. ):
  243. try:
  244. new_review = ResourceProvisionProcurementReview(
  245. id=new_guid(),
  246. reviewer=user_id,
  247. declaration_id=body['declaration_id'],
  248. review_result=body['review_result'],
  249. review_comments=body['review_comments'],
  250. create_by = user_id
  251. )
  252. db.add(new_review)
  253. db.commit()
  254. return {"code": 200, "msg": "审核成功", "data": None}
  255. except Exception as e:
  256. traceback.print_exc()
  257. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  258. @router.get("/review/list")
  259. async def get_pattern_list(
  260. page: int = Query(1, gt=0, description='页码'),
  261. pageSize: int = Query(None, gt=0, description='每页条目数量'),
  262. db: Session = Depends(get_db)
  263. ):
  264. try:
  265. review_list = [info.declaration_id for info in get_declaration_review_list(db)]
  266. query = db.query(ResourceProvisionProcurementDeclaration)
  267. query = query.filter_by(del_flag='0')
  268. query = query.filter(~ResourceProvisionProcurementDeclaration.id.in_(review_list))
  269. total_items = query.count()
  270. # 排序
  271. if pageSize is None:
  272. pageSize=total_items
  273. query = query.order_by(ResourceProvisionProcurementDeclaration.declaration_date.desc())
  274. # 执行分页查询
  275. lists = query.offset((page - 1) * pageSize).limit(pageSize).all()
  276. data = []
  277. for info in lists:
  278. dept_info = dept_id_get_dept_info(db, info.declaration_unit)
  279. user_info = user_id_get_user_info(db, info.declaration_person)
  280. # review_info = declaration_id_get_declaration_review_info(db, info.id)
  281. # if review_info is None:
  282. # declaration_details = info.declaration_details
  283. # else:
  284. # declaration_details = review_info.review_result
  285. data = {"id": info.id,
  286. "declaration_date": info.declaration_date,
  287. "declaration_amount": info.declaration_amount,
  288. "declaration_unit": info.declaration_unit,
  289. "declaration_unit_name": dept_info.dept_name,
  290. "declaration_person": info.declaration_person,
  291. "declaration_person_name": user_info.user_name,
  292. "declaration_details": info.declaration_details,
  293. "create_time": info.create_time}
  294. return {"code": 200, "msg": "查询成功", "data": data,
  295. "total": total_items,
  296. "page": page,
  297. "pageSize": pageSize,
  298. "totalPages": (total_items + pageSize - 1) // pageSize
  299. }
  300. except Exception as e:
  301. traceback.print_exc()
  302. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")