from DBHelper.MongoHelperInstance import DB_GUA
from Modules.Common.CommonUtils import CommonUtils


class RatingUtils(object):

    @staticmethod
    def make_new_rid():
        """新建评级ID"""
        new_rid = CommonUtils.random_code(8)
        case = DB_GUA.find_single_column(
            "评级数据",
            "评级记录",
            {"评级ID": new_rid},
            "评级ID"
        ) is not None
        while case:
            new_rid = CommonUtils.random_code(8)
        return new_rid

    @staticmethod
    def in_progress_rating(cid):
        """查找进行中的评级"""
        data = DB_GUA.find_single_data(
            "评级数据",
            "评级记录",
            {"企业ID": cid, "评级状态": "进行"},
            ["评级ID", "企业ID", "企业名称", "报告期"]
        )
        return data

    @staticmethod
    def last_report_date(report_date):
        """输入期末得到期初"""
        curr_year = int(report_date.split("-")[0])
        last_year = '{}-12-31'.format(curr_year-1)
        return last_year

    @staticmethod
    def check_cid_legal(cid):
        _cid = DB_GUA.find_single_column(
            "企业数据",
            "工商信息",
            {"企业ID": cid},
            "企业ID"
        )
        assert _cid, "错误的企业ID"