from DBHelper.MongoHelperInstance import DB_TYC, DB_TEST
from Company.CompanyUtils import CompanyUtils
from Utils.CommonUtil import CommonUtils


class OperationalImpl(object):
    """经营风险"""
    @staticmethod
    def basic_information(**kwargs):
        """工商信息"""
        name = kwargs['name']
        cid = CompanyUtils.make_new_cid()

        record = DB_TYC.find_single_data(
            '公司背景',
            '基本信息',
            {'企业名称': name},
            ['基本信息']
        )

        record = record['基本信息']
        data = dict()
        data['企业ID'] = cid
        data['企业名称'] = name
        data['企业类型'] = record['companyOrgType']
        data['经营状态'] = record['regStatus']
        data['法定代表人'] = record['legalPersonName']
        data['注册资本'] = record['regCapital']
        data['实缴资本'] = record['actualCapital']
        data['经营范围'] = record['businessScope']
        data['社会统一信用代码'] = record['creditCode']
        data['成立日期'] = CommonUtils.time_stamp(record['estiblishTime'])
        data['注册地址'] = record['regLocation']
        data['登记机关'] = record['regInstitute']
        data['核准日期'] = CommonUtils.time_stamp(record['approvedTime'])
        data['更新时间'] = CommonUtils.get_current_time()

        DB_TEST.upsert_single_data(
            "WR_DataBase_v2_test",
            "C3.1_工商信息",
            {"企业名称": name},
            data
        )

        return '导入成功'

    @staticmethod
    def common_view_method(**kwargs):
        """查看接口（通用）"""
        search = kwargs['search']
        cid = search['cid']
        sheet = kwargs['sheet']
        page_no = int(search['page_no'])
        page_size = int(search['page_size'])
        page_size = 10 if page_size > 10 else page_size

        total = DB_TEST.find_all_data_with_count(
            'WR_DataBase_v2_test',
            sheet,
            {"企业ID": cid}
        )

        record = DB_TEST.find_data_by_page_with_sort(
            "WR_DataBase_v2_test",
            sheet,
            {"企业ID": cid},
            [],
            {'_id': 1},
            page_size,
            page_no
        )

        result = {
            "total": total,
            "records": record
        }

        return result

    def get_cid(self, **kwargs):
        """查询工商信息中的企业ID"""
        name = kwargs['name']
        cid = DB_TEST.find_single_column(
            "WR_DataBase_v2_test",
            "C3.1_工商信息",
            {"企业名称": name},
            '企业ID'
        )
        if not cid:
            self.basic_information(name=name)
            cid = DB_TEST.find_single_column(
                "WR_DataBase_v2_test",
                "C3.1_工商信息",
                {"企业名称": name},
                '企业ID'
            )
        return cid

    def equity_pledge(self, **kwargs):
        """股权出质"""
        name = kwargs['name']
        cid = self.get_cid(name=name)

        records = DB_TYC.find_single_column(
            '经营风险',
            '股权出质',
            {'企业名称': name},
            '股权出质'
        )

        records = records['result']
        if records:
            for record in records:
                dict_data = dict()
                dict_data['企业ID'] = cid
                dict_data['登记日期'] = CommonUtils.time_stamp(record['regDate'])
                dict_data['登记编号'] = record['regNumber']
                dict_data['出质人'] = record['pledgor']
                dict_data['出质股权标的企业'] = record['targetCompany']
                dict_data['质权人'] = record['pledgee']
                dict_data['出质股权数额'] = record['equityAmount']
                dict_data['状态'] = record['state']
                dict_data['更新时间'] = CommonUtils.get_current_time()
                DB_TEST.insert_single_data(
                    "WR_DataBase_v2_test",
                    "C10.1_股权出质",
                    dict_data
                )

        return '导入成功'

    def mortgage_movable_property(self, **kwargs):
        """动产抵押"""
        name = kwargs['name']
        cid = self.get_cid(name=name)

        records = DB_TYC.find_single_column(
            '经营风险',
            '动产抵押',
            {'企业名称': name},
            '动产抵押'
        )

        records = records['result']
        if records:
            for record in records:
                dict_data = dict()
                dict_data['企业ID'] = cid
                dict_data['登记编号'] = record['baseInfo']['regNum']
                dict_data['登记机关'] = record['baseInfo']['regDepartment']
                dict_data['登记日期'] = record['baseInfo']['regDate']
                dict_data['抵押人'] = None
                dict_data['抵押权人'] = record['peopleInfo'][0]['peopleName']
                dict_data['被担保债权类型'] = record['baseInfo']['type']
                dict_data['被担保债权数额'] = record['baseInfo']['amount']
                dict_data['债务期限'] = record['baseInfo']['term']
                dict_data['抵押物信息'] = record['pawnInfoList']
                dict_data['状态'] = record['baseInfo']['status']
                dict_data['更新时间'] = CommonUtils.get_current_time()
                DB_TEST.insert_single_data(
                    "WR_DataBase_v2_test",
                    "C10.2_动产抵押",
                    dict_data
                )

        return '导入成功'

    def land_mortgage(self, **kwargs):
        """土地抵押"""
        name = kwargs['name']
        cid = self.get_cid(name=name)

        records = DB_TYC.find_single_column(
            '经营风险',
            '土地抵押',
            {'企业名称': name},
            '土地抵押'
        )

        records = records['result']
        if records:
            for record in records:
                dict_data = dict()
                dict_data['企业ID'] = cid
                dict_data['土地坐落'] = record['land_loc']
                dict_data['土地用途'] = record['mortgageToUser']
                dict_data['起始时间'] = record['startDate']
                dict_data['结束日期'] = record['endDate']
                dict_data['行政区'] = record['landAministrativeArea']
                dict_data['抵押面积'] = record['mortgageArea']
                dict_data['评估金额'] = None
                dict_data['抵押金额'] = None
                dict_data['更新时间'] = CommonUtils.get_current_time()
                DB_TEST.insert_single_data(
                    "WR_DataBase_v2_test",
                    "C10.3_土地抵押",
                    dict_data
                )

        return '导入成功'