风险防控-风险防控-小屏开发。

routers/api/riskManagement/risk_router.py

@@ -336,8 +336,441 @@ async def delete_inspection_task(
         raise HTTPException(status_code=500, detail=str(e))
 
 
+####小屏子任务查询
+
+@router.get('/children/task/list')
+async def get_risk_task_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:
+        # 构建查询
+        # 查询用户负责层级
+        task_range = user_id_get_task_range(db,user_id)
+        #查询未办结巡查任务列表
+        Task_list = db.query(RiskManagementRiskTask)\
+            .filter(RiskManagementRiskTask.del_flag!='2')\
+            .filter(RiskManagementRiskTask.task_range.in_(task_range))\
+            .filter(RiskManagementRiskTask.task_status!='3').all()
+        #获取所有巡查任务id
+        task_ids = [i.id for i in Task_list]
+        #用户所有负责的区域
+        user_area_code_list = db.query(RiskManagementInspectionUser)\
+            .filter(RiskManagementInspectionUser.del_flag!='2')\
+            .filter(RiskManagementInspectionUser.user_id==user_id).all()
+        user_area_codes = [i.area_code for i in user_area_code_list]
+        #查询子任务
+        query = db.query(RiskManagementRiskTaskChildrenTask)
+        query = query.filter(RiskManagementRiskTaskChildrenTask.del_flag != '2')
+        # 应用查询条件
+        # 查询小于今天
+        query = query.filter(RiskManagementRiskTaskChildrenTask.tsak_time <=datetime.now())
+        # 查询层级、未办结子任务
+        query = query.filter(RiskManagementRiskTaskChildrenTask.task_range.in_(task_range))
+        query = query.filter(RiskManagementRiskTaskChildrenTask.task_id.in_(task_ids))
+        # 计算总条目数
+        if type:
+            query = query.filter(RiskManagementRiskTaskChildrenTask.type==type)
+        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:
+            # 判断这个任务该user_id完成了没
+            tasklog = db.query(RiskManagementRiskTaskChildrenTaskLog).\
+                filter(RiskManagementRiskTaskChildrenTaskLog.del_flag!='2').\
+                filter(RiskManagementRiskTaskChildrenTaskLog.children_task_id==task.id).\
+                filter(RiskManagementRiskTaskChildrenTaskLog.area_code.in_(user_area_codes)).all()
+            task_area_code_list = [i.area_code for i in tasklog]
+            query = db.query(RiskManagementInspectionUser)\
+                .filter(RiskManagementInspectionUser.del_flag!='2')\
+                .filter(RiskManagementInspectionUser.user_id==user_id)
+            if task.task_range == '0' and '440900000000' not in task_area_code_list:
+                user_range_area_codes= ['440900000000']
+            elif task.task_range == '1':
+                query = query.filter(
+                    and_(RiskManagementInspectionUser.area_code.like('%000000'),
+                         RiskManagementInspectionUser.area_code.notlike('%00000000'),
+                         ~RiskManagementInspectionUser.area_code.in_(task_area_code_list)))
+                user_range_area_codes= list({i.area_code for i in query.all()})
+            elif task.task_range == '2':
+                query = query.filter(and_(RiskManagementInspectionUser.area_code.like('%000'),
+                                          RiskManagementInspectionUser.area_code.notlike('%000000'),
+                                          ~RiskManagementInspectionUser.area_code.in_(task_area_code_list)))
+                user_range_area_codes= list({i.area_code for i in query.all()})
+            elif task.task_range == '3':
+                query = query.filter(RiskManagementInspectionUser.area_code.notlike('%000'),
+                                     ~RiskManagementInspectionUser.area_code.in_(task_area_code_list))
+                user_range_area_codes= list({i.area_code for i in query.all()})
+            else:
+                user_range_area_codes= []
+
+            for area_code in user_range_area_codes:
+                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_time": task.tsak_time.strftime('%Y-%m-%d'),
+                    "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/calendar/list')
+async def get_inspection_task_list(
+    year_1: int =Query(None, description='年份'),
+    month_1: int =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:
+        # 构建查询
+        import datetime as datetime_1
+
+        task_range = user_id_get_task_range(db,user_id)
+        Task_list = db.query(RiskManagementRiskTask).\
+            filter(RiskManagementRiskTask.del_flag!='2').\
+            filter(RiskManagementRiskTask.task_range.in_(task_range)).all() #.filter(RiskManagementInspectionTask.task_status!='3')
+        task_ids = [i.id for i in Task_list]
+        user_area_code_list = db.query(RiskManagementInspectionUser).\
+            filter(RiskManagementInspectionUser.del_flag!='2').\
+            filter(RiskManagementInspectionUser.user_id==user_id).all()
+        user_area_codes = [i.area_code for i in user_area_code_list]
+        query = db.query(RiskManagementRiskTaskChildrenTask)
+        query = query.filter(RiskManagementRiskTaskChildrenTask.del_flag != '2')
+        # 应用查询条件
+        current_date = datetime.today()
+        if year_1 is None:
+            year_1 =current_date.year
+        if month_1 is None:
+            month_1 = current_date.month
+        query = query.filter(RiskManagementRiskTaskChildrenTask.tsak_time < datetime_1.date(year_1, month_1 + 1, 1))
+        query = query.filter(RiskManagementRiskTaskChildrenTask.tsak_time >= 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() and data['status']<1:
+                            data['status'] = 1 #有事已完成
+                        # elif tasklog.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,
+                "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 ,