#!/usr/bin/env python3 # -*- coding: utf-8 -*- from fastapi import APIRouter, Request, Depends,Query, HTTPException, status,BackgroundTasks 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 router = APIRouter() @router.get('/list') async def get_inspection_task_list( type: str = Query(None, description='类型'), cycle :str = Query(None, description='周期'), page: int = Query(1, gt=0, description='页码'), pageSize: int = Query(10, gt=0, description='每页条目数量'), db: Session = Depends(get_db), user_id = Depends(valid_access_token) ): try: # 构建查询 query = db.query(RiskManagementRiskTask) query = query.filter(RiskManagementRiskTask.del_flag != '2') # 应用查询条件 if type: query = query.filter(RiskManagementRiskTask.risk_type == type) if cycle: query = query.filter(RiskManagementRiskTask.task_cycle == cycle) # 计算总条目数 total_items = query.count() # 排序 query = query.order_by(RiskManagementRiskTask.create_time.desc()) # 执行分页查询 RiskTasks = query.offset((page - 1) * pageSize).limit(pageSize).all() # 将查询结果转换为列表形式的字典 RiskTasks_list = [] for task in RiskTasks: if task.task_status=='3': task_status = '3' #'已完结' else: if datetime.now()= datetime_1.date(year_1, month_1, 1)) # query = query.filter(RiskManagementInspectionTaskChildrenTask.tsak_time <=datetime.now()) query = query.filter(RiskManagementRiskTaskChildrenTask.task_range.in_(task_range)) query = query.filter(RiskManagementRiskTaskChildrenTask.task_id.in_(task_ids)) # 计算总条目数 total_items = query.count() print(total_items) result = [] # 获取指定月份的第一天是周几 # first_day = datetime.date(yaer_1, month_1, 1) # 计算该月的天数 number_of_days = (datetime_1.date(year_1, month_1 + 1, 1) - datetime_1.timedelta(days=1)).day # 遍历该月的每一天 for day in range(1, number_of_days + 1): # 创建日期对象 date_obj = datetime_1.date(year_1, month_1, day) # 获取星期（0是周一，6是周日） week_day = date_obj.weekday() # 将星期转换为中文 week_day_cn = "一二三四五六日"[week_day] data = {"day":day,"date":f"{year_1}-{month_1}-{day}","week":f'周{week_day_cn}',"status":0} if date_obj==datetime_1.date.today(): data['status']=4 # 排序 # total_items = query.count() # print( '前',total_items) # print(datetime_1.date(year_1, month_1, day)) query_1 = query.filter( RiskManagementRiskTaskChildrenTask.tsak_time == datetime_1.date(year_1, month_1, day)) # query = query.order_by(RiskManagementInspectionTaskChildrenTask.tsak_time.asc()) # total_items = query_1.count() # print( '后',total_items) InspectionTasks = query_1.all() # 将查询结果转换为列表形式的字典 InspectionTasks_list = [] for task in InspectionTasks: # 判断这个任务该user_id完成了没 # tasklog = db.query(RiskManagementInspectionTaskChildrenTaskLog).\ # filter(RiskManagementInspectionTaskChildrenTaskLog.del_flag!='2').\ # filter(RiskManagementInspectionTaskChildrenTaskLog.children_task_id==task.id).\ # filter(RiskManagementInspectionTaskChildrenTaskLog.area_code.in_(user_area_codes)).all() # task_area_code_list = [i.area_code for i in tasklog] query_2 = db.query(RiskManagementInspectionUser).\ filter(RiskManagementInspectionUser.del_flag!='2').\ filter(RiskManagementInspectionUser.user_id==user_id) if task.task_range == '0': user_range_area_codes= ['440900000000'] elif task.task_range == '1': query_2 = query_2.filter( and_(RiskManagementInspectionUser.area_code.like('%000000'), RiskManagementInspectionUser.area_code.notlike('%00000000'))) user_range_area_codes= list({i.area_code for i in query_2.all()}) elif task.task_range == '2': query_2 = query_2.filter(and_(RiskManagementInspectionUser.area_code.like('%000'), RiskManagementInspectionUser.area_code.notlike('%000000'))) user_range_area_codes= list({i.area_code for i in query_2.all()}) elif task.task_range == '3': query_2 = query_2.filter(RiskManagementInspectionUser.area_code.notlike('%000')) user_range_area_codes= list({i.area_code for i in query_2.all()}) else: user_range_area_codes= [] print(task.task_range) print(user_range_area_codes) for area_code in user_range_area_codes: tasklog = db.query(RiskManagementRiskTaskChildrenTaskLog). \ filter(RiskManagementRiskTaskChildrenTaskLog.del_flag != '2'). \ filter(RiskManagementRiskTaskChildrenTaskLog.children_task_id == task.id). \ filter(RiskManagementRiskTaskChildrenTaskLog.area_code==area_code).first() task_info = { "id": task.id, "type": task.type, "cycle": task.cycle, "area_code":area_code, "area": area_code_get_ancestors_names(db,area_code_get_area_info(db,area_code)), "task_status": tasklog is not None, "task_time": task.tsak_time.strftime('%Y-%m-%d'), "create_time": task.create_time.strftime('%Y-%m-%d %H:%M') } if tasklog : task_info['create_time'] = tasklog.create_time.strftime('%Y-%m-%d %H:%M') if task.tsak_time datetime.today(): # data['status'] = 3 # 未来有事 else: if task.tsak_time < datetime.today() and data['status'] < 2: data['status'] = 2 # 有事未完成 elif task.tsak_time > datetime.today(): data['status'] = 3 # 未来有事 InspectionTasks_list.append(task_info) data['task_list']=InspectionTasks_list result.append(data) # 返回结果 return { "code": 200, "msg": "成功", "data": result } except Exception as e: # 处理异常 traceback.print_exc() raise HTTPException(status_code=500, detail=str(e)) @router.get('/children/task/records') async def get_children_task_records_list( type: str =Query(None, description='类型'), page: int = Query(1, gt=0, description='页码'), pageSize: int = Query(10, gt=0, description='每页条目数量'), db: Session = Depends(get_db), user_id = Depends(valid_access_token) ): try: # 获取子任务id query = db.query(RiskManagementRiskTaskChildrenTask) query = query.filter(RiskManagementRiskTaskChildrenTask.del_flag != '2') if type: query = query.filter(RiskManagementRiskTaskChildrenTask.type==type) children_ids = [i.id for i in query.all()] print(children_ids) # 构建查询 query = db.query(RiskManagementRiskTaskChildrenTaskLog) query = query.filter(RiskManagementRiskTaskChildrenTaskLog.del_flag != '2') query = query.filter(RiskManagementRiskTaskChildrenTaskLog.user_id == user_id) query = query.filter(RiskManagementRiskTaskChildrenTaskLog.children_task_id.in_(children_ids)) total_items = query.count() # 排序 query = query.order_by(RiskManagementRiskTaskChildrenTaskLog.create_time.desc()) # 执行分页查询 InspectionTasks = query.offset((page - 1) * pageSize).limit(pageSize).all() # 将查询结果转换为列表形式的字典 InspectionTasks_list = [] for task in InspectionTasks: children_task = risk_task_children_task_id_get_risk_task_children_task_info(db,task.children_task_id) task_info = { "id": task.children_task_id, "type": children_task.type, "task_range": children_task.task_range, "cycle": children_task.cycle, "area_code": task.area_code, "area": area_code_get_ancestors_names(db, area_code_get_area_info(db, task.area_code)), "task_time": children_task.tsak_time.strftime('%Y-%m-%d'), "create_time": task.create_time.strftime('%Y-%m-%d %H:%M') } InspectionTasks_list.append(task_info) # 返回结果 return { "code": 200, "msg": "成功", "data": InspectionTasks_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('/children/task/result/{children_task_id}') async def get_children_task_result( children_task_id: str , db: Session = Depends(get_db), page: int = Query(1, gt=0, description='页码'), pageSize: int = Query(10, gt=0, description='每页条目数量'), user_id = Depends(valid_access_token) ): try: # 构建查询 query = db.query(RiskManagementRiskTaskChildrenTaskResult) query = query.filter(RiskManagementRiskTaskChildrenTaskResult.del_flag != '2') # 应用查询条件 query = query.filter(RiskManagementRiskTaskChildrenTaskResult.children_task_id == children_task_id) # 计算总条目数 total_items = query.count() # 排序 query = query.order_by(RiskManagementRiskTaskChildrenTaskResult.create_time.desc()) # 执行分页查询 InspectionTasks = query.offset((page - 1) * pageSize).limit(pageSize).all() InspectionTasks_list = [] for task in InspectionTasks: area_code = task.area_code area = area_code_get_area_info(db, area_code) area = area_code_get_ancestors_names(db, area) task_info = { "id": task.id, # "children_task_id": task.children_task_id, "type":risk_task_children_task_id_get_risk_task_children_task_info(db,task.children_task_id).type, "point_name": task.inspection_point_name, # "area": area, # "create_time": task.create_time.strftime('%Y-%m-%d'), # "nick_name": task.nick_name, "task_result": task.inspection_result, "remark": task.remark, "fileList": get_file_query_fun(db=db, from_scenario='RiskManagementRiskTaskChildrenTaskResult', foreign_key=task.id) } InspectionTasks_list.append(task_info) # 返回结果 return { "code": 200, "msg": "成功", "data": InspectionTasks_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.post('/children/task/result/create') async def create_inspection_task( db: Session = Depends(get_db), body = Depends(remove_xss_json), user_id = Depends(valid_access_token) ): try: children_task_id = body['children_task_id'] result = body['result'] area_code = body['area_code'] area =area_code_get_ancestors_names(db, area_code_get_area_info(db, area_code)) # task_time = body['task_time'] # 创建新的 new_task_log = RiskManagementRiskTaskChildrenTaskLog( id=new_guid(), children_task_id=children_task_id, area_code = area_code, area = area, task_status = '1', user_id = user_id, nick_name = user_id_get_user_info(db,user_id).nick_name, create_by = user_id ) for info in result: new_file_list = info['fileList'] inspection_point_name= info['point_name'] inspection_result = info['task_result'] remark = info['remark'] new_task_result = RiskManagementRiskTaskChildrenTaskResult( id=new_guid(), children_task_id=children_task_id, inspection_point_name=inspection_point_name, area_code=area_code, inspection_result=inspection_result, remark = remark, user_id = user_id, nick_name = user_id_get_user_info(db,user_id).nick_name, create_by = user_id ) db.add(new_task_result) for file in new_file_list: file_name = file['file_name'] file_name_desc = file['file_name_desc'] status = file['status'] new_file = RiskManagementFile( file_id=new_guid(), foreign_key=new_task_result.id, from_scenario='RiskManagementRiskTaskChildrenTaskResult', file_name=file_name, file_name_desc=file_name_desc, status=status ) db.add(new_file) # 添加到数据库会话并提交 db.add(new_task_log) db.commit() # 返回创建成功的响应 return { "code": 200, "msg": "成功", "data": None } except Exception as e: # 处理异常 traceback.print_exc() raise HTTPException(status_code=500, detail=str(e)) ####中屏子任务 @router.get('/children/task/{task_id}/list') async def get_inspection_task_list( task_id: str , page: int = Query(1, gt=0, description='页码'), pageSize: int = Query(10, gt=0, description='每页条目数量'), db: Session = Depends(get_db), user_id = Depends(valid_access_token) ): try: # 构建查询 query = db.query(RiskManagementRiskTaskChildrenTask) query = query.filter(RiskManagementRiskTaskChildrenTask.del_flag != '2') # 应用查询条件 query = query.filter(RiskManagementRiskTaskChildrenTask.task_id == task_id) # 计算总条目数 total_items = query.count() # 排序 query = query.order_by(RiskManagementRiskTaskChildrenTask.tsak_time.asc()) # 执行分页查询 InspectionTasks = query.offset((page - 1) * pageSize).limit(pageSize).all() # 将查询结果转换为列表形式的字典 InspectionTasks_list = [] for task in InspectionTasks: task_num=task.task_num completed_num = db.query(RiskManagementRiskTaskChildrenTaskLog)\ .filter(RiskManagementRiskTaskChildrenTaskLog.del_flag!='2')\ .filter(RiskManagementRiskTaskChildrenTaskLog.children_task_id==task.id).count() incomplete_num = task_num-completed_num task_info = { "id": task.id, "task_id":task.task_id, "task_number": task.task_number, "type": task.type, "task_time": task.tsak_time.strftime('%Y-%m-%d'), "cycle": task.cycle, "task_range": task.task_range, "completed_num": completed_num, "incomplete_num": incomplete_num, "create_time": task.create_time.strftime('%Y-%m-%d') } InspectionTasks_list.append(task_info) # 返回结果 return { "code": 200, "msg": "成功", "data": InspectionTasks_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('/children/task/log/{children_task_id}/{status}/list') async def get_inspection_task_list( children_task_id: str , status:str, area_code: str = Query(None, description='区划编码'), page: int = Query(1, gt=0, description='页码'), pageSize: int = Query(10, gt=0, description='每页条目数量'), db: Session = Depends(get_db), user_id = Depends(valid_access_token) ): try: # 构建查询 total_items= 0 InspectionTasks_list = [] if status=='completed': query = db.query(RiskManagementRiskTaskChildrenTaskLog) query = query.filter(RiskManagementRiskTaskChildrenTaskLog.del_flag != '2') # 应用查询条件 query = query.filter(RiskManagementRiskTaskChildrenTaskLog.children_task_id == children_task_id) if area_code: query = query.filter(RiskManagementRiskTaskChildrenTaskLog.area_code == area_code) # 计算总条目数 total_items = query.count() # 排序 query = query.order_by(RiskManagementRiskTaskChildrenTaskLog.create_time.desc()) # 执行分页查询 InspectionTasks = query.offset((page - 1) * pageSize).limit(pageSize).all() # 将查询结果转换为列表形式的字典 for task in InspectionTasks: task_info = { "id": task.id, "children_task_id":children_task_id, "area_code": task.area_code, "area": task.area, "task_status": task.task_status, "user_id": task.user_id, "nick_name": task.nick_name, "create_time": task.create_time.strftime('%Y-%m-%d') } InspectionTasks_list.append(task_info) # 返回结果 elif status == 'incomplete': children_task = risk_task_children_task_id_get_risk_task_children_task_info(db,children_task_id) if children_task: task = risk_task_id_get_risk_task_info(db,children_task.task_id) if task: complete_area_code_list =[i.area_code for i in risk_task_children_task_id_get_risk_task_children_task_log_info(db,children_task_id)] area_code_list = get_area_code_exclude_list(db,task.task_range,complete_area_code_list) if area_code: area_code_list = area_code_rang_get_area_children_list(db, children_task.task_range, area_code, complete_area_code_list) for area_code in area_code_list: task_info = { "id": new_guid(), "children_task_id": children_task_id, "area_code": area_code, "area": area_code_get_ancestors_names(db,area_code_get_area_info(db,area_code)), "task_status": '0', "user_id": '', "nick_name": '', "create_time": children_task.tsak_time.strftime('%Y-%m-%d') } InspectionTasks_list.append(task_info) return { "code": 200, "msg": "成功", "data": InspectionTasks_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('/children/task/result/{children_task_id}/list') async def get_inspection_task_list( children_task_id: str , area_code: str = Query(None, description='区划编码'), result: str = Query(None, description='巡查结果'), nick_name: str = Query(None, description='姓名'), page: int = Query(1, gt=0, description='页码'), pageSize: int = Query(10, gt=0, description='每页条目数量'), db: Session = Depends(get_db), user_id = Depends(valid_access_token) ): try: # 构建查询 query = db.query(RiskManagementRiskTaskChildrenTaskResult) query = query.filter(RiskManagementRiskTaskChildrenTaskResult.del_flag != '2') query = query.filter(RiskManagementRiskTaskChildrenTaskResult.children_task_id == children_task_id) # 应用查询条件 if area_code: query = query.filter(RiskManagementRiskTaskChildrenTaskResult.area_code == area_code) if result: query = query.filter(RiskManagementRiskTaskChildrenTaskResult.inspection_result == result) if nick_name: query = query.filter(RiskManagementRiskTaskChildrenTaskResult.nick_name.like(f'%{nick_name}%') ) # 计算总条目数 total_items = query.count() # 排序 query = query.order_by(RiskManagementRiskTaskChildrenTaskResult.create_time.desc()) # 执行分页查询 InspectionTasks = query.offset((page - 1) * pageSize).limit(pageSize).all() # 将查询结果转换为列表形式的字典 InspectionTasks_list = [] for task in InspectionTasks: area_code = task.area_code area = area_code_get_area_info(db, area_code) area = area_code_get_ancestors_names(db, area) task_info = { "id": task.id, "children_task_id": task.children_task_id, "point_name":task.inspection_point_name, "area": area, "create_time": task.create_time.strftime('%Y-%m-%d'), "nick_name": task.nick_name, "result": task.inspection_result, "fileList": get_file_query_fun(db=db,from_scenario='RiskManagementRiskTaskChildrenTaskResult', foreign_key=task.id), "remark":task.remark } InspectionTasks_list.append(task_info) # 返回结果 return { "code": 200, "msg": "成功", "data": InspectionTasks_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))