import xlrd

from DBHelper.MongoHelperInstance import DB_TEST
from Utils.CommonUtil import CommonUtils


class CompanyUtils(object):

    @staticmethod
    def make_new_cid():
        new_cid = CommonUtils.random_code(8)
        case = DB_TEST.find_single_column(
            "WR_DataBase_v2_test",
            "C3.1_工商信息",
            {"企业ID": new_cid},
            "企业ID"
        ) is not None
        while case:
            new_cid = CommonUtils.random_code(8)
        return new_cid

    @staticmethod
    def make_new_change_id():
        new_cid = CommonUtils.random_code(8)
        case = DB_TEST.find_single_column(
            "WR_DataBase_v2_test",
            "C3.6_工商变更",
            {"变更ID": new_cid},
            "变更ID"
        ) is not None
        while case:
            new_cid = CommonUtils.random_code(8)
        return new_cid

    @staticmethod
    def make_new_land_id():
        new_cid = CommonUtils.random_code(8)
        case = DB_TEST.find_single_column(
            "WR_DataBase_v2_test",
            "C10.3_土地抵押",
            {"抵押ID": new_cid},
            "抵押ID"
        ) is not None
        while case:
            new_cid = CommonUtils.random_code(8)
        return new_cid

    @staticmethod
    def make_new_abnormal_id():
        new_cid = CommonUtils.random_code(8)
        case = DB_TEST.find_single_column(
            "WR_DataBase_v2_test",
            "C11.1_经营异常",
            {"异常ID": new_cid},
            "异常ID"
        ) is not None
        while case:
            new_cid = CommonUtils.random_code(8)
        return new_cid

    @staticmethod
    def make_new_penalties_id():
        new_cid = CommonUtils.random_code(8)
        case = DB_TEST.find_single_column(
            "WR_DataBase_v2_test",
            "C11.2_行政处罚",
            {"处罚ID": new_cid},
            "处罚ID"
        ) is not None
        while case:
            new_cid = CommonUtils.random_code(8)
        return new_cid

    @staticmethod
    def make_new_violation_id():
        new_cid = CommonUtils.random_code(8)
        case = DB_TEST.find_single_column(
            "WR_DataBase_v2_test",
            "C11.3_严重违法",
            {"违法ID": new_cid},
            "违法ID"
        ) is not None
        while case:
            new_cid = CommonUtils.random_code(8)
        return new_cid

    @staticmethod
    def make_new_tax_id():
        new_cid = CommonUtils.random_code(8)
        case = DB_TEST.find_single_column(
            "WR_DataBase_v2_test",
            "C11.4_税收违法",
            {"违法ID": new_cid},
            "违法ID"
        ) is not None
        while case:
            new_cid = CommonUtils.random_code(8)
        return new_cid

    @staticmethod
    def make_new_arrears_id():
        new_cid = CommonUtils.random_code(8)
        case = DB_TEST.find_single_column(
            "WR_DataBase_v2_test",
            "C11.6_欠税公告",
            {"欠税ID": new_cid},
            "欠税ID"
        ) is not None
        while case:
            new_cid = CommonUtils.random_code(8)
        return new_cid

    @staticmethod
    def excel_sheet_parser(**kwargs):
        file = kwargs['file']
        work_book = xlrd.open_workbook(file_contents=file.read())
        sheet = work_book.sheet_by_index(0)
        length = len(sheet.col_values(0))
        retur_list = list()
        if length > 1:
            for item in range(1, length):
                key_list = sheet.row_values(0)
                return_dict = dict()
                for num in range(len(key_list)):
                    return_dict[key_list[num]] = sheet.row_values(item)[num]
                retur_list.append(return_dict)
        return retur_list