__init__.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663
  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. import json
  19. import traceback
  20. router = APIRouter()
  21. @router.post("/create")
  22. async def create_pattern(
  23. user_id=Depends(valid_access_token),
  24. body = Depends(remove_xss_json),
  25. db: Session = Depends(get_db)
  26. ):
  27. try:
  28. new_pattern = TpPatternList(
  29. id = new_guid(),
  30. pattern_name=body['pattern_name'],
  31. content=body['content'],
  32. create_dept = user_id
  33. )
  34. db.add(new_pattern)
  35. db.commit()
  36. return {"code": 200, "msg": "创建成功", "data": None}
  37. except Exception as e:
  38. traceback.print_exc()
  39. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  40. @router.put("/update/{pattern_id}")
  41. async def update_pattern(
  42. pattern_id :int ,
  43. user_id=Depends(valid_access_token),
  44. body=Depends(remove_xss_json),
  45. db: Session = Depends(get_db)
  46. ):
  47. try:
  48. query = db.query(TpPatternList)
  49. query = query.filter(TpPatternList.id == pattern_id)
  50. query = query.filter(TpPatternList.del_flag != '2')
  51. update_pattern = query.first()
  52. if not update_pattern:
  53. raise HTTPException(status_code=404, detail="图案不存在")
  54. update_pattern.pattern_name = body['pattern_name']
  55. update_pattern.content = body['content']
  56. update_pattern.create_dept = user_id
  57. db.commit()
  58. return {"code": 200, "msg": "更新成功"}
  59. except Exception as e:
  60. traceback.print_exc()
  61. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  62. @router.get("/info/{pattern_id}")
  63. async def get_pattern_info(
  64. pattern_id: int,
  65. db: Session = Depends(get_db)
  66. ):
  67. try:
  68. query = db.query(TpPatternList)
  69. query = query.filter(TpPatternList.id == pattern_id)
  70. query = query.filter(TpPatternList.del_flag != '2')
  71. pattern = query.first()
  72. # pattern = db.query(TpPatternList).filter(TpPatternList.id == pattern_id).first()
  73. if not pattern:
  74. raise HTTPException(status_code=404, detail="图案不存在")
  75. return {"code": 200, "msg": "获取成功", "data": {"pattern_name": pattern.pattern_name, "content": pattern.content}}
  76. except Exception as e:
  77. traceback.print_exc()
  78. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  79. @router.get("/list")
  80. async def get_pattern_list(
  81. pattern_name: str = Query(None, description='预案名称'),
  82. page: int = Query(1, gt=0, description='页码'),
  83. pageSize: int = Query(5, gt=0, description='每页条目数量'),
  84. db: Session = Depends(get_db)
  85. ):
  86. try:
  87. query = db.query(TpPatternList)
  88. query = query.filter(TpPatternList.del_flag != '2')
  89. if pattern_name:
  90. query = query.filter(TpPatternList.pattern_name.like(f'%{pattern_name}%'))
  91. total_items = query.count()
  92. # 排序
  93. query = query.order_by(TpPatternList.create_time.desc())
  94. # 执行分页查询
  95. patterns = query.offset((page - 1) * pageSize).limit(pageSize).all()
  96. return {"code": 200, "msg": "查询成功", "data": [{"id": p.id, "pattern_name": p.pattern_name, "content": p.content} for p in patterns],
  97. "total": total_items,
  98. "page": page,
  99. "pageSize": pageSize,
  100. "totalPages": (total_items + pageSize - 1) // pageSize
  101. }
  102. except Exception as e:
  103. traceback.print_exc()
  104. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  105. @router.delete("/delete/{pattern_id}")
  106. async def delete_pattern(
  107. pattern_id: int,
  108. db: Session = Depends(get_db)
  109. ):
  110. try:
  111. # 检查图案是否存在
  112. query = db.query(TpPatternList)
  113. query = query.filter(TpPatternList.id == pattern_id)
  114. query = query.filter(TpPatternList.del_flag != '2')
  115. pattern = query.first()
  116. if not pattern:
  117. raise HTTPException(status_code=404, detail="图案不存在")
  118. # 执行删除操作
  119. pattern.del_flag='2'
  120. db.commit()
  121. return {"code": 200, "msg": "删除成功"}
  122. except Exception as e:
  123. traceback.print_exc()
  124. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  125. def user_id_and_pattern_id_get_tp_pattern_ws_info(user_id:str,pattern_id:str,db: Session):
  126. query = db.query(TpPatternWSList)
  127. query = query.filter(TpPatternWSList.del_flag != '2')
  128. query = query.filter(TpPatternWSList.pattern_id == pattern_id)
  129. query = query.filter(TpPatternWSList.create_by == user_id)
  130. query.order_by(TpPatternWSList.create_time.desc())
  131. return query.first()
  132. def pattern_id_get_tp_pattern_ws_users(pattern_id:str,db: Session):
  133. query = db.query(TpPatternWSList)
  134. query = query.filter(TpPatternWSList.del_flag != '2')
  135. query = query.filter(TpPatternWSList.pattern_id == pattern_id)
  136. query = query.group_by(TpPatternWSList.create_by)
  137. return query.all()
  138. def pattern_id_get_tp_pattern_ws_list(pattern_id:str,db: Session):
  139. query = db.query(TpPatternWSList)
  140. query = query.filter(TpPatternWSList.del_flag != '2')
  141. query = query.filter(TpPatternWSList.pattern_id == pattern_id)
  142. # query = query.group_by(TpPatternWSList.create_id)
  143. return query.all()
  144. def pattern_id_get_tp_pattern_ws_count(pattern_id:str,db: Session):
  145. query = db.query(TpPatternWSList)
  146. query = query.filter(TpPatternWSList.del_flag != '2')
  147. query = query.filter(TpPatternWSList.pattern_id == pattern_id)
  148. return query.count()
  149. def pattern_id_get_tp_pattern_ws_max_time(pattern_id:str,db: Session):
  150. query = db.query(TpPatternWSList.update_time)
  151. query = query.filter(TpPatternWSList.del_flag != '2')
  152. query = query.filter(TpPatternWSList.pattern_id == pattern_id)
  153. query.order_by(TpPatternWSUserList.update_time.desc())
  154. return query.first()
  155. def pattern_id_get_tp_pattern_ws_user_list(pattern_id:str,db: Session):
  156. query = db.query(TpPatternWSUserList)
  157. query = query.filter(TpPatternWSUserList.del_flag != '2')
  158. query = query.filter(TpPatternWSUserList.pattern_id == pattern_id)
  159. query.order_by(TpPatternWSUserList.create_time.desc())
  160. return query.all()
  161. def pattern_id_get_tp_pattern_ws_group_def_info(pattern_id:str,db: Session):
  162. query = db.query(TpPatternWSGroupList)
  163. query = query.filter(TpPatternWSGroupList.del_flag != '2')
  164. query = query.filter(TpPatternWSGroupList.pattern_id == pattern_id)
  165. query.order_by(TpPatternWSGroupList.create_time.desc())
  166. return query.first()
  167. def user_id_get_tp_pattern_ws_user_list(user_id:str,db: Session):
  168. query = db.query(TpPatternWSUserList)
  169. query = query.filter(TpPatternWSUserList.del_flag != '2')
  170. query = query.filter(TpPatternWSUserList.user_id == user_id)
  171. query.order_by(TpPatternWSUserList.create_time.desc())
  172. return query.all()
  173. def user_id_and_pattern_id_get_tp_pattern_ws_user_info(user_id:str,pattern_id:str,db: Session):
  174. query = db.query(TpPatternWSUserList)
  175. query = query.filter(TpPatternWSUserList.del_flag != '2')
  176. query = query.filter(TpPatternWSUserList.user_id == user_id)
  177. query = query.filter(TpPatternWSUserList.pattern_id == pattern_id)
  178. query.order_by(TpPatternWSUserList.create_time.desc())
  179. return query.all()
  180. def pattern_id_get_tp_pattern_ws_group_list(pattern_id:str,db: Session):
  181. query = db.query(TpPatternWSGroupList)
  182. query = query.filter(TpPatternWSGroupList.del_flag != '2')
  183. query = query.filter(TpPatternWSGroupList.pattern_id == pattern_id)
  184. query.order_by(TpPatternWSGroupList.create_time.desc())
  185. return query.all()
  186. class ConnectionManager:
  187. def __init__(self):
  188. self.active_connections = {} #: List[WebSocket]
  189. async def connect(self, websocket: WebSocket,pattern_id:str,db: Session):
  190. await websocket.accept()
  191. if pattern_id not in self.active_connections:
  192. self.active_connections[pattern_id] = [websocket]
  193. else:
  194. self.active_connections[pattern_id].append(websocket)
  195. print('连接成功')
  196. # data = pattern_id_get_tp_pattern_ws_list(pattern_id, db)
  197. # message = []
  198. # for info in data:
  199. # user = user_id_get_user_info(db, info.create_by)
  200. # dept = dept_id_get_dept_info(db, user.dept_id)
  201. # message.append({"id": info.id,
  202. # "name": info.name,
  203. # "pattern_id": info.pattern_id,
  204. # "content": info.content,
  205. # "visible": info.visible,
  206. # "user_id": info.create_by,
  207. # "nick_name": user.nick_name,
  208. # "dept_name": dept.dept_name})
  209. # message = json.dumps(message)
  210. # await websocket.send_text(message)
  211. # await websocket.send_text('连接成功')
  212. def disconnect(self, websocket: WebSocket,pattern_id:str):
  213. # await websocket.send_text('已断开')
  214. print('已断开')
  215. self.active_connections[pattern_id].remove(websocket)
  216. if not self.active_connections[pattern_id]:
  217. del self.active_connections[pattern_id]
  218. async def broadcast(self,pattern_id:str,db: Session):#, message: str
  219. data = pattern_id_get_tp_pattern_ws_list(pattern_id,db)
  220. message = []
  221. for info in data:
  222. user= user_id_get_user_info(db,info.create_by)
  223. dept = dept_id_get_dept_info(db,user.dept_id)
  224. message.append({"id":info.id,
  225. "name":info.name,
  226. "pattern_id":info.pattern_id,
  227. "content":info.content,
  228. "visible":info.visible,
  229. "user_id":info.create_by,
  230. "nick_name":user.nick_name,
  231. "dept_name":dept.dept_name})
  232. message = json.dumps(message)
  233. for connection in self.active_connections[pattern_id]:
  234. await connection.send_text(message)
  235. manager = ConnectionManager()
  236. @router.websocket("/{pattern_id}/ws")
  237. async def websocket_endpoint(pattern_id:str ,websocket: WebSocket,db: Session = Depends(get_db)):
  238. await manager.connect(websocket,pattern_id,db)
  239. # return JSONResponse(status_code=404, content={
  240. # 'code': 404,
  241. # 'msg': '抱歉,您无权限,请联系系统管理员'
  242. # })
  243. try:
  244. data = await websocket.receive_text()
  245. user_id = valid_websocket_token(data)
  246. user_list = [i.user_id for i in pattern_id_get_tp_pattern_ws_user_list(pattern_id, db)]
  247. if user_id not in user_list:
  248. # await websocket.send_text('抱歉,您无权限,请联系系统管理员')
  249. manager.disconnect(websocket, pattern_id)
  250. # await manager.broadcast(pattern_id, db) # 广播消息给所有连接
  251. # now_max_time = pattern_id_get_tp_pattern_ws_max_time(pattern_id, db).update_time
  252. while True:
  253. # data = await websocket.receive_text()
  254. # time.sleep(0.5)
  255. # max_time = pattern_id_get_tp_pattern_ws_max_time(pattern_id, db).update_time
  256. # if now_max_time<max_time:
  257. await manager.broadcast(pattern_id,db) # 广播消息给所有连接
  258. except WebSocketDisconnect:
  259. manager.disconnect(websocket,pattern_id)
  260. @router.post("/ws/bz_add")
  261. async def create_pattern(
  262. user_id=Depends(valid_access_token),
  263. body = Depends(remove_xss_json),
  264. db: Session = Depends(get_db)
  265. ):
  266. try:
  267. new_pattern_ws = TpPatternWSList(
  268. id = new_guid(),
  269. name=body['name'],
  270. pattern_id=body['pattern_id'],
  271. content=body['content'],
  272. visible = body['visible'],
  273. create_by = user_id
  274. )
  275. db.add(new_pattern_ws)
  276. db.commit()
  277. # await manager.broadcast(body['pattern_id'], db)
  278. return {"code": 200, "msg": "创建成功", "data": None}
  279. except Exception as e:
  280. traceback.print_exc()
  281. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  282. @router.put("/ws/bz_update")
  283. async def update_pattern(
  284. user_id=Depends(valid_access_token),
  285. body = Depends(remove_xss_json),
  286. db: Session = Depends(get_db)
  287. ):
  288. try:
  289. id = body['id']
  290. query = db.query(TpPatternWSList)
  291. query = query.filter(TpPatternWSList.del_flag!='2')
  292. query = query.filter(TpPatternWSList.id == id)
  293. info = query.first()
  294. if 'name' in body:
  295. info.name=body['name']
  296. info.update_by=user_id
  297. if 'pattern_id' in body:
  298. info.pattern_id=body['pattern_id']
  299. info.update_by=user_id
  300. if 'content' in body:
  301. info.content=body['content']
  302. info.update_by=user_id
  303. if 'visible' in body:
  304. info.visible=body['visible']
  305. info.update_by=user_id
  306. db.commit()
  307. # await manager.broadcast(body['pattern_id'], db)
  308. return {"code": 200, "msg": "更新成功", "data": None}
  309. except Exception as e:
  310. traceback.print_exc()
  311. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  312. @router.post("/ws/create")
  313. async def create_pattern(
  314. user_id=Depends(valid_access_token),
  315. body = Depends(remove_xss_json),
  316. db: Session = Depends(get_db)
  317. ):
  318. try:
  319. user_id_list = body['user_id_list']
  320. if str(user_id) not in user_id_list:
  321. user_id_list.append(str(user_id))
  322. for user in user_id_list:
  323. new_pattern_user_ws = TpPatternWSUserList(
  324. id=new_guid(),
  325. pattern_id=body['pattern_id'],
  326. pattern_name=body['pattern_name'],
  327. user_id=user,
  328. create_by=user_id
  329. )
  330. db.add(new_pattern_user_ws)
  331. new_pattern_ws = TpPatternWSList(
  332. id = new_guid(),
  333. name=body['name'],
  334. pattern_id=body['pattern_id'],
  335. content=body['content'],
  336. visible = body['visible'],
  337. create_by = user_id
  338. )
  339. db.add(new_pattern_ws)
  340. new_pattern_group_ws = TpPatternWSGroupList(
  341. group_id=body['pattern_id'],
  342. group_name='默认分组',
  343. pattern_id=body['pattern_id'],
  344. pattern_name=body['pattern_name'],
  345. create_by=user_id
  346. )
  347. db.add(new_pattern_group_ws)
  348. db.commit()
  349. return {"code": 200, "msg": "创建成功", "data": None}
  350. except Exception as e:
  351. traceback.print_exc()
  352. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  353. # @router.put("/ws/rollback")
  354. # async def rollback_pattern(
  355. # user_id=Depends(valid_access_token),
  356. # body = Depends(remove_xss_json),
  357. # db: Session = Depends(get_db)
  358. # ):
  359. # try:
  360. # user_list = [i.user_id for i in pattern_id_get_tp_pattern_ws_user_list(body['pattern_id'],db)]
  361. # if str(user_id) not in user_list:
  362. # return JSONResponse(status_code=404, content={
  363. # 'code': 404,
  364. # 'msg': '抱歉,您无权限,请联系系统管理员'
  365. # })
  366. # if pattern_id_get_tp_pattern_ws_count(body['pattern_id'],db)<=1:
  367. # return JSONResponse(status_code=404, content={
  368. # 'code': 404,
  369. # 'msg': '抱歉,已无法回滚'
  370. # })
  371. # data = user_id_and_pattern_id_get_tp_pattern_ws_info(user_id,body['pattern_id'],db)
  372. # data.del_flag='2'
  373. # data.update_by = user_id
  374. # db.commit()
  375. # return {"code": 200, "msg": "回滚成功", "data": None}
  376. # except Exception as e:
  377. # traceback.print_exc()
  378. # raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  379. @router.get("/ws/list")
  380. async def get_pattern_list(
  381. pattern_name: str = Query(None, description='预案名称'),
  382. page: int = Query(1, gt=0, description='页码'),
  383. pageSize: int = Query(5, gt=0, description='每页条目数量'),
  384. user_id=Depends(valid_access_token),
  385. db: Session = Depends(get_db)
  386. ):
  387. try:
  388. query = db.query(TpPatternWSUserList)
  389. query = query.filter(TpPatternWSUserList.del_flag != '2')
  390. query = query.filter(TpPatternWSUserList.user_id==user_id)
  391. if pattern_name:
  392. query = query.filter(TpPatternWSUserList.pattern_name.like(f'%{pattern_name}%'))
  393. total_items = query.count()
  394. # 排序
  395. query = query.order_by(TpPatternWSUserList.create_time.desc())
  396. # 执行分页查询
  397. patterns = query.offset((page - 1) * pageSize).limit(pageSize).all()
  398. return {"code": 200, "msg": "查询成功", "data": [{"id": p.id,"pattern_id":p.pattern_id, "pattern_name": p.pattern_name} for p in patterns],
  399. "total": total_items,
  400. "page": page,
  401. "pageSize": pageSize,
  402. "totalPages": (total_items + pageSize - 1) // pageSize
  403. }
  404. except Exception as e:
  405. traceback.print_exc()
  406. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  407. @router.get("/ws/user_list")
  408. async def get_pattern_list(
  409. pattern_id: str = Query(None, description='预案名称'),
  410. page: int = Query(1, gt=0, description='页码'),
  411. pageSize: int = Query(5, gt=0, description='每页条目数量'),
  412. user_id=Depends(valid_access_token),
  413. db: Session = Depends(get_db)
  414. ):
  415. try:
  416. query = db.query(TpPatternWSUserList)
  417. query = query.filter(TpPatternWSUserList.del_flag != '2')
  418. query = query.filter(TpPatternWSUserList.pattern_id==pattern_id)
  419. total_items = query.count()
  420. # 排序
  421. query = query.order_by(TpPatternWSUserList.create_time.desc())
  422. # 执行分页查询
  423. patterns = query.offset((page - 1) * pageSize).limit(pageSize).all()
  424. data = []
  425. for p in patterns:
  426. user = user_id_get_user_info(db, p.user_id)
  427. dept = dept_id_get_dept_info(db, user.dept_id)
  428. data.append({"id": p.id,"pattern_id":p.pattern_id, "pattern_name": p.pattern_name,"nick_name":user.nick_name,"dept_name":dept.dept_name})
  429. return {"code": 200, "msg": "查询成功", "data": data,
  430. "total": total_items,
  431. "page": page,
  432. "pageSize": pageSize,
  433. "totalPages": (total_items + pageSize - 1) // pageSize
  434. }
  435. except Exception as e:
  436. traceback.print_exc()
  437. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  438. @router.put("/ws/delete_user")
  439. async def rollback_pattern(
  440. user_id=Depends(valid_access_token),
  441. body = Depends(remove_xss_json),
  442. db: Session = Depends(get_db)
  443. ):
  444. try:
  445. pattern_info = pattern_id_get_tp_pattern_ws_group_def_info(body['pattern_id'],db)
  446. if pattern_info is None:
  447. return JSONResponse(status_code=404, content={
  448. 'code': 404,
  449. 'msg': '抱歉,pattern_id不存在,请联系系统管理员'
  450. })
  451. if pattern_info.create_by!=user_id:
  452. return JSONResponse(status_code=404, content={
  453. 'code': 404,
  454. 'msg': '抱歉,您无权限,请联系系统管理员'
  455. })
  456. user = user_id_and_pattern_id_get_tp_pattern_ws_user_info(body['user_id'],body['pattern_id'],db)
  457. user.del_flag='2'
  458. user.update_by = user_id
  459. db.commit()
  460. return {"code": 200, "msg": "关闭协同成功", "data": None}
  461. except Exception as e:
  462. traceback.print_exc()
  463. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  464. @router.post("/ws/add_user")
  465. async def rollback_pattern(
  466. user_id=Depends(valid_access_token),
  467. body = Depends(remove_xss_json),
  468. db: Session = Depends(get_db)
  469. ):
  470. try:
  471. pattern_info = pattern_id_get_tp_pattern_ws_group_def_info(body['pattern_id'],db)
  472. if pattern_info is None:
  473. return JSONResponse(status_code=404, content={
  474. 'code': 404,
  475. 'msg': '抱歉,pattern_id不存在,请联系系统管理员'
  476. })
  477. if pattern_info.create_by!=user_id:
  478. return JSONResponse(status_code=404, content={
  479. 'code': 404,
  480. 'msg': '抱歉,您无权限,请联系系统管理员'
  481. })
  482. user_id_list = body['user_id_list']
  483. if str(user_id) not in user_id_list:
  484. user_id_list.append(str(user_id))
  485. for user in user_id_list:
  486. new_pattern_ws = TpPatternWSUserList(
  487. id=new_guid(),
  488. pattern_id=body['pattern_id'],
  489. pattern_name=body['pattern_name'],
  490. user_id=user,
  491. create_by=user_id
  492. )
  493. db.add(new_pattern_ws)
  494. db.commit()
  495. return {"code": 200, "msg": "开启协同成功", "data": None}
  496. except Exception as e:
  497. traceback.print_exc()
  498. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  499. @router.post("/ws/add_group")
  500. async def add_group_pattern(
  501. user_id=Depends(valid_access_token),
  502. body = Depends(remove_xss_json),
  503. db: Session = Depends(get_db)
  504. ):
  505. try:
  506. pattern_info = pattern_id_get_tp_pattern_ws_group_def_info(body['pattern_id'],db)
  507. if pattern_info is None:
  508. return JSONResponse(status_code=404, content={
  509. 'code': 404,
  510. 'msg': '抱歉,pattern_id不存在,请联系系统管理员'
  511. })
  512. if pattern_info.create_by!=user_id:
  513. return JSONResponse(status_code=404, content={
  514. 'code': 404,
  515. 'msg': '抱歉,您无权限,请联系系统管理员'
  516. })
  517. new_pattern_ws_group = TpPatternWSGroupList(
  518. group_id=new_guid(),
  519. group_name = body['group_name'],
  520. pattern_id=body['pattern_id'],
  521. pattern_name=body['pattern_name'],
  522. create_by=user_id
  523. )
  524. db.add(new_pattern_ws_group)
  525. db.commit()
  526. return {"code": 200, "msg": "新增分组成功", "data": None}
  527. except Exception as e:
  528. traceback.print_exc()
  529. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  530. @router.put("/ws/update_group")
  531. async def add_group_pattern(
  532. user_id=Depends(valid_access_token),
  533. body = Depends(remove_xss_json),
  534. db: Session = Depends(get_db)
  535. ):
  536. try:
  537. pattern_info = pattern_id_get_tp_pattern_ws_group_def_info(body['pattern_id'],db)
  538. if pattern_info is None:
  539. return JSONResponse(status_code=404, content={
  540. 'code': 404,
  541. 'msg': '抱歉,pattern_id不存在,请联系系统管理员'
  542. })
  543. if pattern_info.create_by!=user_id:
  544. return JSONResponse(status_code=404, content={
  545. 'code': 404,
  546. 'msg': '抱歉,您无权限,请联系系统管理员'
  547. })
  548. query = db.query(TpPatternWSGroupList)
  549. query = query.filter(TpPatternWSGroupList.del_flag!='2')
  550. query = query.filter(TpPatternWSGroupList.group_id==body['gruop_id'])
  551. group = query.first()
  552. if 'group_name' in body:
  553. group.group_name = body['group_name']
  554. group.update_by=user_id
  555. db.commit()
  556. return {"code": 200, "msg": "更新分组成功", "data": None}
  557. except Exception as e:
  558. traceback.print_exc()
  559. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  560. @router.get("/ws/bz_list")
  561. async def get_pattern_list(
  562. pattern_id: str = Query(None, description='预案名称'),
  563. # page: int = Query(1, gt=0, description='页码'),
  564. # pageSize: int = Query(5, gt=0, description='每页条目数量'),
  565. user_id=Depends(valid_access_token),
  566. db: Session = Depends(get_db)
  567. ):
  568. try:
  569. # data = pattern_id_get_tp_pattern_ws_list(pattern_id,db)
  570. group_list = pattern_id_get_tp_pattern_ws_group_list(pattern_id,db)
  571. message = []
  572. for group in group_list:
  573. query = db.query(TpPatternWSList.create_by)
  574. query = query.filter(TpPatternWSList.del_flag != '2')
  575. query = query.filter(TpPatternWSList.pattern_id == group.group_id)
  576. # total_items = query.count()
  577. # 排序
  578. query = query.order_by(TpPatternWSUserList.create_time.desc())
  579. # 执行分页查询
  580. patterns = query.all() # .offset((page - 1) * pageSize).limit(pageSize)
  581. for info in patterns:
  582. user= user_id_get_user_info(db,info.create_by)
  583. dept = dept_id_get_dept_info(db,user.dept_id)
  584. message.append({"id":info.id,
  585. "name":info.name,
  586. "group_id":info.pattern_id,
  587. "content":info.content,
  588. "visible":info.visible,
  589. "user_id":info.create_by,
  590. "nick_name":user.nick_name,
  591. "dept_name":dept.dept_name,
  592. "create_time":info.create_time.strftime('%Y-%m-%d %H:%M:%S')})
  593. return {"code": 200, "msg": "查询成功", "data": message,
  594. # "total": total_items,
  595. # "page": page,
  596. # "pageSize": pageSize,
  597. # "totalPages": (total_items + pageSize - 1) // pageSize
  598. }
  599. except Exception as e:
  600. traceback.print_exc()
  601. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  602. @router.put('/ws/bz_visible')
  603. async def update_bz_visible(
  604. body = Depends(remove_xss_json),
  605. user_id=Depends(valid_access_token),
  606. db: Session = Depends(get_db)
  607. ):
  608. try:
  609. zb_id_list = body['zb_id_list']
  610. for zb in zb_id_list:
  611. query = db.query(TpPatternWSList)
  612. query = query.filter(TpPatternWSList.del_flag!='2')
  613. query = query.filter(TpPatternWSList.id==zb)
  614. info = query.first()
  615. info.visible=body['visible']
  616. db.commit()
  617. except Exception as e:
  618. traceback.print_exc()
  619. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")