from DBHelper.MongoHelperInstance import DB_GUA
from Utils.CommonUtil import trans_fields_name


class RatingResultImpl(object):

    @staticmethod
    def search(**kwargs):
        total = DB_GUA.find_all_data_with_count(
            "企业数据",
            "公司概览",
            {"信用评级": {"$regex": kwargs["level"]}}
        )

        items = DB_GUA.find_data_by_page_with_sort(
            "企业数据",
            "公司概览",
            {"信用评级": {"$regex": kwargs["level"]}},
            ["信用评级", "企业名称", "所在省份", "评级日期"],
            {"评级日期": -1},
            kwargs["page_size"],
            kwargs["page_no"]
        )

        trans_items = list()

        for item in items:
            trans_item = trans_fields_name(
                item,
                ["信用评级", "企业名称", "所在省份", "评级日期"],
                ["级别", "企业名称", "省份", "日期"]
            )
            trans_items.append(trans_item)

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

        return result