BrandCultivation/report.py

import os
import argparse
from utils import ReportUtils


def run():
    parser = argparse.ArgumentParser()

    parser.add_argument("--city_uuid", type=str, default="00000000000000000000000011445301")
    parser.add_argument("--product_id", type=str, default="350139")
    parser.add_argument("--cust_code_list", type=str, nargs="*", default=[])

    args = parser.parse_args()

    # 查找该城市的gbdt模型是否存在
    args.gbdtlr_model_path = os.path.join("./models/rank/weights/", args.city_uuid, "gbdtlr_model.pkl")
    if not os.path.exists(args.gbdtlr_model_path):
        print("该城市的模型还未训练，请先启动训练！！！")

    # 初始化report生成工具
    report_utils = ReportUtils(args.city_uuid, args.product_id)

    # 生成报告
    report_utils.generate_all_data(args.cust_code_list)


if __name__ == "__main__":
    run()