from DBHelper.MongoHelperInstance import DB_TEST


class CommonImpl(object):
    """企业数据管理"""

    @staticmethod
    def search_company(**kwargs):
        """查询"""
        data = kwargs['data']
        search = dict()
        if 'cid' in data:
            search['企业ID'] = {"$regex": data['cid']}
        elif 'cname' in data:
            search['企业名称'] = {"$regex": data['cname']}

        page_no = int(data['page_no'])
        page_size = int(data['page_size'])
        page_size = 10 if page_size > 10 else page_size

        total = DB_TEST.find_all_data_with_count(
            "WR_DataBase_v2_test",
            "C3.1_工商信息",
            search,
        )

        records = DB_TEST.find_data_by_page_with_sort(
            "WR_DataBase_v2_test",
            "C3.1_工商信息",
            search,
            ['企业ID', '企业名称', '注册资本', '实缴资本', '法定代表人'],
            {'_id': 1},
            page_size,
            page_no
        )

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

        return result