import json

from client.AIGCClient import AIGCClient
from client.ApolloClient import ApolloClient
from util import json_util

client = ApolloClient("http://apolloconfig-internal.piaoquantv.com")
aigc_client = AIGCClient(token="8bf14f27fc3a486788f3383452422d72", base_url="https://aigc-api.aiddit.com")


def main():
    config_str = client.get_value(
        app_id="crawler-scheduler",
        cluster="default",
        namespace="application",
        key="hot.video.keywords.provide.task.config"
    )
    config_json = json.loads(json_util.remove_comments(config_str))
    for merge_cate2 in config_json['mergeCate2PatternPlanIdMap']:
        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)


if __name__ == '__main__':
    main()