#!/usr/bin/env python3 # -*- coding: utf-8 -*- from fastapi import APIRouter, Request, Depends, Query, HTTPException, status,WebSocket,WebSocketDisconnect from common.security import valid_access_token from fastapi.responses import JSONResponse from sqlalchemy.orm import Session from sqlalchemy.sql import func from common.auth_user import * from sqlalchemy import text from pydantic import BaseModel from common.BigDataCenterAPI import * from database import get_db from typing import List from models import * from utils import * from utils.spatial import * import json import traceback router = APIRouter() @router.post("/create") async def create_pattern( user_id=Depends(valid_access_token), body = Depends(remove_xss_json), db: Session = Depends(get_db) ): try: new_pattern = TpPatternList( id = new_guid(), pattern_name=body['pattern_name'], content=body['content'], create_dept = user_id ) db.add(new_pattern) db.commit() return {"code": 200, "msg": "创建成功", "data": None} except Exception as e: traceback.print_exc() raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") @router.put("/update/{pattern_id}") async def update_pattern( pattern_id :int , user_id=Depends(valid_access_token), body=Depends(remove_xss_json), db: Session = Depends(get_db) ): try: query = db.query(TpPatternList) query = query.filter(TpPatternList.id == pattern_id) query = query.filter(TpPatternList.del_flag != '2') update_pattern = query.first() if not update_pattern: raise HTTPException(status_code=404, detail="图案不存在") update_pattern.pattern_name = body['pattern_name'] update_pattern.content = body['content'] update_pattern.create_dept = user_id db.commit() return {"code": 200, "msg": "更新成功"} except Exception as e: traceback.print_exc() raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") @router.get("/info/{pattern_id}") async def get_pattern_info( pattern_id: int, db: Session = Depends(get_db) ): try: query = db.query(TpPatternList) query = query.filter(TpPatternList.id == pattern_id) query = query.filter(TpPatternList.del_flag != '2') pattern = query.first() # pattern = db.query(TpPatternList).filter(TpPatternList.id == pattern_id).first() if not pattern: raise HTTPException(status_code=404, detail="图案不存在") return {"code": 200, "msg": "获取成功", "data": {"pattern_name": pattern.pattern_name, "content": pattern.content}} except Exception as e: traceback.print_exc() raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") @router.get("/list") async def get_pattern_list( pattern_name: str = Query(None, description='预案名称'), page: int = Query(1, gt=0, description='页码'), pageSize: int = Query(5, gt=0, description='每页条目数量'), db: Session = Depends(get_db) ): try: query = db.query(TpPatternList) query = query.filter(TpPatternList.del_flag != '2') if pattern_name: query = query.filter(TpPatternList.pattern_name.like(f'%{pattern_name}%')) total_items = query.count() # 排序 query = query.order_by(TpPatternList.create_time.desc()) # 执行分页查询 patterns = query.offset((page - 1) * pageSize).limit(pageSize).all() return {"code": 200, "msg": "查询成功", "data": [{"id": p.id, "pattern_name": p.pattern_name, "content": p.content} for p in patterns], "total": total_items, "page": page, "pageSize": pageSize, "totalPages": (total_items + pageSize - 1) // pageSize } except Exception as e: traceback.print_exc() raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") @router.delete("/delete/{pattern_id}") async def delete_pattern( pattern_id: int, db: Session = Depends(get_db) ): try: # 检查图案是否存在 query = db.query(TpPatternList) query = query.filter(TpPatternList.id == pattern_id) query = query.filter(TpPatternList.del_flag != '2') pattern = query.first() if not pattern: raise HTTPException(status_code=404, detail="图案不存在") # 执行删除操作 pattern.del_flag='2' db.commit() return {"code": 200, "msg": "删除成功"} except Exception as e: traceback.print_exc() raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") def pattern_id_get_tp_pattern_ws_info(pattern_id:str,db: Session): query = db.query(TpPatternWSList) query = query.filter(TpPatternWSList.del_flag != '2') query = query.filter(TpPatternWSList.pattern_id == pattern_id) query.order_by(TpPatternWSList.create_time.desc()) return query.first() def pattern_id_get_tp_pattern_ws_user_list(pattern_id:str,db: Session): query = db.query(TpPatternWSUserList) query = query.filter(TpPatternWSUserList.del_flag != '2') query = query.filter(TpPatternWSUserList.pattern_id == pattern_id) query.order_by(TpPatternWSUserList.create_time.desc()) return query.all() def user_id_get_tp_pattern_ws_user_list(user_id:str,db: Session): query = db.query(TpPatternWSUserList) query = query.filter(TpPatternWSUserList.del_flag != '2') query = query.filter(TpPatternWSUserList.user_id == user_id) query.order_by(TpPatternWSUserList.create_time.desc()) return query.all() def user_id_and_pattern_id_get_tp_pattern_ws_user_info(user_id:str,pattern_id:str,db: Session): query = db.query(TpPatternWSUserList) query = query.filter(TpPatternWSUserList.del_flag != '2') query = query.filter(TpPatternWSUserList.user_id == user_id) query = query.filter(TpPatternWSUserList.pattern_id == pattern_id) query.order_by(TpPatternWSUserList.create_time.desc()) return query.all() class ConnectionManager: def __init__(self): self.active_connections = {} #: List[WebSocket] async def connect(self, websocket: WebSocket,pattern_id:str,db: Session): await websocket.accept() if pattern_id not in self.active_connections: data = pattern_id_get_tp_pattern_ws_info(pattern_id,db) if data: await websocket.send_text(data.content) self.active_connections[pattern_id] = [websocket] else: self.active_connections[pattern_id].append(websocket) def disconnect(self, websocket: WebSocket,pattern_id:str): self.active_connections[pattern_id].remove(websocket) if not self.active_connections[pattern_id]: del self.active_connections[pattern_id] async def broadcast(self, message: str,pattern_id:str,user_id,db: Session): new_pattern = TpPatternWSList( id = new_guid(), pattern_id=pattern_id, content=pattern_id, create_dept = user_id ) db.add(new_pattern) db.commit() for connection in self.active_connections[pattern_id]: await connection.send_text(message) manager = ConnectionManager() @router.websocket("/{pattern_id}/ws") async def websocket_endpoint(pattern_id:str ,websocket: WebSocket,user_id=Depends(valid_access_token),db: Session = Depends(get_db)): user_list = [i.user_id for i in pattern_id_get_tp_pattern_ws_user_list(pattern_id, db)] if user_id not in user_list: return JSONResponse(status_code=404, content={ 'code': 404, 'msg': '抱歉，您无权限，请联系系统管理员' }) await manager.connect(websocket,pattern_id,db) try: while True: data = await websocket.receive_text() await manager.broadcast(data,pattern_id,user_id,db) # 广播消息给所有连接 except WebSocketDisconnect: manager.disconnect(websocket,pattern_id) @router.post("/ws/create") async def create_pattern( user_id=Depends(valid_access_token), body = Depends(remove_xss_json), db: Session = Depends(get_db) ): try: user_id_list = body['user_id_list'] if user_id not in user_id_list: user_id_list.append(user_id) for user in user_id_list: new_pattern_ws = TpPatternWSUserList( id=new_guid(), pattern_id=body['pattern_id'], user_id=user, create_dept=user_id ) db.add(new_pattern_ws) new_pattern_ws = TpPatternWSList( id = new_guid(), pattern_id=body['pattern_id'], content=body['content'], create_dept = user_id ) db.add(new_pattern_ws) db.commit() return {"code": 200, "msg": "创建成功", "data": None} except Exception as e: traceback.print_exc() raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") @router.put("/ws/rollback") async def rollback_pattern( user_id=Depends(valid_access_token), body = Depends(remove_xss_json), db: Session = Depends(get_db) ): try: user_list = [i.user_id for i in pattern_id_get_tp_pattern_ws_user_list(body['pattern_id'],db)] if user_id not in user_list: return JSONResponse(status_code=404, content={ 'code': 404, 'msg': '抱歉，您无权限，请联系系统管理员' }) data = pattern_id_get_tp_pattern_ws_info(body['pattern_id'],db) data.del_flag='2' data.update_by = user_id db.commit() return {"code": 200, "msg": "回滚成功", "data": None} except Exception as e: traceback.print_exc() raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") @router.get("/ws/list") async def get_pattern_list( pattern_name: str = Query(None, description='预案名称'), page: int = Query(1, gt=0, description='页码'), pageSize: int = Query(5, gt=0, description='每页条目数量'), user_id=Depends(valid_access_token), db: Session = Depends(get_db) ): try: query = db.query(TpPatternWSUserList) query = query.filter(TpPatternWSUserList.del_flag != '2') query = query.filter(TpPatternWSUserList.user_id==user_id) if pattern_name: query = query.filter(TpPatternWSUserList.pattern_name.like(f'%{pattern_name}%')) total_items = query.count() # 排序 query = query.order_by(TpPatternWSUserList.create_time.desc()) # 执行分页查询 patterns = query.offset((page - 1) * pageSize).limit(pageSize).all() return {"code": 200, "msg": "查询成功", "data": [{"id": p.id,"pattern_id":p.pattern_id, "pattern_name": p.pattern_name} for p in patterns], "total": total_items, "page": page, "pageSize": pageSize, "totalPages": (total_items + pageSize - 1) // pageSize } except Exception as e: traceback.print_exc() raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") @router.put("/ws/delete_user") async def rollback_pattern( user_id=Depends(valid_access_token), body = Depends(remove_xss_json), db: Session = Depends(get_db) ): try: pattern_info = pattern_id_get_tp_pattern_ws_info(body['pattern_id'],db) if pattern_info.create_by!=user_id: return JSONResponse(status_code=404, content={ 'code': 404, 'msg': '抱歉，您无权限，请联系系统管理员' }) user = user_id_and_pattern_id_get_tp_pattern_ws_user_info(body['user_id'],body['pattern_id'],db) user.del_flag='2' user.update_by = user_id db.commit() return {"code": 200, "msg": "关闭协同成功", "data": None} except Exception as e: traceback.print_exc() raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") @router.put("/ws/delete_user") async def rollback_pattern( user_id=Depends(valid_access_token), body = Depends(remove_xss_json), db: Session = Depends(get_db) ): try: pattern_info = pattern_id_get_tp_pattern_ws_info(body['pattern_id'],db) if pattern_info.create_by!=user_id: return JSONResponse(status_code=404, content={ 'code': 404, 'msg': '抱歉，您无权限，请联系系统管理员' }) new_pattern_ws = TpPatternWSUserList( id=new_guid(), pattern_id=body['pattern_id'], user_id=body['user_id'], create_dept=user_id ) db.add(new_pattern_ws) db.commit() return {"code": 200, "msg": "开启协同成功", "data": None} except Exception as e: traceback.print_exc() raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")