#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from fastapi import APIRouter, Request, Depends, Form, Body, File, UploadFile
from database import get_db
from utils.StripTagsHTMLParser import *
from sqlalchemy.orm import Session
from datetime import datetime, timedelta
import jwt
from passlib.context import CryptContext
from models import *
from sqlalchemy import text, exists, and_, or_, not_
from sqlalchemy.sql import func
from models import *
from extensions import logger
from utils import *
import traceback

router = APIRouter()

# 2.2.4.智能巡检作业信息
# 2.2.4.1.巡检计划信息
@router.post('/plan/info', description='提交巡检计划信息')
async def index(
    request: Request, 
    db: Session = Depends(get_db),
    data: dict = Depends(remove_xss_json)
):
    try:
        id = get_req_param(data, 'id')
        info = db.query(DangerInspectionPlanInfo).filter(DangerInspectionPlanInfo.id == id).first()
        if info is None:
            new_entity = DangerInspectionPlanInfo(**data)
            db.add(new_entity)
        else:
            db.query(DangerInspectionPlanInfo).filter(DangerInspectionPlanInfo.id == id).update(data)
        db.commit()

        return {
            "code":1,
            "message":"成功",
            "data":{}
        }
    except Exception as e:
        traceback.print_exc()
        return {
            "code":0,
            "message":"更新异常，错误信息：" + str(e),
            "data":{}
        }
    


# 2.2.4.2.巡检节点信息
@router.post('/plan/point', description='提交巡检节点信息')
async def index(
    request: Request, 
    db: Session = Depends(get_db),
    data: dict = Depends(remove_xss_json)
):
    try:
        id = get_req_param(data, 'id')
        info = db.query(DangerInspectionPlanPoint).filter(DangerInspectionPlanPoint.id == id).first()
        if info is None:
            new_entity = DangerInspectionPlanPoint(**data)
            db.add(new_entity)
        else:
            db.query(DangerInspectionPlanPoint).filter(DangerInspectionPlanPoint.id == id).update(data)
        db.commit()

        return {
            "code":1,
            "message":"成功",
            "data":{}
        }
    except Exception as e:
        traceback.print_exc()
        return {
            "code":0,
            "message":"更新异常，错误信息：" + str(e),
            "data":{}
        }
    


# 2.2.4.3.检查标准信息
@router.post('/plan/rule', description='提交检查标准信息')
async def index(
    request: Request, 
    db: Session = Depends(get_db),
    data: dict = Depends(remove_xss_json)
):
    try:
        id = get_req_param(data, 'id')
        info = db.query(DangerInspectionPlanRule).filter(DangerInspectionPlanRule.id == id).first()
        if info is None:
            new_entity = DangerInspectionPlanRule(**data)
            db.add(new_entity)
        else:
            db.query(DangerInspectionPlanRule).filter(DangerInspectionPlanRule.id == id).update(data)
        db.commit()

        return {
            "code":1,
            "message":"成功",
            "data":{}
        }
    except Exception as e:
        traceback.print_exc()
        return {
            "code":0,
            "message":"更新异常，错误信息：" + str(e),
            "data":{}
        }
    


# 2.2.4.4.巡检对象信息
@router.post('/plan/obj', description='提交巡检对象信息')
async def index(
    request: Request, 
    db: Session = Depends(get_db),
    data: dict = Depends(remove_xss_json)
):
    try:
        id = get_req_param(data, 'id')
        info = db.query(DangerInspectionPlanObj).filter(DangerInspectionPlanObj.id == id).first()
        if info is None:
            new_entity = DangerInspectionPlanObj(**data)
            db.add(new_entity)
        else:
            db.query(DangerInspectionPlanObj).filter(DangerInspectionPlanObj.id == id).update(data)
        db.commit()

        return {
            "code":1,
            "message":"成功",
            "data":{}
        }
    except Exception as e:
        traceback.print_exc()
        return {
            "code":0,
            "message":"更新异常，错误信息：" + str(e),
            "data":{}
        }
    




# 2.2.4.5.巡检任务信息
@router.post('/task/info', description='提交巡检任务信息')
async def index(
    request: Request, 
    db: Session = Depends(get_db),
    data: dict = Depends(remove_xss_json)
):
    try:
        id = get_req_param(data, 'id')
        info = db.query(DangerInspectionTaskInfo).filter(DangerInspectionTaskInfo.id == id).first()
        if info is None:
            new_entity = DangerInspectionTaskInfo(**data)
            db.add(new_entity)
        else:
            db.query(DangerInspectionTaskInfo).filter(DangerInspectionTaskInfo.id == id).update(data)
        db.commit()

        return {
            "code":1,
            "message":"成功",
            "data":{}
        }
    except Exception as e:
        traceback.print_exc()
        return {
            "code":0,
            "message":"更新异常，错误信息：" + str(e),
            "data":{}
        }
    



# 2.2.4.6.巡检记录信息
@router.post('/task/record', description='提交巡检记录信息')
async def index(
    request: Request, 
    db: Session = Depends(get_db),
    data: dict = Depends(remove_xss_json)
):
    try:
        id = get_req_param(data, 'id')
        info = db.query(DangerInspectionTaskRecord).filter(DangerInspectionTaskRecord.id == id).first()
        if info is None:
            new_entity = DangerInspectionTaskRecord(**data)
            db.add(new_entity)
        else:
            db.query(DangerInspectionTaskRecord).filter(DangerInspectionTaskRecord.id == id).update(data)
        db.commit()

        return {
            "code":1,
            "message":"成功",
            "data":{}
        }
    except Exception as e:
        traceback.print_exc()
        return {
            "code":0,
            "message":"更新异常，错误信息：" + str(e),
            "data":{}
        }
    



# 2.2.4.7.巡检统计结果信息
@router.post('/task/statistics', description='提交巡检统计结果信息')
async def index(
    request: Request, 
    db: Session = Depends(get_db),
    data: dict = Depends(remove_xss_json)
):
    try:
        id = get_req_param(data, 'id')
        info = db.query(DangerInspectionTaskStatistics).filter(DangerInspectionTaskStatistics.id == id).first()
        if info is None:
            new_entity = DangerInspectionTaskStatistics(**data)
            db.add(new_entity)
        else:
            db.query(DangerInspectionTaskStatistics).filter(DangerInspectionTaskStatistics.id == id).update(data)
        db.commit()

        return {
            "code":1,
            "message":"成功",
            "data":{}
        }
    except Exception as e:
        traceback.print_exc()
        return {
            "code":0,
            "message":"更新异常，错误信息：" + str(e),
            "data":{}
        }