#!/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.2.重大危险源安全管理系统
# 2.2.2.1.评价/评估报告管理及隐患管理
# 2.2.2.1.1.评价/评估报告数据表
@router.post('/evaluation', description='提交评价/评估报告')
async def index(
    request: Request, 
    db: Session = Depends(get_db),
    data: dict = Depends(remove_xss_json)
):
    try:
        id = get_req_param(data, 'id')
        company_industry = db.query(DangerSourceEvaluation).filter(DangerSourceEvaluation.id == id).first()
        if company_industry is None:
            new_entity = DangerSourceEvaluation(**data)
            db.add(new_entity)
        else:
            db.query(DangerSourceEvaluation).filter(DangerSourceEvaluation.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.2.4.企业重大危险源信息
@router.post('/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(DangerSourceInfo).filter(DangerSourceInfo.id == id).first()
        if info is None:
            new_entity = DangerSourceInfo(**data)
            db.add(new_entity)
        else:
            db.query(DangerSourceInfo).filter(DangerSourceInfo.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.2.3.隐患信息
@router.post('/problem', 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(DangerSourceProblem).filter(DangerSourceProblem.id == id).first()
        if info is None:
            new_entity = DangerSourceProblem(**data)
            db.add(new_entity)
        else:
            db.query(DangerSourceProblem).filter(DangerSourceProblem.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.2.4.危险源包保责任人信息
@router.post('/takes', 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(DangerSourceTakes).filter(DangerSourceTakes.id == id).first()
        if info is None:
            new_entity = DangerSourceTakes(**data)
            db.add(new_entity)
        else:
            db.query(DangerSourceTakes).filter(DangerSourceTakes.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.2.8.安全风险评估与管控信息
@router.post('/warning', 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(DangerSourceWarning).filter(DangerSourceWarning.id == id).first()
        if info is None:
            new_entity = DangerSourceWarning(**data)
            db.add(new_entity)
        else:
            db.query(DangerSourceWarning).filter(DangerSourceWarning.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":{}
        }