#!/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,valid_websocket_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 * from utils.ry_system_util import * from common.websocketManager import * import json import xlrd,os import traceback router = APIRouter() def template_id_get_template_info(db,id): query = db.query(TpPatternTemplate) query = query.filter_by(template_id = id,del_flag = '0') return query.first() @router.post('/createImport') async def create_contact( request: Request, db: Session = Depends(get_db), body=Depends(remove_xss_json), auth_user: AuthUser = Depends(find_auth_user), user_id=Depends(valid_access_token) ): try: # 提取请求数据 filename = body['filename'] # file_name_desc = body['file_name_desc'] if len(filename) == 0: raise Exception() file_path = f'/data/upload/mergefile/uploads/{filename}' # 检查文件是否存在 if not os.path.isfile(file_path): return JSONResponse(status_code=404, content={ 'errcode': 404, 'errmsg': f'{filename}不存在' }) msg = '成功' code =200 try: book = xlrd.open_workbook(file_path) sheet = book.sheet_by_index(0) except Exception as e: traceback.print_exc() msg = f'\n文件打开失败，请核实文件格式为xlsx/xlx>{str(e)}' code = 500 return JSONResponse(status_code=code, content={ "code": code, "msg": msg, "data": None }) data = [] import_status = True for row in range(1, sheet.nrows): name = sheet.cell(row, 0).value if name == '' or name==0: import_status = False msg = f'\n行<{row + 1}>模型名称不能为空<{name}>' code = 500 value = sheet.cell(row, 1).value if value == '': import_status = False msg = f'\n行<{row + 1}>模型值不能为空<{value}>' code = 500 Template = TpPatternTemplate( template_id=new_guid(), name=name, value=value, order_num=0, visible='1', create_by = user_id ) data.append(Template) if import_status: db.add_all(data) db.commit() except Exception as e: traceback.print_exc() # 处理异常 db.rollback() code = 500 msg = str(e) return JSONResponse(status_code=code, content={ "code": code, "msg": msg, "data": None }) @router.post("/create") async def create_template( user_id=Depends(valid_access_token), body = Depends(remove_xss_json), db: Session = Depends(get_db) ): try: new_template = TpPatternTemplate( template_id=new_guid(), name=body['name'], value=body['value'], order_num=body['order_num'], visible=body['visible'], create_by = user_id ) db.add(new_template) 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/{id}") async def update_pattern( id :str , user_id=Depends(valid_access_token), body=Depends(remove_xss_json), db: Session = Depends(get_db) ): try: update_template = template_id_get_template_info(db,id) if not update_template: return JSONResponse(status_code=404,content={"code":404,"msg":"template not found"}) update_template.name = body['name'] update_template.value = body['value'] update_template.order_num=body['order_num'] update_template.visible=body['visible'] update_template.update_by = 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/{id}") async def get_pattern_info( id: str, db: Session = Depends(get_db) ): try: info = template_id_get_template_info(db,id) if not info: return JSONResponse(status_code=404,content={"code":404,"msg":"template not found"}) data = {"template_id": info.template_id, "name": info.name, "value": info.value, "order_num": info.order_num, "visible": info.visible, "create_time": info.create_time} return {"code": 200, "msg": "获取成功", "data": data} 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( name: str = Query(None, description='名称'), page: int = Query(1, gt=0, description='页码'), pageSize: int = Query(None, gt=0, description='每页条目数量'), db: Session = Depends(get_db) ): try: query = db.query(TpPatternTemplate) query = query.filter_by(del_flag='0') if name: query = query.filter(TpPatternTemplate.name.like(f'%{name}%')) total_items = query.count() # 排序 if pageSize is None: pageSize=total_items query = query.order_by(TpPatternTemplate.order_num.asc()) # 执行分页查询 lists = query.offset((page - 1) * pageSize).limit(pageSize).all() data = [{"template_id": info.template_id, "name": info.name, "value": info.value, "order_num": info.order_num, "visible": info.visible, "create_time": info.create_time} for info in lists] return {"code": 200, "msg": "查询成功", "data": data, "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('/changeVisible') async def change_classification_visible( db: Session = Depends(get_db), body=Depends(remove_xss_json), user_id=Depends(valid_access_token) ): try: template_id = body['template_id'] visible = body['visible'] info = template_id_get_template_info(db, template_id) if not info: return JSONResponse(status_code=404, content={"code": 404, "msg": "classification not found"}) info.visible= visible info.update_by=user_id db.commit() return { "code": 200, "msg": "操作成功" } except Exception as e: # 处理异常 traceback.print_exc() raise HTTPException(status_code=500, detail=str(e)) @router.delete("/delete/{id}") async def delete_pattern( id: str, db: Session = Depends(get_db) ): try: # 检查图案是否存在 info = template_id_get_template_info(db, id) if not info: return JSONResponse(status_code=404, content={"code": 404, "msg": "template not found"}) info.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)}")