python-fastapi-mm-zhcs-yj-api/routers/api/riskManagement/user.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from fastapi import APIRouter, Request, Depends,Query, HTTPException, status
from common.security import valid_access_token
from fastapi.responses import JSONResponse
from sqlalchemy.orm import Session
from sqlalchemy import and_, or_
from pydantic import BaseModel
from datetime import datetime
from database import get_db
from typing import List
from models import *
from utils import *
from utils.ry_system_util import *
from utils.riskManagement_uitl import *
import json
import traceback
import xlrd
import os
from common.enc import mpfun

router = APIRouter()



# 目录在文档上传接口写死
UPLOAD_mergefile_PATH = '/data/upload/mergefile'

@router.get('/list')
async def get_inspection_user_list(
    nickName: str = Query(None, description='姓名'),
    deptId :str = Query(None, description='部门id'),
    page: int = Query(1, gt=0, description='页码'),
    pageSize: int = Query(5, gt=0, description='每页条目数量'),
    db: Session = Depends(get_db),
    user_id = Depends(valid_access_token)
):
    try:

        # 应用查询条件
        query = db.query(SysUser)
        query = query.filter(SysUser.del_flag != '2')

        if nickName:
            query = query.filter(SysUser.nick_name.like(f'%{nickName}%'))

        if deptId:
            user_ids = [i.user_id for i in dept_id_get_user_info(db,deptId)]
            query = query.filter(SysUser.user_id.in_(user_ids))
        user_ids = [i.user_id for i in query.all()]
        # 构建查询
        query = db.query(RiskManagementInspectionUser)
        query = query.filter(RiskManagementInspectionUser.del_flag != '2')
        query = query.filter(RiskManagementInspectionUser.user_id.in_(user_ids))

        # 计算总条目数
        total_items = query.count()

        # 排序

        query = query.order_by(RiskManagementInspectionUser.create_time.desc())
        # 执行分页查询
        InspectionUsers = query.offset((page - 1) * pageSize).limit(pageSize).all()

        # 将查询结果转换为列表形式的字典
        InspectionUsers_list = []
        for user in InspectionUsers:
            user_info = user_id_get_user_info(db,user.user_id)
            dept = dept_id_get_dept_info(db,user_info.dept_id)
            # print(dept)
            ancestors_names = dept_id_get_ancestors_names(db,dept)
            area_code = user.area_code
            area = area_code_get_area_info(db,area_code)
            area = area_code_get_ancestors_names(db,area)
            user_info = {
                "id": user.id,
                "user_id": user.user_id,
                "dept_id": user_info.dept_id,
                "dept_name": dept.dept_name,
                "ancestors_names": ancestors_names,
                "user_name": user_info.user_name,
                "create_time": user.create_time.strftime('%Y-%m-%d'),
                "phonenumber": mpfun.dec_data(user_info.phonenumber),
                "nick_name": user_info.nick_name,
                "area_code": user.area_code,
                "area": area,
                "yzy_account":user_info.yzy_account
                }
            InspectionUsers_list.append(user_info)
        # 返回结果
        return {
            "code": 200,
            "msg": "成功",
            "data": InspectionUsers_list,
            "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=str(e))


@router.get('/{id}')
async def get_inspection_user(
    id: str ,
    db: Session = Depends(get_db),
    user_id = Depends(valid_access_token)
):
    try:
        # 构建查询
        query = db.query(RiskManagementInspectionUser)
        query = query.filter(RiskManagementInspectionUser.del_flag != '2')
        # 应用查询条件
        if id:
            query = query.filter(RiskManagementInspectionUser.id == id)

        # 执行查询
        user = query.first()
        if not user:
            detail = "用户不存在"
            raise HTTPException(status_code=404, detail="用户不存在")
        # 将查询结果转换为列表形式的字典
        user_info = user_id_get_user_info(db, user.user_id)
        dept = dept_id_get_dept_info(db, user_info.dept_id)
        # print(dept)
        ancestors_names = dept_id_get_ancestors_names(db, dept)
        area_code = user.area_code
        area = area_code_get_area_info(db, area_code)
        area = area_code_get_ancestors_names(db, area)
        inspection_user_result = {
            "id": user.id,
            "user_id": user.user_id,
            "dept_id": user_info.dept_id,
            "dept_name": dept.dept_name,
            "ancestors_names": ancestors_names,
            "user_name": user_info.user_name,
            "create_time": user.create_time.strftime('%Y-%m-%d'),
            "phonenumber": user_info.phonenumber,
            "nick_name": user_info.nick_name,
            "area_code": user.area_code,
            "area": area,
            "yzy_account": user_info.yzy_account
        }
        # inspection_user_result = {
        #         "id": user.id,
        #         "user_id": user.user_id,
        #         "dept_id": user.dept_id,
        #         "dept_name": user.dept_name,
        #         "ancestors_names": user.ancestors_names,
        #         "user_name": user.user_name,
        #         "create_time": user.create_time.strftime('%Y-%m-%d'),
        #         "phonenumber": user.phonenumber,
        #         "nick_name": user.nick_name,
        #         "area_code": user.area_code,
        #         "area": user.area,
        #         "yzy_account":user.yzy_account
        #     }

        # 返回结果
        return {
            "code": 200,
            "msg": "成功",
            "data": inspection_user_result
        }
    except Exception as e:
        # 处理异常
        traceback.print_exc()
        if str(e)=='':
            e = detail
        raise HTTPException(status_code=500, detail=str(e))




@router.post('/create')
async def create_inspection_user(
    db: Session = Depends(get_db),
    body = Depends(remove_xss_json),
    user_id = Depends(valid_access_token)
):
    try:
        # user_id_1 = body['user_id']
        # area_code = body['area_code']
        # dept_id = None
        # ancestors_names=None
        # if 'dept_id' in body:
        #     dept_id=body['dept_id']
        #     dept = dept_id_get_dept_info(db,dept_id)
        #     # print(dept)
        #     ancestors_names = dept_id_get_ancestors_names(db,dept)
        # dept_name = None
        # if 'dept_name' in body:
        #     dept_name=body['dept_name']
        # phonenumber = None
        # if 'phonenumber' in body:
        #     phonenumber=body['phonenumber']
        # 创建新的预案记录
        new_user = RiskManagementInspectionUser(
            user_id=body['user_id'],
            # dept_id = dept_id,#body['dept_id'],
            # dept_name = dept_name,#body['dept_name'],
            # ancestors_names = ancestors_names,#body['ancestors_names'],
            # user_name = '',#body['user_name'],
            # nick_name = '',#body['nick_name'],
            # phonenumber = phonenumber,#body['phonenumber'],
            area_code = body['area_code'],
            # area = '',#body['area'],
            # yzy_account = '',#body['yzy_account'],
            create_by = user_id
        )
        # 添加到数据库会话并提交
        db.add(new_user)
        db.commit()
        db.refresh(new_user)  # 可选，如果需要刷新实例状态
        # 返回创建成功的响应
        return {
            "code": 200,
            "msg": "成功",
            "data": None
        }
    except Exception as e:
        # 处理异常
        traceback.print_exc()
        raise HTTPException(status_code=500, detail=str(e))


@router.put('/update')
async def update_inspection_user(
    db: Session = Depends(get_db),
    body = Depends(remove_xss_json),
    user_id = Depends(valid_access_token)
):
    try:
        # 提取请求数据
        query = db.query(RiskManagementInspectionUser)
        query = query.filter(RiskManagementInspectionUser.id == body['id'])
        query = query.filter(RiskManagementInspectionUser.del_flag != '2')
        user = query.first()

        if not user:
            detail = "用户不存在"
            raise HTTPException(status_code=404, detail="用户不存在")

        if 'user_id' in body:
            user.user_id = body['user_id']
        # if 'dept_id' in body:
        #     user.dept_id = body['dept_id']
        # if 'dept_name' in body:
        #     user.dept_name = body['dept_name']
        # if 'ancestors_names' in body:
        #     user.ancestors_names = body['ancestors_names']
        # if 'user_name' in body:
        #     user.user_name = body['user_name']
        # if 'nick_name' in body:
        #     user.nick_name = body['nick_name']
        # if 'phonenumber' in body:
        #     user.phonenumber = body['phonenumber']
        if 'area_code' in body:
            user.area_code = body['area_code']
        # if 'area' in body:
        #     user.area = body['area']

        if user_id:
            user.update_by = user_id

        # 更新到数据库会话并提交
        db.commit()
        db.refresh(user)  # 可选，如果需要刷新实例状态

        # 返回创建成功的响应
        return {
            "code": 200,
            "msg": "成功",
            "data": None
        }
    except Exception as e:
        # 处理异常
        traceback.print_exc()
        if str(e)=='':
            e = detail
        raise HTTPException(status_code=500, detail=str(e))

@router.delete('/delete')
async def delete_inspection_users(
        userIds: list,
        db: Session = Depends(get_db),
    body = Depends(remove_xss_json),
    user_id = Depends(valid_access_token)
):
    try:
        # 提取请求数据
        query = db.query(RiskManagementInspectionUser)
        query = query.filter(RiskManagementInspectionUser.del_flag != '2')
        query = query.filter(RiskManagementInspectionUser.id.in_(userIds))
        users = query.all()
        print(userIds)
        if not users:
            detail = "用户不存在"
            raise HTTPException(status_code=404, detail="用户不存在")
        for user in users:
            user.del_flag = '2'
            user.update_by=user_id
        # 更新到数据库会话并提交
        db.commit()

        # 返回创建成功的响应
        return {
            "code": 200,
            "msg": "删除成功",
            "data": None
        }
    except Exception as e:
        # 处理异常
        traceback.print_exc()
        if str(e) == '':
            e = detail
        raise HTTPException(status_code=500, detail=str(e))

@router.delete('/delete/{userId}')
async def delete_inspection_user(
        userId: str,
        db: Session = Depends(get_db),
    body = Depends(remove_xss_json),
    user_id = Depends(valid_access_token)
):
    try:
        # 提取请求数据
        query = db.query(RiskManagementInspectionUser)
        query = query.filter(RiskManagementInspectionUser.del_flag != '2')
        query = query.filter(RiskManagementInspectionUser.id==userId)
        user = query.first()
        if not user:
            detail = "用户不存在"
            raise HTTPException(status_code=404, detail="用户不存在")

        user.del_flag = '2'
        user.update_by = user_id
        # 更新到数据库会话并提交
        db.commit()
        db.refresh(user)  # 可选，如果需要刷新实例状态

        # 返回创建成功的响应
        return {
            "code": 200,
            "msg": "删除成功",
            "data": None
        }
    except Exception as e:
        # 处理异常
        traceback.print_exc()
        if str(e) == '':
            e = detail
        raise HTTPException(status_code=500, detail=str(e))
    

# 导入
@router.post('/import')
async def import_doc(
    request: Request, 
    db: Session = Depends(get_db), 
    body = Depends(remove_xss_json), 
    user_id = Depends(valid_access_token)
):
    
    try:
        filename = body['filename']
        if len(filename) == 0:
            raise Exception()
        
        file = filename[0]
        url = file['url']
        file_path = f"{UPLOAD_mergefile_PATH}/uploads/{url}"
        file_path = os.path.abspath(file_path)
        print(file_path)
        
        book = xlrd.open_workbook(file_path)
        sheet = book.sheet_by_index(0)

        data = []

        return {
            'code': 200,
            'msg': '导入成功'
        }
    except Exception:
        traceback.print_exc()
        return {
            'code': 500,
            'msg': '导入发生异常'
        }