__init__.py 36 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. from fastapi import APIRouter, Request, Depends,Query, HTTPException, status
  4. from common.security import valid_access_token
  5. from fastapi.responses import JSONResponse
  6. from sqlalchemy.orm import Session
  7. from sqlalchemy import and_, or_
  8. from pydantic import BaseModel
  9. from datetime import datetime
  10. from database import get_db
  11. from typing import List
  12. from models import *
  13. from utils import *
  14. import json
  15. import traceback
  16. from . import event
  17. router = APIRouter()
  18. router.include_router(event.router, prefix="/event", tags=["事件相关"])
  19. class File(BaseModel):
  20. status :str
  21. name : str
  22. url : str
  23. class PlanCreateForm(BaseModel):
  24. planName : str
  25. planType : int
  26. # planTypeDesc : str
  27. publishDate: str
  28. organizingUnit : str
  29. document : str
  30. fileList : List[File]
  31. class PlanUpdateForm(BaseModel):
  32. planUid : str
  33. planName : str = None
  34. planType : int = None
  35. publishDate: str = None
  36. organizingUnit : str = None
  37. document : str = None
  38. fileList : List[File] = []
  39. class DrillCreateForm(BaseModel):
  40. planId : str
  41. drillName : str
  42. drillUnit : str
  43. year : int
  44. drillTime: str
  45. drillAddress : str
  46. lon : float
  47. lat : float
  48. drillProject : List[File]
  49. drillVideo : List[File]
  50. drillPicture : List[File]
  51. class DrillUpdateForm(BaseModel):
  52. drillId : str
  53. planId : str = None
  54. drillName : str = None
  55. drillUnit : str = None
  56. year : int = None
  57. drillTime: str = None
  58. drillAddress : str = None
  59. lon : float = None
  60. lat : float = None
  61. drillProject : List[File] = []
  62. drillVideo : List[File] = []
  63. drillPicture : List[File] = []
  64. class TrainingCreateForm(BaseModel):
  65. planId:str
  66. theme : str
  67. unitName : str
  68. trainingWay : str
  69. peopleNum: int
  70. startTime : str
  71. endTime : str
  72. Content :str
  73. address : str
  74. lon : float
  75. lat : float
  76. class TrainingUpdateForm(BaseModel):
  77. trainingId : str
  78. planId:str = None
  79. theme : str = None
  80. unitName : str = None
  81. trainingWay : str = None
  82. peopleNum: int = None
  83. startTime : str = None
  84. endTime : str = None
  85. Content :str = None
  86. address : str = None
  87. lon : float = None
  88. lat : float = None
  89. def get_file_query_fun(from_scenario,foreign_key,db: Session):
  90. file_query = db.query(EmergencyFile)
  91. file_query = file_query.filter(EmergencyFile.del_flag != '2')
  92. file_query = file_query.filter(EmergencyFile.from_scenario == from_scenario)
  93. file_query = file_query.filter(EmergencyFile.foreign_key == foreign_key)
  94. files = file_query.all()
  95. result = [{
  96. "uid": file.file_id,
  97. "status": file.status,
  98. "name": file.file_name_desc,
  99. "url": file.file_name #"/api/file/download/%s" %
  100. } for file in files]
  101. return result
  102. def delete_file_fun(from_scenario,foreign_key,db: Session):
  103. file_query = db.query(EmergencyFile)
  104. file_query = file_query.filter(EmergencyFile.del_flag != '2')
  105. file_query = file_query.filter(EmergencyFile.from_scenario == from_scenario)
  106. file_query = file_query.filter(EmergencyFile.foreign_key == foreign_key)
  107. files = file_query.all()
  108. for file in files:
  109. file.del_flag='2'
  110. # db.commit()
  111. def create_file_fun(file,from_scenario,foreign_key,db: Session):
  112. file_name = file.url
  113. file_name_desc = file.name
  114. status = file.status
  115. new_file = EmergencyFile(
  116. file_id=new_guid(),
  117. foreign_key=foreign_key,
  118. from_scenario=from_scenario,
  119. file_name=file_name,
  120. file_name_desc=file_name_desc,
  121. status=status
  122. )
  123. db.add(new_file)
  124. # db.commit()
  125. # db.refresh(new_file)
  126. @router.get('/plan/list')
  127. async def get_emergency_plan_list(
  128. planType: str = Query(None, description='预案类型'),
  129. # publishDate: datetime = Query(None, description='发布日期'),
  130. beginTime :str = Query(None, description='发布日期起始'),
  131. endTime :str = Query(None, description='发布日期结束'),
  132. # planName: str = Query(None, description='预案名称'),
  133. keywords : str = Query(None, description='预案名称/编制单位'),
  134. page: int = Query(1, gt=0, description='页码'),
  135. pageSize: int = Query(5, gt=0, description='每页条目数量'),
  136. db: Session = Depends(get_db),
  137. user_id = Depends(valid_access_token)
  138. ):
  139. try:
  140. # 构建查询
  141. query = db.query(EmergencyPlan)
  142. query = query.filter(EmergencyPlan.del_flag != '2')
  143. # 应用查询条件
  144. if planType:
  145. query = query.filter(EmergencyPlan.plan_type == planType)
  146. # if publishDate:
  147. # query = query.filter(EmergencyPlan.publish_date == publishDate)
  148. if beginTime:
  149. query = query.filter(EmergencyPlan.publish_date >= beginTime)
  150. if endTime:
  151. query = query.filter(EmergencyPlan.publish_date <= endTime)
  152. if keywords:
  153. query = query.filter(
  154. or_(
  155. EmergencyPlan.plan_name.like(f'%{keywords}%'),
  156. EmergencyPlan.organizing_unit.like(f'%{keywords}%')
  157. )
  158. )
  159. # query = query.filter(EmergencyPlan.plan_name.like(f'%{planName}%'))
  160. # 计算总条目数
  161. total_items = query.count()
  162. # 排序
  163. query = query.order_by(EmergencyPlan.create_time.desc())
  164. # 执行分页查询
  165. emergency_plans = query.offset((page - 1) * pageSize).limit(pageSize).all()
  166. # 将查询结果转换为列表形式的字典
  167. emergency_plans_list = [
  168. {
  169. "planUid": plan.plan_id,
  170. "planId": plan.plan_number,
  171. "planName": plan.plan_name,
  172. "planType": plan.plan_type,
  173. "organizingUnit": plan.organizing_unit,
  174. "document": plan.document_number,
  175. "publishDate": plan.publish_date.strftime('%Y-%m-%d')
  176. }
  177. for plan in emergency_plans
  178. ]
  179. # 返回结果
  180. return {
  181. "code": 200,
  182. "msg": "成功获取预案列表",
  183. "data": emergency_plans_list,
  184. "total": total_items,
  185. "page": page,
  186. "pageSize": pageSize,
  187. "totalPages": (total_items + pageSize - 1) // pageSize
  188. }
  189. except Exception as e:
  190. # 处理异常
  191. raise HTTPException(status_code=500, detail=str(e))
  192. @router.get('/plan/{planId}')
  193. async def get_emergency_plan(
  194. planId: str ,
  195. db: Session = Depends(get_db),
  196. user_id = Depends(valid_access_token)
  197. ):
  198. try:
  199. # 构建查询
  200. query = db.query(EmergencyPlan)
  201. query = query.filter(EmergencyPlan.del_flag != '2')
  202. # 应用查询条件
  203. if planId:
  204. query = query.filter(EmergencyPlan.plan_number == planId)
  205. # 执行查询
  206. emergency_plan = query.first()
  207. if not emergency_plan:
  208. detail = "预案不存在"
  209. raise HTTPException(status_code=404, detail="预案不存在")
  210. # 将查询结果转换为列表形式的字典
  211. emergency_plan_result = {
  212. "planUid": emergency_plan.plan_id,
  213. "planId": emergency_plan.plan_number,
  214. "planName": emergency_plan.plan_name,
  215. "planType": emergency_plan.plan_type,
  216. "organizingUnit": emergency_plan.organizing_unit,
  217. "document": emergency_plan.document_number,
  218. "publishDate": emergency_plan.publish_date.strftime('%Y-%m-%d'),
  219. "fileList": get_file_query_fun(from_scenario='emergencyPlans_plan',foreign_key=emergency_plan.plan_id,db=db)
  220. }
  221. # 返回结果
  222. return {
  223. "code": 200,
  224. "msg": "成功获取预案详情",
  225. "data": emergency_plan_result
  226. }
  227. except Exception as e:
  228. # 处理异常
  229. if str(e)=='':
  230. e = detail
  231. raise HTTPException(status_code=500, detail=str(e))
  232. @router.post('/plan/create')
  233. async def create_emergency_plan(
  234. form_data:PlanCreateForm,
  235. db: Session = Depends(get_db),
  236. body = Depends(remove_xss_json),
  237. user_id = Depends(valid_access_token)
  238. ):
  239. try:
  240. # 提取请求数据
  241. plan_name = form_data.planName
  242. plan_type = form_data.planType
  243. # plan_type_desc = form_data.planTypeDesc
  244. publish_date = form_data.publishDate # 如果没有提供发布日期,则使用当前时间
  245. organizing_unit = form_data.organizingUnit # 使用用户位置作为编制单位
  246. document_number = form_data.document
  247. # 创建新的预案记录
  248. new_plan = EmergencyPlan(
  249. plan_id=new_guid(), # 假设使用 UUID 作为预案 UID
  250. plan_name=plan_name,
  251. plan_type=plan_type,
  252. # plan_type_desc=plan_type_desc,
  253. publish_date=publish_date,
  254. organizing_unit=organizing_unit,
  255. document_number=document_number,
  256. create_by = user_id
  257. )
  258. # 添加到数据库会话并提交
  259. db.add(new_plan)
  260. db.commit()
  261. db.refresh(new_plan) # 可选,如果需要刷新实例状态
  262. new_plan.plan_number= f'YJYA{str(new_plan.id).zfill(10)}'
  263. db.commit()
  264. for file in form_data.fileList:
  265. create_file_fun(file=file, from_scenario='emergencyPlans_plan', foreign_key=new_plan.plan_id, db=db)
  266. db.commit()
  267. # 返回创建成功的响应
  268. return {
  269. "code": 200,
  270. "msg": "预案创建成功",
  271. "data": None
  272. }
  273. except Exception as e:
  274. # 处理异常
  275. raise HTTPException(status_code=500, detail=str(e))
  276. @router.put('/plan/update')
  277. async def update_emergency_plan(
  278. form_data: PlanUpdateForm,
  279. db: Session = Depends(get_db),
  280. body = Depends(remove_xss_json),
  281. user_id = Depends(valid_access_token)
  282. ):
  283. try:
  284. # 提取请求数据
  285. query = db.query(EmergencyPlan)
  286. query = query.filter(EmergencyPlan.plan_id == form_data.planUid)
  287. query = query.filter(EmergencyPlan.del_flag != '2')
  288. plan = query.first()
  289. if not plan:
  290. detail = "预案不存在"
  291. raise HTTPException(status_code=404, detail="预案不存在")
  292. if form_data.planName:
  293. plan.plan_name = form_data.planName
  294. if form_data.planType:
  295. plan.plan_type = form_data.planType
  296. if form_data.publishDate:
  297. plan.publish_date = form_data.publishDate
  298. if form_data.organizingUnit:
  299. plan.organizing_unit = form_data.organizingUnit
  300. if form_data.document:
  301. plan.document_number = form_data.document
  302. if len(form_data.fileList)>0:
  303. delete_file_fun(from_scenario='emergencyPlans_plan',foreign_key=plan.plan_id,db=db)
  304. for file in form_data.fileList:
  305. create_file_fun(file=file,from_scenario='emergencyPlans_plan',foreign_key=plan.plan_id,db=db)
  306. if user_id:
  307. plan.create_by = user_id
  308. # 更新到数据库会话并提交
  309. db.commit()
  310. db.refresh(plan) # 可选,如果需要刷新实例状态
  311. # 返回创建成功的响应
  312. return {
  313. "code": 200,
  314. "msg": "预案更新成功",
  315. "data": None
  316. }
  317. except Exception as e:
  318. # 处理异常
  319. if str(e)=='':
  320. e = detail
  321. raise HTTPException(status_code=500, detail=str(e))
  322. @router.delete('/plan/delete')
  323. async def delete_emergency_plans(
  324. planUids: list,
  325. db: Session = Depends(get_db),
  326. body = Depends(remove_xss_json),
  327. user_id = Depends(valid_access_token)
  328. ):
  329. try:
  330. # 提取请求数据
  331. query = db.query(EmergencyPlan)
  332. query = query.filter(EmergencyPlan.del_flag != '2')
  333. query = query.filter(EmergencyPlan.plan_id.in_(planUids))
  334. plans = query.all()
  335. if not plans:
  336. detail = "预案不存在"
  337. raise HTTPException(status_code=404, detail="预案不存在")
  338. for plan in plans:
  339. plan.del_flag = '2'
  340. plan.create_by=user_id
  341. # 更新到数据库会话并提交
  342. db.commit()
  343. # 返回创建成功的响应
  344. return {
  345. "code": 200,
  346. "msg": "预案删除成功",
  347. "data": None
  348. }
  349. except Exception as e:
  350. # 处理异常
  351. if str(e) == '':
  352. e = detail
  353. raise HTTPException(status_code=500, detail=str(e))
  354. @router.delete('/plan/delete/{planUid}')
  355. async def delete_emergency_plan(
  356. planUid: str,
  357. db: Session = Depends(get_db),
  358. body = Depends(remove_xss_json),
  359. user_id = Depends(valid_access_token)
  360. ):
  361. try:
  362. # 提取请求数据
  363. query = db.query(EmergencyPlan)
  364. query = query.filter(EmergencyPlan.del_flag != '2')
  365. query = query.filter(EmergencyPlan.plan_id == planUid)
  366. plan = query.first()
  367. if not plan:
  368. detail = "预案不存在"
  369. raise HTTPException(status_code=404, detail="预案不存在")
  370. plan.del_flag = '2'
  371. plan.create_by = user_id
  372. # 更新到数据库会话并提交
  373. db.commit()
  374. db.refresh(plan) # 可选,如果需要刷新实例状态
  375. # 返回创建成功的响应
  376. return {
  377. "code": 200,
  378. "msg": "预案删除成功",
  379. "data": None
  380. }
  381. except Exception as e:
  382. # 处理异常
  383. if str(e) == '':
  384. e = detail
  385. raise HTTPException(status_code=500, detail=str(e))
  386. @router.get('/drill/list')
  387. async def get_emergency_drill_list(
  388. planNum: str = Query(None, description='预案编号'),
  389. page: int = Query(1, gt=0, description='页码'),
  390. pageSize: int = Query(5, gt=0, description='每页条目数量'),
  391. db: Session = Depends(get_db),
  392. user_id = Depends(valid_access_token)
  393. ):
  394. try:
  395. # 构建查询
  396. query = db.query(EmergencyDrill)
  397. query = query.filter(EmergencyDrill.del_flag != '2')
  398. # 应用查询条件
  399. if planNum:
  400. query = query.filter(EmergencyDrill.plan_number == planNum)
  401. else:
  402. detail = "planNum不存在"
  403. raise HTTPException(status_code=404, detail="planNum不存在")
  404. # 计算总条目数
  405. total_items = query.count()
  406. query = query.order_by(EmergencyDrill.create_time.desc())
  407. # 执行分页查询
  408. emergency_drill = query.offset((page - 1) * pageSize).limit(pageSize).all() #
  409. print(emergency_drill)
  410. # 将查询结果转换为列表形式的字典
  411. emergency_drill_list = []
  412. for drill in emergency_drill:
  413. emergency_drill = {
  414. "drillId": drill.drill_id,
  415. "drillName": drill.drill_name,
  416. "drillUnit": drill.organizing_unit,
  417. "year": drill.planned_annual,
  418. "drillTime": drill.planned_time.strftime('%Y-%m-%d %H:%M:%S'),
  419. "drillAddress": drill.drill_location,
  420. "drillProject": get_file_query_fun(from_scenario='emergencyPlans_drill_project', foreign_key=drill.drill_id,db=db),
  421. "drillVideo": get_file_query_fun(from_scenario='emergencyPlans_drill_video', foreign_key=drill.drill_id,db=db),
  422. "drillPicture": get_file_query_fun(from_scenario='emergencyPlans_drill_pic', foreign_key=drill.drill_id,db=db)
  423. }
  424. emergency_drill_list.append(emergency_drill)
  425. # 返回结果
  426. return {
  427. "code": 200,
  428. "msg": "成功",
  429. "data": emergency_drill_list,
  430. "total": total_items,
  431. "page": page,
  432. "pageSize": pageSize,
  433. "totalPages": (total_items + pageSize - 1) // pageSize
  434. }
  435. except Exception as e:
  436. # 处理异常
  437. if str(e) == '':
  438. e = detail
  439. raise HTTPException(status_code=500, detail=str(e))
  440. @router.get('/drill/{drillId}')
  441. async def get_emergency_drill(
  442. drillId: str = Query(None, description='演练编号'),
  443. db: Session = Depends(get_db),
  444. body = Depends(remove_xss_json),
  445. user_id = Depends(valid_access_token)
  446. ):
  447. try:
  448. # 构建查询
  449. query = db.query(EmergencyDrill)
  450. query = query.filter(EmergencyDrill.del_flag != '2')
  451. # 应用查询条件
  452. if drillId:
  453. query = query.filter(EmergencyDrill.drill_id == drillId)
  454. else:
  455. detail = "drillId不存在"
  456. raise HTTPException(status_code=404, detail="drillId不存在")
  457. # 执行查询
  458. drill = query.first()
  459. # 将查询结果转换为字典形式
  460. emergency_drill = {
  461. "drillId": drill.drill_id,
  462. "drillName": drill.drill_name,
  463. "drillUnit": drill.organizing_unit,
  464. "year": drill.planned_annual,
  465. "drillTime": drill.planned_time.strftime('%Y-%m-%d %H:%M:%S'),
  466. "drillAddress": drill.drill_location,
  467. "drillProject": get_file_query_fun(from_scenario='emergencyPlans_drill_project',foreign_key=drill.drill_id,db=db),
  468. "drillVideo": get_file_query_fun(from_scenario='emergencyPlans_drill_video', foreign_key=drill.drill_id,db=db),
  469. "drillPicture": get_file_query_fun(from_scenario='emergencyPlans_drill_pic', foreign_key=drill.drill_id,db=db)
  470. }
  471. # 返回结果
  472. return {
  473. "code": 200,
  474. "msg": "成功",
  475. "data": emergency_drill
  476. }
  477. except Exception as e:
  478. # 处理异常
  479. if str(e) == '':
  480. e = detail
  481. raise HTTPException(status_code=500, detail=str(e))
  482. @router.post('/drill/create')
  483. async def create_emergency_drill(
  484. form_data:DrillCreateForm,
  485. db: Session = Depends(get_db),
  486. body = Depends(remove_xss_json),
  487. user_id = Depends(valid_access_token)
  488. ):
  489. try:
  490. # 提取请求数据
  491. plan_number = form_data.planId
  492. drill_name = form_data.drillName
  493. organizing_unit = form_data.drillUnit
  494. planned_annual = form_data.year
  495. planned_time = form_data.drillTime
  496. drill_location = form_data.drillAddress
  497. drill_lon = form_data.lon
  498. drill_lat = form_data.lat
  499. # 创建新的预案记录
  500. new_drill = EmergencyDrill(
  501. drill_id=new_guid(), # 假设使用 UUID 作为预案 UID
  502. plan_number = plan_number,
  503. drill_name=drill_name,
  504. organizing_unit = organizing_unit,
  505. planned_annual = planned_annual,
  506. planned_time = planned_time,
  507. drill_location = drill_location,
  508. drill_lon = drill_lon,
  509. drill_lat = drill_lat,
  510. create_by = user_id
  511. )
  512. # 添加到数据库会话并提交
  513. db.add(new_drill)
  514. db.commit()
  515. db.refresh(new_drill) # 可选,如果需要刷新实例状态
  516. for file in form_data.drillProject:
  517. create_file_fun(file=file, from_scenario='emergencyPlans_drill_project', foreign_key=new_drill.drill_id, db=db)
  518. for file in form_data.drillVideo:
  519. create_file_fun(file=file, from_scenario='emergencyPlans_drill_video', foreign_key=new_drill.drill_id, db=db)
  520. for file in form_data.drillPicture:
  521. create_file_fun(file=file, from_scenario='emergencyPlans_drill_pic', foreign_key=new_drill.drill_id, db=db)
  522. db.commit()
  523. # 返回创建成功的响应
  524. return {
  525. "code": 200,
  526. "msg": "演练创建成功",
  527. "data": None
  528. }
  529. except Exception as e:
  530. # 处理异常
  531. raise HTTPException(status_code=500, detail=str(e))
  532. @router.put('/drill/update')
  533. async def update_emergency_drill(
  534. form_data:DrillUpdateForm,
  535. db: Session = Depends(get_db),
  536. body = Depends(remove_xss_json),
  537. user_id = Depends(valid_access_token)
  538. ):
  539. try:
  540. query = db.query(EmergencyDrill)
  541. query = query.filter(EmergencyDrill.drill_id == form_data.drillId)
  542. query = query.filter(EmergencyDrill.del_flag != '2')
  543. drill = query.first()
  544. # 判断预案编号是否存在,存在即更新
  545. if form_data.planId:
  546. drill.plan_number=form_data.planId
  547. # 判断演练名称是否存在,存在即更新
  548. if form_data.drillName:
  549. drill.drill_name=form_data.drillName
  550. # 判断演练单位是否存在,存在即更新
  551. if form_data.drillUnit:
  552. drill.organizing_unit=form_data.drillUnit
  553. # 判断演练年份是否存在,存在即更新
  554. if form_data.year:
  555. drill.planned_annual=form_data.year
  556. # 判断演练日期是否存在,存在即更新
  557. if form_data.drillTime:
  558. drill.planned_time=form_data.drillTime
  559. # 判断演练地址是否存在,存在即更新
  560. if form_data.drillAddress:
  561. drill.drill_location=form_data.drillAddress
  562. # 判断经度是否存在,存在即更新
  563. if form_data.lon:
  564. drill.drill_lon=form_data.lon
  565. # 判断纬度是否存在,存在即更新
  566. if form_data.lat:
  567. drill.drill_lat=form_data.lat
  568. # 判断演练方案是否存在,存在即更新
  569. if len(form_data.drillProject)>0:
  570. delete_file_fun(from_scenario='emergencyPlans_drill_project',foreign_key=drill.drill_id,db=db)
  571. for file in form_data.drillProject:
  572. create_file_fun(file=file,from_scenario='emergencyPlans_drill_project',foreign_key=drill.drill_id,db=db)
  573. # 判断演练视频是否存在,存在即更新
  574. if len(form_data.drillVideo) > 0:
  575. delete_file_fun(from_scenario='emergencyPlans_drill_video', foreign_key=drill.drill_id, db=db)
  576. for file in form_data.drillVideo:
  577. create_file_fun(file=file, from_scenario='emergencyPlans_drill_video', foreign_key=drill.drill_id,db=db)
  578. # 判断演练图片是否存在,存在即更新
  579. if len(form_data.drillPicture) > 0:
  580. delete_file_fun(from_scenario='emergencyPlans_drill_pic', foreign_key=drill.drill_id, db=db)
  581. for file in form_data.drillPicture:
  582. create_file_fun(file=file, from_scenario='emergencyPlans_drill_pic', foreign_key=drill.drill_id,db=db)
  583. if user_id:
  584. drill.create_by = user_id
  585. # 提交所有更新
  586. db.commit()
  587. # 返回更新成功的响应
  588. return {
  589. "code": 200,
  590. "msg": "演练更新成功",
  591. "data": None
  592. }
  593. except Exception as e:
  594. # 处理异常
  595. raise HTTPException(status_code=500, detail=str(e))
  596. @router.delete('/drill/delete')
  597. async def delete_emergency_drill(
  598. drillUids: list,
  599. db: Session = Depends(get_db),
  600. body = Depends(remove_xss_json),
  601. user_id = Depends(valid_access_token)
  602. ):
  603. try:
  604. # 提取请求数据
  605. query = db.query(EmergencyDrill)
  606. query = query.filter(EmergencyDrill.del_flag != '2')
  607. query = query.filter(EmergencyDrill.drill_id.in_(drillUids))
  608. drills = query.all()
  609. if not drills:
  610. detail = "演练不存在"
  611. raise HTTPException(status_code=404, detail="演练不存在")
  612. for drill in drills:
  613. drill.del_flag = '2'
  614. drill.create_by=user_id
  615. # 更新到数据库会话并提交
  616. db.commit()
  617. # 返回创建成功的响应
  618. return {
  619. "code": 200,
  620. "msg": "演练删除成功",
  621. "data": None
  622. }
  623. except Exception as e:
  624. # 处理异常
  625. if str(e) == '':
  626. e = detail
  627. raise HTTPException(status_code=500, detail=str(e))
  628. @router.delete('/drill/delete/{drillUid}')
  629. async def delete_emergency_drill(
  630. drillUid: str,
  631. db: Session = Depends(get_db),
  632. user_id = Depends(valid_access_token)
  633. ):
  634. try:
  635. # 提取请求数据
  636. query = db.query(EmergencyDrill)
  637. query = query.filter(EmergencyDrill.del_flag != '2')
  638. query = query.filter(EmergencyDrill.drill_id == drillUid)
  639. drill = query.first()
  640. if not drill:
  641. detail = "演练不存在"
  642. raise HTTPException(status_code=404, detail="演练不存在")
  643. drill.del_flag = '2'
  644. drill.create_by = user_id
  645. # 更新到数据库会话并提交
  646. db.commit()
  647. db.refresh(drill) # 可选,如果需要刷新实例状态
  648. # 返回创建成功的响应
  649. return {
  650. "code": 200,
  651. "msg": "演练删除成功",
  652. "data": None
  653. }
  654. except Exception as e:
  655. # 处理异常
  656. if str(e) == '':
  657. e = detail
  658. raise HTTPException(status_code=500, detail=str(e))
  659. @router.get('/training/list')
  660. async def get_emergency_training_list(
  661. planNum: str = Query(None, description='预案编号'),
  662. page: int = Query(1, gt=0, description='页码'),
  663. pageSize: int = Query(5, gt=0, description='每页条目数量'),
  664. db: Session = Depends(get_db),
  665. user_id = Depends(valid_access_token)
  666. ):
  667. try:
  668. # 构建查询
  669. query = db.query(EmergencyTrainingSession)
  670. query = query.filter(EmergencyTrainingSession.del_flag != '2')
  671. # 应用查询条件
  672. if planNum:
  673. query = query.filter(EmergencyTrainingSession.plan_number == planNum)
  674. else:
  675. detail = "planNum不存在"
  676. raise HTTPException(status_code=404, detail="planNum不存在")
  677. # 计算总条目数
  678. total_items = query.count()
  679. query = query.order_by(EmergencyTrainingSession.start_time.desc())
  680. # 执行分页查询
  681. emergency_training = query.offset((page - 1) * pageSize).limit(pageSize).all() #
  682. # print(emergency_drill)
  683. # 将查询结果转换为列表形式的字典
  684. emergency_training_list = [
  685. {
  686. "trainingId": training.training_id,
  687. "theme": training.training_theme,
  688. "unitName": training.training_unit,
  689. "peopleNum": training.participant_count,
  690. "trainingWay":training.training_method,
  691. "startTime":training.start_time.strftime('%Y-%m-%d %H:%M:%S'),
  692. "endTime": training.end_time.strftime('%Y-%m-%d %H:%M:%S'),
  693. "address": training.training_location,
  694. "Content":training.training_content
  695. }
  696. for training in emergency_training
  697. ]
  698. # 返回结果
  699. return {
  700. "code": 200,
  701. "msg": "成功",
  702. "data": emergency_training_list,
  703. "total": total_items,
  704. "page": page,
  705. "pageSize": pageSize,
  706. "totalPages": (total_items + pageSize - 1) // pageSize
  707. }
  708. except Exception as e:
  709. # 处理异常
  710. if str(e) == '':
  711. e = detail
  712. raise HTTPException(status_code=500, detail=str(e))
  713. @router.get('/training/{trainingId}')
  714. async def get_emergency_training(
  715. trainingId: str = Query(None, description='培训编号'),
  716. db: Session = Depends(get_db),
  717. user_id = Depends(valid_access_token)
  718. ):
  719. try:
  720. # 构建查询
  721. query = db.query(EmergencyTrainingSession)
  722. query = query.filter(EmergencyTrainingSession.del_flag != '2')
  723. # 应用查询条件
  724. if trainingId:
  725. query = query.filter(EmergencyTrainingSession.training_id == trainingId)
  726. else:
  727. detail = "trainingId不存在"
  728. raise HTTPException(status_code=404, detail="trainingId不存在")
  729. # 执行查询
  730. training = query.first() #
  731. # 将查询结果转换为列表形式的字典
  732. emergency_training = {
  733. "trainingId": training.training_id,
  734. "theme": training.training_theme,
  735. "unitName": training.training_unit,
  736. "peopleNum": training.participant_count,
  737. "trainingWay":training.training_method,
  738. "startTime":training.start_time.strftime('%Y-%m-%d %H:%M:%S'),
  739. "endTime": training.end_time.strftime('%Y-%m-%d %H:%M:%S'),
  740. "address": training.training_location,
  741. "Content":training.training_content
  742. }
  743. # 返回结果
  744. return {
  745. "code": 200,
  746. "msg": "成功",
  747. "data": emergency_training
  748. }
  749. except Exception as e:
  750. # 处理异常
  751. if str(e) == '':
  752. e = detail
  753. raise HTTPException(status_code=500, detail=str(e))
  754. @router.post('/training/create')
  755. async def create_emergency_training(
  756. form_data:TrainingCreateForm,
  757. db: Session = Depends(get_db),
  758. user_id = Depends(valid_access_token)
  759. ):
  760. try:
  761. # 创建新的预案记录
  762. new_training = EmergencyTrainingSession(
  763. training_id=new_guid(), # 假设使用 UUID 作为预案 UID
  764. training_theme=form_data.theme,
  765. training_unit=form_data.unitName,
  766. training_method=form_data.trainingWay,
  767. participant_count=form_data.peopleNum,
  768. start_time=form_data.startTime,
  769. end_time=form_data.endTime,
  770. training_content=form_data.Content,
  771. training_location=form_data.address,
  772. training_lon=form_data.lon,
  773. training_lat=form_data.lat,
  774. plan_number = form_data.planId
  775. )
  776. # 添加到数据库会话并提交
  777. db.add(new_training)
  778. db.commit()
  779. db.refresh(new_training) # 可选,如果需要刷新实例状态
  780. # 返回创建成功的响应
  781. return {
  782. "code": 200,
  783. "msg": "培训创建成功",
  784. "data": None
  785. }
  786. except Exception as e:
  787. # 处理异常
  788. raise HTTPException(status_code=500, detail=str(e))
  789. @router.put('/training/update')
  790. async def update_emergency_training(
  791. form_data:TrainingUpdateForm,
  792. db: Session = Depends(get_db),
  793. user_id = Depends(valid_access_token)
  794. ):
  795. try:
  796. query = db.query(EmergencyTrainingSession)
  797. query = query.filter(EmergencyTrainingSession.training_id == form_data.trainingId)
  798. query = query.filter(EmergencyTrainingSession.del_flag != '2')
  799. training = query.first()
  800. # 判断预案编号是否存在,存在即更新
  801. if form_data.planId:
  802. training.plan_number = form_data.planId
  803. # 判断预案编号是否存在,存在即更新
  804. if form_data.theme:
  805. training.training_theme = form_data.theme
  806. # 判断预案编号是否存在,存在即更新
  807. if form_data.unitName:
  808. training.training_unit = form_data.unitName
  809. # 判断预案编号是否存在,存在即更新
  810. if form_data.trainingWay:
  811. training.training_method = form_data.trainingWay
  812. # 判断预案编号是否存在,存在即更新
  813. if form_data.peopleNum:
  814. training.participant_count = form_data.peopleNum
  815. # 判断预案编号是否存在,存在即更新
  816. if form_data.startTime:
  817. training.start_time = form_data.startTime
  818. # 判断预案编号是否存在,存在即更新
  819. if form_data.endTime:
  820. training.end_time = form_data.endTime
  821. # 判断预案编号是否存在,存在即更新
  822. if form_data.Content:
  823. training.training_content = form_data.Content
  824. # 判断预案编号是否存在,存在即更新
  825. if form_data.address:
  826. training.training_location = form_data.address
  827. # 判断预案编号是否存在,存在即更新
  828. if form_data.lon:
  829. training.training_lon = form_data.lon
  830. # 判断预案编号是否存在,存在即更新
  831. if form_data.lat:
  832. training.training_lat = form_data.lat
  833. if user_id:
  834. training.create_by = user_id
  835. # 提交
  836. db.commit()
  837. # 返回创建成功的响应
  838. return {
  839. "code": 200,
  840. "msg": "培训更新成功",
  841. "data": None
  842. }
  843. except Exception as e:
  844. # 处理异常
  845. raise HTTPException(status_code=500, detail=str(e))
  846. @router.delete('/training/delete')
  847. async def delete_emergency_training(
  848. trainingUids: list,
  849. db: Session = Depends(get_db),
  850. user_id = Depends(valid_access_token)
  851. ):
  852. try:
  853. # 提取请求数据
  854. query = db.query(EmergencyTrainingSession)
  855. query = query.filter(EmergencyTrainingSession.del_flag != '2')
  856. query = query.filter(EmergencyTrainingSession.training_id.in_(trainingUids))
  857. trainings = query.all()
  858. if not trainings:
  859. detail = "培训不存在"
  860. raise HTTPException(status_code=404, detail="培训不存在")
  861. for training in trainings:
  862. training.del_flag = '2'
  863. training.create_by = user_id
  864. # 更新到数据库会话并提交
  865. db.commit()
  866. # 返回创建成功的响应
  867. return {
  868. "code": 200,
  869. "msg": "培训删除成功",
  870. "data": None
  871. }
  872. except Exception as e:
  873. # 处理异常
  874. if str(e) == '':
  875. e = detail
  876. raise HTTPException(status_code=500, detail=str(e))
  877. @router.delete('/training/delete/{trainingUid}')
  878. async def delete_emergency_training(
  879. trainingUid: str,
  880. db: Session = Depends(get_db),
  881. user_id = Depends(valid_access_token)
  882. ):
  883. try:
  884. # 提取请求数据
  885. query = db.query(EmergencyTrainingSession)
  886. query = query.filter(EmergencyTrainingSession.del_flag != '2')
  887. query = query.filter(EmergencyTrainingSession.training_id == trainingUid)
  888. training = query.first()
  889. if not training:
  890. detail = "培训不存在"
  891. raise HTTPException(status_code=404, detail="培训不存在")
  892. training.del_flag = '2'
  893. training.create_by = user_id
  894. # 更新到数据库会话并提交
  895. db.commit()
  896. db.refresh(training) # 可选,如果需要刷新实例状态
  897. # 返回创建成功的响应
  898. return {
  899. "code": 200,
  900. "msg": "培训删除成功",
  901. "data": None
  902. }
  903. except Exception as e:
  904. # 处理异常
  905. if str(e) == '':
  906. e = detail
  907. raise HTTPException(status_code=500, detail=str(e))
  908. @router.get('/response/list')
  909. async def get_emergency_response_list(
  910. planNum:str,
  911. page: int = Query(1, gt=0, description='页码'),
  912. pageSize: int = Query(5, gt=0, description='每页条目数量'),
  913. db: Session = Depends(get_db),
  914. user_id = Depends(valid_access_token)
  915. ):
  916. try:
  917. # 查询所有响应事件
  918. query = db.query(EmergencyResponse)
  919. query = query.filter(EmergencyResponse.del_flag!='2')
  920. query = query.filter(EmergencyResponse.plan_id==planNum)
  921. # 计算总条目数
  922. total_items = query.count()
  923. # 排序
  924. query = query.order_by(EmergencyResponse.response_start_time.desc())
  925. # 执行分页查询
  926. responses = query.offset((page - 1) * pageSize).limit(pageSize).all()
  927. # 将查询结果转换为列表形式的字典
  928. response_list = [
  929. {
  930. "responseId": response.response_id,
  931. "eventName": response.event_name,
  932. "eventType": response.event_type,
  933. "responseUnit": response.response_unit,
  934. "eventAddress": response.event_address,
  935. "responseLevel": response.response_level,
  936. "responseStartTime": response.response_start_time.strftime('%Y-%m-%d %H:%M:%S') if response.response_start_time else '',
  937. "responseEndTime": response.response_end_time.strftime('%Y-%m-%d %H:%M:%S') if response.response_end_time else '',
  938. }
  939. for response in responses
  940. ]
  941. # 返回结果
  942. return {
  943. "code": 200,
  944. "msg": "成功获取响应列表",
  945. "data": response_list,
  946. "total": total_items,
  947. "page": page,
  948. "pageSize": pageSize,
  949. "totalPages": (total_items + pageSize - 1) // pageSize
  950. }
  951. except Exception as e:
  952. # 处理异常
  953. traceback.print_exc()
  954. raise HTTPException(status_code=500, detail=str(e))