from TestCenter.tc_impl import *


class TestObj:

    def __init__(self):
        self.rid = None
        self.uid = None
        self.email = None
        self.company = None
        self.code = None
        self.legal_person = None
        self.cid = None
        self.types = None
        self.record = None
        self.data = None

    def search_evaluate_record(self):
        """
        根据评价类型返回评价数据库
        """
        data = search_evaluate_record_scripts(self.types)
        return data

    def delete_cid_cascade(self):
        """
        级联删除与uid有关的所有数据
        """
        info = delete_cid_cascade_scripts(self.uid, self.data)
        return info

    def delete_rid_cascade(self):
        """
        级联删除与评价ID有关的所有数据
        """
        info = delete_rid_cascade_scripts(self.rid, self.record, self.data)
        return info

    def company_verify(self):
        """
        企业认证
        Returns:
            info: 认证结果
        """
        info = company_verify_script(self.company, self.code, self.legal_person, self.cid)
        return info

    def change_verified(self):
        """
        修改企业认证状态
        Returns:
            info: 修改结果
        """
        info = change_verified_script(self.cid)
        return info

    def new_test_company(self):
        """
        新增一家测试企业
        """
        info = new_test_company_impl(self.email, self.company, self.code, self.legal_person)
        return info

    def generate_rid_information(self):
        """
        生成企业主页数据
        """
        info = generate_rid_information_script(self.rid, self.types)
        return info

    def get_company_process(self):
        """
        获取企业测评流程
        """
        data = get_company_process_script(self.types, self.rid, self.cid)
        return data