from DBHelper.MongoHelperInstance import DB_GUA


class RatingResultImpl(object):

    @staticmethod
    def search(**kwargs):
        total = DB_GUA.find_all_data_with_count(
            "评级数据",
            "得分级别",
            {"信用级别": kwargs["level"]}
        )

        items = DB_GUA.find_data_by_page_with_sort(
            "评级数据",
            "得分级别",
            {"企业名称": kwargs["level"]},
            ["企业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