alessandrolollo / prefect-transform Goto Github PK

    # Raise error if both api_key and api_key_env_var are missing
    if not (api_key or api_key_env_var):
        msg = "Both `api_key` and `api_key_env_var` are missing."
        raise TransformConfigurationException(msg)

    # Raise error if api_key is missing and env var is not found
    if not api_key and api_key_env_var not in os.environ.keys():
        msg = "`api_key` is missing and `api_key_env_var` not found in env vars."
        raise TransformConfigurationException(msg)

    # Raise error if both mql_server_url and mql_server_url_env_var are missing
    if not (mql_server_url or mql_server_url_env_var):
        msg = "Both `mql_server_url` and `mql_server_url_env_var` are missing."
        raise TransformConfigurationException(msg)

    # Raise error if mql_server_url is missing and env var is not found
    if not mql_server_url and mql_server_url_env_var not in os.environ.keys():
        msg = """
        `mql_server_url` is missing and `mql_server_url_env_var` not found in env vars.
        """
        raise TransformConfigurationException(msg)

    if not materialization_name:
        msg = "`materialization_name` is missing."
        raise TransformConfigurationException(msg)

    mql_api_key = api_key or os.environ[api_key_env_var]
    mql_url = mql_server_url or os.environ[mql_server_url_env_var]
    use_async = not wait_for_creation

    try:
        mql_client = MQLClient(
            api_key=mql_api_key, mql_server_url=mql_url, use_async=use_async
        )
    except (AuthException, URLException) as e:
        msg = f"Cannot connect to Transform server! Error is: {e.msg}"
        raise TransformAuthException(msg)

class TransformCredentials(Block):

    api_key: Optional[str] = None
    api_key_env_var: Optional[str] = None
    mql_server_url: Optional[str] = None
    mql_server_url_env_var: Optional[str] = None

    # some validator like https://github.com/PrefectHQ/prefect-aws/blob/main/prefect_aws/s3.py#L266-L302

    def get_client(self):
        ...