aigc_produce_plan_refresh.py 929 B

1234567891011121314151617181920212223242526
  1. import json
  2. from client.AIGCClient import AIGCClient
  3. from client.ApolloClient import ApolloClient
  4. from util import json_util
  5. client = ApolloClient("http://apolloconfig-internal.piaoquantv.com")
  6. aigc_client = AIGCClient(token="8bf14f27fc3a486788f3383452422d72", base_url="https://aigc-api.aiddit.com")
  7. def main():
  8. config_str = client.get_value(
  9. app_id="crawler-scheduler",
  10. cluster="default",
  11. namespace="application",
  12. key="hot.video.keywords.provide.task.config"
  13. )
  14. config_json = json.loads(json_util.remove_comments(config_str))
  15. for merge_cate2 in config_json['mergeCate2PatternPlanIdMap']:
  16. for mode in config_json['mergeCate2PatternPlanIdMap'][merge_cate2]:
  17. for produce_plan_id in config_json['mergeCate2PatternPlanIdMap'][merge_cate2][mode]:
  18. aigc_client.update_produce_plan_status(produce_plan_id, 0)
  19. if __name__ == '__main__':
  20. main()