from DBHelper.MongoHelperInstance import DB_GUA
from Utils.CommonUtil import trans_fields_name, sub_dict, get_attr


class CompanyManageImpl(object):

    @staticmethod
    def search(**kwargs):

        total = DB_GUA.find_all_data_with_count(
            "企业数据",
            "公司概览",
            {"企业名称": {"$regex": kwargs["company_name"]}}
        )

        items = DB_GUA.find_data_by_page_with_sort(
            "企业数据",
            "公司概览",
            {"企业名称": {"$regex": kwargs["company_name"]}},
            ["企业ID", "企业名称", "所在省份", "所在城市", "信用评级", "评级日期", "公司标签"],
            {"更新时间": -1},
            kwargs["page_size"],
            kwargs["page_no"]
        )

        trans_items = list()

        for item in items:
            trans_item = trans_fields_name(
                item,
                ["企业ID", "所在省份", "所在城市", "信用评级", "评级日期", "公司标签"],
                ["cid", "省份", "地级市", "级别", "日期", "标签"]
            )
            trans_items.append(trans_item)

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

        return result

    @staticmethod
    def basic_data(**kwargs):

        item = DB_GUA.find_single_data(
            "企业数据",
            "工商信息",
            {"企业ID": kwargs["cid"]},
            [
                "企业名称", "企业简称", "企业类型", "担保类型", "统一社会信用代码",
                "组织代码", "成立时间", "注册资本（万元）", "实缴资本（万元）", "经营范围",
                "所在省份", "所在城市", "注册地址", "通信地址", "电子邮箱",
                "联系电话", "参保人数", "公司定位", "法定代表人", "实际控制人",
                "高管信息", "股东信息"
            ]
        )

        if not item:
            result = {}
        else:
            result = dict()

            business_info = sub_dict(
                item,
                [
                    "企业名称", "企业简称", "企业类型", "担保类型", "统一社会信用代码",
                    "组织代码", "成立时间", "注册资本（万元）", "实缴资本（万元）", "经营范围",
                    "所在省份", "所在城市", "注册地址", "通信地址", "电子邮箱",
                    "联系电话", "参保人数", "公司定位", "法定代表人", "实际控制人"
                ]
            )

            result["工商信息"] = business_info
            result["高管信息"] = get_attr(item, ["高管信息"], default=[])
            result["股东信息"] = get_attr(item, ["股东信息"], default=[])

        return result