refactor: update report.py CLI and api_test.py for cust_code_list interface

api_test.py

@@ -5,8 +5,7 @@ url = "http://127.0.0.1:7960/brandcultivation/api/v1/recommend"
 payload = {
     "city_uuid": "00000000000000000000000011445301",
     "product_code": "310101",
-    "recall_cust_count": 100,
-    "delivery_count": 80,
+    "cust_code_list": ["445300108802", "445300108803"],
     "cultivacation_id": "10000001",
     "limit_cycle_name": "202606W1(06.01-06.07)"
 }

report.py

@@ -1,35 +1,28 @@
 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("--recall_count", type=int, default=500)
-    parser.add_argument("--delivery_count", type=int, default=5000)
-    
-    # parser.add_argument()
-    # parser.add_argument()
-    
+    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.recall_count, args.delivery_count)
-    
-    
+    report_utils.generate_all_data(args.cust_code_list)
+
+
 if __name__ == "__main__":
     run()
-    
-