from App.Crud.CategoryCrud import create_major_category, create_sub_category
from App.Schema.CategorySchema import CreateMajorCategoryReqBody, CreateSubCategoryReqBody
from Utils.DataBase.SqlAlchemyUtils import Session


def init_major_category():
    db = Session()
    items = [
        {"name": "主体评级"},
        {"name": "债项评级"},
        {"name": "结构融资评级"},
        {"name": "绿色金融评级"},
    ]
    for item in items:
        create_major_category(db=db, schema=CreateMajorCategoryReqBody(**item))


def init_sub_category():
    db = Session()
    items = [
        {"name": "国家主权评级", "major_category_id": "M_CATEGORY_00001"},
        {"name": "地方政府评级", "major_category_id": "M_CATEGORY_00001"},
        {"name": "城投公司评级", "major_category_id": "M_CATEGORY_00001"},
        {"name": "工商企业评级", "major_category_id": "M_CATEGORY_00001"},
        {"name": "金融机构评级", "major_category_id": "M_CATEGORY_00001"},

        {"name": "企业债", "major_category_id": "M_CATEGORY_00002"},
        {"name": "公司债", "major_category_id": "M_CATEGORY_00002"},
        {"name": "中期票据", "major_category_id": "M_CATEGORY_00002"},
        {"name": "短期融资券", "major_category_id": "M_CATEGORY_00002"},
        {"name": "地方政府专项债", "major_category_id": "M_CATEGORY_00002"},

        {"name": "资产支持票据", "major_category_id": "M_CATEGORY_00003"},
        {"name": "资产支持证券", "major_category_id": "M_CATEGORY_00003"},
        {"name": "其他", "major_category_id": "M_CATEGORY_00003"},

        {"name": "绿色认证", "major_category_id": "M_CATEGORY_00004"},
        {"name": "ESG评级", "major_category_id": "M_CATEGORY_00004"},
        {"name": "企业碳资信评价", "major_category_id": "M_CATEGORY_00004"},
        {"name": "生态产品价值(GEP)核算", "major_category_id": "M_CATEGORY_00004"}

    ]
    for item in items:
        create_sub_category(db=db, schema=CreateSubCategoryReqBody(**item))


if __name__ == '__main__':
    init_major_category()
    init_sub_category()