from flask import Blueprint, request

from Test.TestImpl import CompanyAuthImpl, CompanyIndexImpl, DeleteCompanyDataImpl, GenerateFillInData
from Utils.ErrorUtil import ReturnConditionCheckFailed
from common.APIAuth import api_secret

test_route = Blueprint('test', __name__)


@test_route.route('/enterprise_certification', methods=['POST'])
@api_secret
def enterprise_certification():
    """测试用企业认证"""
    try:
        company_auth = CompanyAuthImpl()
        company_auth.cid = request.json['cid']
        verify = company_auth.VerifyInfo()
        verify.name = request.json['company']
        verify.code = request.json['code']
        verify.legal_person = request.json['legal_person']
        company_auth.verify_info = verify
        info = company_auth.enterprise_certification()
        return info
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.failed_info}, e.status_code
    except KeyError:
        return {"info": "参数异常"}, 400


@test_route.route('/change_verified', methods=['GET'])
@api_secret
def change_verified():
    """企业认证状态修改"""
    try:
        company_auth = CompanyAuthImpl()
        company_auth.cid = request.args['cid']
        company_auth.change_verified()
        return {"info": "执行成功"}, 200
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.failed_info}, e.status_code
    except KeyError:
        return {"info": "参数异常"}, 400


@test_route.route('/generate_rid_information', methods=['POST'])
@api_secret
def generate_information():
    """根据rid生成企业主页信息"""
    try:
        company_index = CompanyIndexImpl()
        company_index.rid = request.json['rid']
        company_index.project = request.json['project']
        info = company_index.generate_homepage()
        return info
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.failed_info}, e.status_code
    except KeyError:
        return {"info": "参数异常"}, 400


@test_route.route('/new_company', methods=['POST'])
@api_secret
def new_company_route():
    """新增一家测试企业"""
    try:
        company_auth = CompanyAuthImpl()
        company_auth.email = request.json['email']
        company_auth.name = request.json['company']
        res = company_auth.company_register()
        if type(res) is str:
            company_auth.cid = res
            verify = company_auth.VerifyInfo()
            verify.name = request.json['company']
            verify.code = request.json['code']
            verify.legal_person = request.json['legal_person']
            company_auth.verify_info = verify
            info = company_auth.enterprise_certification()
        elif res is False:
            info = '注册失败'
        else:
            info = res
        return info
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.failed_info}, e.status_code
    except KeyError:
        return {"info": "参数异常"}, 400


@test_route.route('/delete_rid_cascade', methods=['POST'])
@api_secret
def delete_rid_cascade():
    """级联删除与评价ID有关的所有数据"""
    try:
        delete_data = DeleteCompanyDataImpl()
        delete_data.rid = request.json['rid']
        record = request.json['record']
        data = request.json['data']
        info = delete_data.form_rid_delete(record, data)
        return {"info": info}, 200
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.failed_info}, e.status_code
    except KeyError:
        return {"info": "参数异常"}, 400


@test_route.route('/delete_cid_cascade', methods=['POST'])
@api_secret
def delete_cid_cascade():
    """级联删除与企业ID有关的所有数据"""
    try:
        delete_data = DeleteCompanyDataImpl()
        delete_data.cid = request.json['cid']
        data = request.json['data']
        info = delete_data.form_cid_delete(data)
        return {"info": info}, 200
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.failed_info}, e.status_code
    except KeyError:
        return {"info": "参数异常"}, 400


@test_route.route('/insert_esg_data', methods=['POST'])
@api_secret
def insert_esg_data():
    """自动化生成esg问卷填报数据"""
    try:
        generate_data = GenerateFillInData()
        generate_data.rid = request.json['rid']
        info = generate_data.insert_esg_data()
        return {"info": info}, 200
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.failed_info}, e.status_code
    except KeyError:
        return {"info": "参数异常"}, 400