from Company.Basic.Business.BusinessImpl import BusinessImpl
from Company.Operational.EquityPledge.EquityPledgeObj import EquityPledgeObj
from Company.Operational.Mortgage.MortgageObj import MortgageMovableoObj
from DBHelper.MongoHelperInstance import DB_TYC, DB_TEST
from Utils.CommonUtil import CommonUtils


class MortgageMovableImpl(object):
    """动产抵押"""

    @staticmethod
    def auto_update(**kwargs):
        """新增/更新（自动）"""
        condition = kwargs['condition']

        def save_data(**args):
            """查询保存数据"""
            company_name = args['cname']
            company_id = args['cid']
            records = DB_TYC.find_all_data(
                '经营风险',
                '动产抵押',
                {'企业名称': company_name},
                ['动产抵押']
            )
            if records:
                list_data = list()
                for record in records:
                    records_item = record['动产抵押']['result']
                    for record_item in records_item:
                        dict_data = dict()
                        dict_data['企业ID'] = company_id
                        dict_data['登记编号'] = record_item['baseInfo']['regNum']
                        dict_data['登记机关'] = record_item['baseInfo']['regDepartment']
                        dict_data['登记日期'] = record_item['baseInfo']['regDate']
                        dict_data['抵押人'] = None
                        dict_data['抵押权人'] = record_item['peopleInfo'][0]['peopleName']
                        dict_data['被担保债权类型'] = record_item['baseInfo']['type']
                        dict_data['被担保债权数额'] = record_item['baseInfo']['amount']
                        dict_data['债务期限'] = record_item['baseInfo']['term']
                        dict_data['抵押物信息'] = record_item['pawnInfoList']
                        dict_data['状态'] = record_item['baseInfo']['status']
                        dict_data['更新时间'] = CommonUtils.get_current_time()
                        list_data.append(dict_data)
                return list_data
            else:
                return False

        if len(condition['cid']):
            cid = condition['cid']
            cname = DB_TEST.find_single_column(
                'WR_DataBase_v2_test',
                'C3.1_工商信息',
                {'企业ID': cid},
                '企业名称'
            )
            update = save_data(cname=cname, cid=cid)
            if update:
                for item in update:
                    DB_TEST.update_single_data(
                        "WR_DataBase_v2_test",
                        "C10.2_动产抵押",
                        {"企业ID": cid, "登记编号": item['登记编号']},
                        item
                    )
                return 'success'
            else:
                return 'update failed'
        elif len(condition['cname']):
            cname = condition['cname']
            cid = DB_TEST.find_single_column(
                'WR_DataBase_v2_test',
                'C3.1_工商信息',
                {'企业名称': cname},
                '企业ID'
            )
            if not cid:
                cid = BusinessImpl.auto_update(condition={"cname": cname, "cid": ""})
            insert = save_data(cname=cname, cid=cid)
            if insert:
                for item in insert:
                    DB_TEST.upsert_single_data(
                        "WR_DataBase_v2_test",
                        "C10.2_动产抵押",
                        {"企业ID": cid, "登记编号": item['登记编号']},
                        item
                    )
                return 'success'
            else:
                return 'new failed'
        else:
            return 'parameter exception'

    @staticmethod
    def delete(**kwargs):
        """删除"""
        param = kwargs['param']
        cid = param['cid']
        check_id = param['check_id']

        res = DB_TEST.delete_single_data(
            "WR_DataBase_v2_test",
            "C10.2_动产抵押",
            {"企业ID": cid, "登记编号": check_id},
        )

        if res:
            result = {
                "info": "success",
            }
        else:
            result = {
                "info": "no result"
            }
        return result

    @staticmethod
    def edit(**kwargs):
        """编辑"""
        param = kwargs['param']
        cid = param['cid']
        check_id = param['check_id']
        body = param['body']
        try:
            mortgage = MortgageMovableoObj()
            mortgage.registration_organ = body['登记机关']
            mortgage.registration_date = body['登记日期']
            mortgage.mortgagor = body['抵押人']
            mortgage.mortgagee = body['抵押权人']
            mortgage.secured_claim_type = body['被担保债权类型']
            mortgage.secured_claim_amount = body['被担保债权数额']
            mortgage.debt_duration = body['债务期限']
            mortgage.collateral_info = body['抵押物信息']
            mortgage.status = body['状态']
            mortgage.update_time = CommonUtils.get_current_time()
            update_data = mortgage.fields_toggle()
            DB_TEST.update_single_data(
                "WR_DataBase_v2_test",
                "C10.2_动产抵押",
                {"企业ID": cid, "登记编号": check_id},
                update_data
            )
            return 'success'
        except Exception:
            return 'edit failed'

    @staticmethod
    def view(**kwargs):
        """查看"""
        query = kwargs['query']
        cid = query['cid']
        page_num = int(query['page_num'])
        page_size = int(query['page_size'])
        page_size = 10 if page_size > 10 else page_size

        total = DB_TEST.find_all_data_with_count(
            'WR_DataBase_v2_test',
            'C10.2_动产抵押',
            {"企业ID": cid}
        )

        record = DB_TEST.find_data_by_page_with_sort(
            "WR_DataBase_v2_test",
            "C10.2_动产抵押",
            {"企业ID": cid},
            [],
            {'_id': 1},
            page_size,
            page_num
        )

        if record:
            update_time = record[-1]['更新时间']
            result = {
                "info": "success",
                "result": {
                    "total": total,
                    "update_time": update_time,
                    "records": record
                }
            }
        else:
            result = {
                "info": "no result"
            }
        return result