from cc_rating.cc_impl import *
from cc_rating.cc_utils import get_company_type


class CCRating:

    def __init__(self):
        self.rid = None
        self.cid = None

    def get_rating_records(self):
        """
        查询企业的评价结果
        Parameters:
            -
        Returns:
            records: 企业的评价记录
        """
        records = get_rating_records_impl(self.cid)
        return records

    def start_cc_rating(self):
        """
        开始企业综合评价
        Parameters:
            param desc
        Returns:
            info 信息
            result 综合评价填报数据
        """
        info, result = start_cc_rating_impl(self.cid)
        return info, result

    def save_cc_rating_input(self, input_form):
        """
        保存综合评价填报数据
        Parameters:
            input_form: desc
        Returns:
            res: desc
        """
        save_cc_rating_input_form(self.cid, input_form)
        return "保存成功"

    def exec_cc_rating(self):
        """
        提交数据，执行综合信用评价
        Parameters:
            -
        Returns:
            评分执行结果
        """
        res, info = exec_general_rating_impl(self.cid)
        if res:
            return "评分成功，请前往主页查看", 200
        else:
            return info, 210

    @staticmethod
    def choose_industry_obj():
        """
        选择行业
        Parameters:
            -
        Returns:
            res: desc
        """
        return choose_industry_script()

    def questionnaire_obj(self):
        """
        选择行业
        Parameters:
            -
        Returns:
            行业选项
        """
        industry = get_company_type(self.cid)
        return questionnaire_script(industry)