import json

import requests

from DBHelper.MongoHelper import MongoHelper
from Test.TestObj import CompanyUser, RatingRecord


class CompanyAuthImpl(CompanyUser):
    """企业相关类"""
    db = MongoHelper("tfse_v0.21")

    def company_register(self):
        """企业注册"""

    def enterprise_certification(self):
        """企业认证（测试用）"""
        insert_data = self.VerifyInfo().dict_to_save()
        self.db.update_single_data(
            "应用端",
            "企业用户",
            {"企业ID": self.cid},
            {"认证信息": insert_data}
        )

        def certification_interface():
            """调用应用端认证接口"""
            url = "https://api.fecribd.com/app/test/company_verify?cid={}".format(self.cid)
            secret = 'EZgo9ykxrYuBMYnYmmKIh'
            headers = {'secret': secret, "Content-Type": "application/json;charset=UTF-8"}
            res = requests.get(url, headers=headers)
            return json.loads(res.text)

        return certification_interface()

    def change_verified(self):
        """企业认证状态修改"""
        verify_info = self.VerifyInfo()
        verify_info.name = None
        verify_info.code = None
        verify_info.legal_person = None
        insert_data = verify_info.dict_to_save()
        self.db.update_single_data(
            "应用端",
            "企业用户",
            {"企业ID": self.cid},
            {"已认证": "否", "认证信息": insert_data}
        )


class CompanyIndexImpl(RatingRecord):
    """企业主页"""
    db = MongoHelper("tfse_v0.21")

    def generate_homepage(self):
        """生成主页数据"""
        self.cid = self.db.find_single_data(
            '企业数据',
            '评价记录',
            {"评价ID": self.rid},
            ['企业ID']
        )['企业ID']
        data = self.dict_to_save(columns=['企业ID', '评价项目'])

        def generate_interface():
            """调用打分接口"""
            url = "https://api.fecribd.com/app/test/general_rating"
            secret = 'EZgo9ykxrYuBMYnYmmKIh'
            headers = {'secret': secret, "Content-Type": "application/json;charset=UTF-8"}
            insert = json.dumps(data)
            res = requests.post(url, headers=headers, data=insert)
            return json.loads(res.text)

        return generate_interface()