feat:添加脚本

script/aigc_produce_plan_refresh.py

@@ -8,18 +8,29 @@ client = ApolloClient("http://apolloconfig-internal.piaoquantv.com")
 aigc_client = AIGCClient(token="8bf14f27fc3a486788f3383452422d72", base_url="https://aigc-api.aiddit.com")
 
 
-def main():
+def produce_plan_status_update():
     config_str = client.get_value(
         app_id="crawler-scheduler",
         cluster="default",
         namespace="application",
         key="hot.video.keywords.provide.task.config"
     )
+    supply_merge_cate2 = ['历史名人', '当代正能量人物', '国家科技力量', '国家力量']
+    produce_plan_id_set = set()
     config_json = json.loads(json_util.remove_comments(config_str))
     for merge_cate2 in config_json['mergeCate2PatternPlanIdMap']:
+        # if merge_cate2 not in supply_merge_cate2:
+        #     continue
         for mode in config_json['mergeCate2PatternPlanIdMap'][merge_cate2]:
             for produce_plan_id in config_json['mergeCate2PatternPlanIdMap'][merge_cate2][mode]:
-                aigc_client.update_produce_plan_status(produce_plan_id, 0)
+                produce_plan_id_set.add(produce_plan_id)
+
+    # for produce_plan_id in produce_plan_id_set:
+    #     aigc_client.update_produce_plan_status(produce_plan_id, 1)
+
+
+def main():
+    produce_plan_status_update()
 
 
 if __name__ == '__main__':