import pyapollos


class ApolloApi:
    def __init__(self, app_id="LongArticlesJob", env="pre"):
        match env:
            case "pre":
                config_server_url = 'http://preapolloconfig-internal.piaoquantv.com/'
            case "dev":
                config_server_url = 'https://devapolloconfig-internal.piaoquantv.com/'
            case "prod":
                config_server_url = 'https://apolloconfig-internal.piaoquantv.com/'
            case _:
                raise ValueError("env must be 'pre' or 'dev' or 'prod'")

        self.apollo_connection = pyapollos.ApolloClient(
            app_id=app_id,
            config_server_url=config_server_url,
            timeout=10
        )

    def get_config_value(self, key):
        return self.apollo_connection.get_value(key)