python-fastapi-mm-zhcs-yj-api/routers/api/pattern/template.py

#!/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)}")