python-fastapi-mm-zhcs-yj-api/routers/api/taskRegistration/__init__.py

import json
import random
from sqlalchemy import create_engine, select
from utils.StripTagsHTMLParser import *
from sqlalchemy.orm import Session
from sqlalchemy import text, exists, and_, or_, not_
import traceback
from models import *
from common.db import db_task
from common.security import valid_access_token
from datetime import datetime, timedelta
from fastapi import APIRouter, Depends, HTTPException, Response, Query
from database import get_db
from pydantic import BaseModel
from utils import *
import copy
from common.db import db_event_management, db_user

router = APIRouter()


@router.post('/create')
async def create_task(
        request: Request,
        db: Session = Depends(get_db),
        body=Depends(remove_xss_json),
        user_id=Depends(valid_access_token)):
    try:
        # 验证必需的字段
        required_fields = ['task_description', 'unit_name', 'registrar','event_code']
        missing_fields = [field for field in required_fields if field not in body]
        if missing_fields:
            raise HTTPException(status_code=401, detail=f"Missing required fields: {', '.join(missing_fields)}")

        event_code = db.query(EventBase).filter(EventBase.event_code == body.get('event_code')).first()
        if not event_code:
            return Response(content="事件不存在", status_code=400)
        task_id = db_task.get_next_event_id(db)

        task_base = TaskRegistration(
            **body,
            task_id=task_id
        )
        db.add(task_base)
        db.commit()

        return {
            "code": 200,
            "msg": "任务创建成功",
            "data": task_id
        }

    except Exception as e:
        db.rollback()
        traceback.print_exc()
        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")


class TaskQuery(BaseModel):
    task_id: str = None
    task_description: str = None
    unit_name: str = None
    registrar: str = None
    creation_time: str = None
    processing_status: str = None



@router.post('/select')
@router.get('/select')
async def select_tasks(
        request: Request,
        db: Session = Depends(get_db),
        query: TaskQuery = Depends(),
        sortBy: str = Query(None, description="排序字段"),
        sortOrder: str = Query(None, description="排序顺序"),
        user_id=Depends(valid_access_token),
        pageNum: int = Query(1, gt=0, description="页码"),
        event_code: str = Query(None, description="事件ID"),
        pageSize: int = Query(10, gt=0, le=100, description="每页大小")):
    try:
        missing_event_code = db.query(EventBase).filter(EventBase.event_code == event_code).first()
        # print(missing_event_code)
        if not missing_event_code:
            return Response(content="事件不存在", status_code=400)


        data_query = db.query(TaskRegistration).filter(TaskRegistration.del_flag != '2')
        data_query = data_query.filter(TaskRegistration.event_code == event_code)

        # 应用过滤条件
        if query.task_id:
            data_query = data_query.filter(TaskRegistration.task_id == query.task_id)
        if query.task_description:
            data_query = data_query.filter(TaskRegistration.task_description == query.task_description)
        if query.unit_name:
            data_query = data_query.filter(TaskRegistration.unit_name == query.unit_name)
        if query.registrar:
            data_query = data_query.filter(TaskRegistration.registrar == query.registrar)
        if query.creation_time:
            data_query = data_query.filter(TaskRegistration.creation_time == query.creation_time)
        if query.processing_status:
            data_query = data_query.filter(TaskRegistration.processing_status == query.processing_status)

        # print(TaskRegistration,sortBy)
        if sortBy:
            if hasattr(TaskRegistration, sortBy):
                print("xx")
                sort_attr = getattr(TaskRegistration, sortBy)
                data_query = data_query.order_by(sort_attr.asc() if sortOrder == 'asc' else sort_attr.desc())



        total_count = data_query.count()
        offset = (pageNum - 1) * pageSize
        data_query = data_query.offset(offset).limit(pageSize)

        tasks = data_query.all()

        return {
            "code": 200,
            "msg": "任务查询成功",
            "data": tasks,
            "total": total_count,
            "pages": (total_count + pageSize - 1) // pageSize,
            "current_page": pageNum,
            "page_size": pageSize
        }

    except Exception as e:
        db.rollback()
        traceback.print_exc()
        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")


@router.put('/update')
async def update_task_status(
        request: Request,
        db: Session = Depends(get_db),
        user_id=Depends(valid_access_token)):
    body = await request.json()
    task_id_to_use = body.get('task_id')
    processing_status = body.get('processing_status')

    event_code = db.query(EventBase).filter(EventBase.event_code == body.get('event_code')).first()
    if not event_code:
        return Response(content="事件不存在", status_code=400)


    if not task_id_to_use:
        return Response(content="Missing required parameter 'task_id'", status_code=400)
    if processing_status not in ['已完成', '处理中', '待处理','未完成']:
        return Response(content="processing_status must be '已完成' or '处理中' or '待处理' or '未完成'", status_code=400)

    task_entry = (db.query(TaskRegistration)
                  .filter(TaskRegistration.del_flag != '2')
                  .filter(TaskRegistration.task_id == task_id_to_use))
    task_entry = task_entry.first()
    if not task_entry:
        raise HTTPException(status_code=404, detail="任务不存在")

    task_entry.processing_status = processing_status
    task_entry.update_time = datetime.now()

    try:
        db.commit()
        return {
            "code": 200,
            "msg": "任务状态更新成功",
            "data": {
                "task_id": task_id_to_use,
                "processing_status": task_entry.processing_status,
                "update_time": task_entry.update_time.isoformat()
            }
        }
    except Exception as e:
        db.rollback()
        raise HTTPException(status_code=500, detail=str(e))

@router.delete('/delete')
async def delete_task(
        request: Request,
        db: Session = Depends(get_db),
        user_id=Depends(valid_access_token)):
    try:
        body = await request.json()
        task_id_to_use = body.get('taskID')
        if not task_id_to_use:
            raise HTTPException(status_code=400, detail="Missing required parameter 'taskID'")

        event_code = db.query(EventBase).filter(EventBase.event_code == body.get('event_code')).first()
        if not event_code:
            return Response(content="事件不存在", status_code=400)

        task_entry = db.query(TaskRegistration).filter(TaskRegistration.task_id == task_id_to_use).first()
        if not task_entry:
            raise HTTPException(status_code=404, detail="任务不存在")

        task_entry.del_flag = '2'

        try:
            db.commit()
            return {
                "code": 200,
                "msg": "任务删除成功",
                "data": {
                        "task_id": task_entry.task_id
                    }
                }
        except Exception as e:
            db.rollback()
            raise HTTPException(status_code=500, detail=f"An error occurred while deleting the task: {str(e)}")

    except HTTPException as e:
        raise e
    except Exception as e:
        db.rollback()
        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")

@router.get('/selectUnit')
async def select_unit(
        request: Request,
        db: Session = Depends(get_db),
        query: TaskQuery = Depends(),
        user_id=Depends(valid_access_token),
        ):
    try:
        data_query = db.query(TaskUnit).filter(TaskUnit.id != '2').all()
        return {
            "code": 200,
            "msg": "查询成功",
            "data": data_query
        }
    except HTTPException as e:
        raise e
    except Exception as e:
        db.rollback()
        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
    

#
# 小屏专用，不敢搞大屏相关代码
#
@router.get('/list')
async def select_tasks(
        request: Request,
        task_type: str = Query(None),
        processing_status: str = Query(None),
        task_description: str = Query(None),
        page: int = Query(1, gt=0, description="页码"),
        page_size: int = Query(10, gt=0, le=100, description="每页大小"),
        db: Session = Depends(get_db)):
    try:
        data_query = db.query(TaskRegistration).filter(TaskRegistration.del_flag != '2').filter(TaskRegistration.event_code is not None)
        # 应用过滤条件
        if task_type is not None:
            data_query = data_query.filter(TaskRegistration.task_type == task_type)
        if task_description is not None:
            data_query = data_query.filter(TaskRegistration.task_description == task_description)
        if processing_status is not None:
            data_query = data_query.filter(TaskRegistration.processing_status == processing_status)

        total_count = data_query.count()
        offset = (page - 1) * page_size
        data_query = data_query.order_by(TaskRegistration.creation_time.desc()).offset(offset).limit(page_size)

        data = []
        tasks = data_query.all()
        for n in tasks:
            task = get_model_dict(n)
            task['creation_time'] = get_datetime_str(task['creation_time'])
            task['event_name'] = db_event_management.get_event_title(db, n.event_code)
            data.append(task)
        return {
            "code": 200,
            "msg": "任务查询成功",
            "data": data,
            "total": total_count
        }

    except Exception as e:
        traceback.print_exc()
        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")


@router.post("/leader_request")
async def leader_request(
    request: Request, 
    db: Session = Depends(get_db), 
    body = Depends(remove_xss_json), 
    user_id = Depends(valid_access_token)
):
    try:
        task_id = body['task_id']
        processing_status = body['processing_status']

        new_feeback = TaskFeeback(
            **body,
            recorded_by=user_id
        )
        db.add(new_feeback)
        db.commit()

        db.query(TaskRegistration).filter(TaskRegistration.task_id == task_id).update({"processing_status": processing_status, "update_time": datetime.now()})
        db.commit()

        return {
            "code": 200,
            "msg": "请示领导保存成功"
        }

    except Exception as e:
        # 处理异常
        db.rollback()
        traceback.print_exc()
        raise HTTPException(status_code=500, detail=str(e))
    

@router.post("/feeback")
async def feeback(
    request: Request, 
    db: Session = Depends(get_db), 
    body = Depends(remove_xss_json), 
    user_id = Depends(valid_access_token)
):
    try:
        feeback_user = get_req_param_optional(body, 'feeback_user')
        task_id = body['task_id']
        processing_status = body['processing_status']

        new_file_list = []
        if 'fileList' in body:
            new_file_list = copy.deepcopy(body['fileList'])
            del body['fileList']

        if feeback_user == '':
            feeback_user = db_user.get_user_info(db, user_id).nick_name

        if 'feeback_user' in body:
            del body['feeback_user']
        

        new_feeback = TaskFeeback(
            **body,
            feeback_user=feeback_user,
            recorded_by=user_id
        )
        db.add(new_feeback)
        db.commit()
        db.refresh(new_feeback)

        print(new_file_list)
        for file in new_file_list:
            file_name = file['file_name']
            file_name_desc = file['file_name_desc']
            status = file['status']
            new_file = TaskFile(
                file_id=new_guid(),
                foreign_key=new_feeback.id,
                from_scenario='task_feeback',
                file_name=file_name,
                file_name_desc=file_name_desc,
                status=status
            )
            db.add(new_file)
            db.commit()

        db.query(TaskRegistration).filter(TaskRegistration.task_id == task_id).update({"processing_status": processing_status, "update_time": datetime.now()})
        db.commit()

        return {
            "code": 200,
            "msg": "任务反馈保存成功"
        }

    except Exception as e:
        # 处理异常
        db.rollback()
        traceback.print_exc()
        raise HTTPException(status_code=500, detail=str(e))
    

@router.get('/detail')
async def get_task_detail(
    request: Request, 
    task_id: str = Query(None, description='任务编号'),
    db: Session = Depends(get_db)):
    
    print('task_id:',task_id)
    try: 
        # 构建查询
        rows = db.query(TaskFeeback).filter(and_(TaskFeeback.task_id == task_id, TaskFeeback.feeback_type == '0')).all()
        feebacks = []
        for row in rows:
            fileList = []

            feebacks.append({
                "content": row.content,
                "create_time": get_datetime_str(row.create_time),
                'fileList': fileList,
                "fileCount": len(fileList)
            })

        
        query = db.query(TaskRegistration)
        query = query.filter(TaskRegistration.task_id == task_id)

        # 执行查询
        row = query.first()
        if row is not None:
            return {
                "code": 200,
                "msg": "查询成功",
                "data": {
                    "id": row.id,
                    "task_id": row.task_id,
                    "task_type": row.task_type,
                    "event_code": row.event_code,
                    "event_name": db_event_management.get_event_title(db, row.event_code),
                    "processing_status": row.processing_status,
                    "expire_time": get_datetime_str(row.expire_time),
                    "update_time": get_datetime_str(row.update_time),
                    "feeback_user": row.feeback_user,
                    "task_description": row.task_description,
                    
                    "feebacks": feebacks
                }
            }
        
        else:
            return {
                "code": 500,
                "msg": "查询失败"
            }

    except Exception as e:
        # 处理异常
        traceback.print_exc()
        raise HTTPException(status_code=500, detail=str(e))