from Manage.rating.scripts import rating_result_script, search_rating_script, fill_detailss_script, risk_info_script, \
    report_edit_script, report_delete_script, report_create_script, get_report_text_model_impl, \
    get_company_process_script


class Rating:

    def __init__(self):
        self.query = None
        self.rid = None
        self.industry = None
        self.cid = None
        self.types = None

    def search_rating(self, skip, limit):
        """
        企业评价记录搜索
        Parameters:
            -
        Returns:
            result: 查询结果
        """
        result = search_rating_script(self.query, skip, limit)
        return result

    def search_rating_result(self):
        """
        查询评价结果
        Parameters:
            -
        Returns:
            result: 查询结果
        """
        result = rating_result_script(self.rid)
        return result

    def search_fill_details(self):
        """
        查看填报详情信息
        Parameters:
            -
        Returns:
            result: 查询结果
        """
        result = fill_detailss_script(self.rid)
        return result

    def search_risk_info(self):
        """
        查看风险数据信息
        Parameters:
            -
        Returns:
            result: 查询结果
        """
        result = risk_info_script(self.rid)
        return result

    def report_edit(self, text_model):
        """
        编辑报告数据
        Parameters:
            text_model: 报告数据
        Returns:
            result: 编辑结果
        """
        result = report_edit_script(self.rid, text_model)
        return result

    def report_delete(self):
        result = report_delete_script(self.rid)
        return result

    def report_create(self):
        """
        报告生成
        Parameters:
            -
        Returns:
            result: 生成结果
        """
        result = report_create_script(self.rid, self.cid)
        return result

    def get_report_text_model(self):
        """
        获取报告text_model
        """
        return get_report_text_model_impl(self.rid)

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