diff --git a/metadata-ingestion/src/datahub/cli/cli_utils.py b/metadata-ingestion/src/datahub/cli/cli_utils.py index 3795ed1ee8..a3e4636271 100644 --- a/metadata-ingestion/src/datahub/cli/cli_utils.py +++ b/metadata-ingestion/src/datahub/cli/cli_utils.py @@ -97,6 +97,17 @@ def guess_entity_type(urn: str) -> str: return urn.split(":")[2] +def get_token(): + _, gms_token_env = get_details_from_env() + if should_skip_config(): + gms_token = gms_token_env + else: + ensure_datahub_config() + _, gms_token_conf = get_details_from_config() + gms_token = first_non_null([gms_token_env, gms_token_conf]) + return gms_token + + def get_session_and_host(): session = requests.Session() diff --git a/metadata-ingestion/src/datahub/cli/delete_cli.py b/metadata-ingestion/src/datahub/cli/delete_cli.py index cb743932e2..d212e3aeb8 100644 --- a/metadata-ingestion/src/datahub/cli/delete_cli.py +++ b/metadata-ingestion/src/datahub/cli/delete_cli.py @@ -186,8 +186,10 @@ def delete_with_filters( ) -> DeletionResult: session, gms_host = cli_utils.get_session_and_host() + token = cli_utils.get_token() + logger.info(f"datahub configured with {gms_host}") - emitter = rest_emitter.DatahubRestEmitter(gms_server=gms_host) + emitter = rest_emitter.DatahubRestEmitter(gms_server=gms_host, token=token) batch_deletion_result = DeletionResult() urns = [ u @@ -238,7 +240,8 @@ def delete_one_urn( # Add removed aspect if not cached_emitter: _, gms_host = cli_utils.get_session_and_host() - emitter = rest_emitter.DatahubRestEmitter(gms_server=gms_host) + token = cli_utils.get_token() + emitter = rest_emitter.DatahubRestEmitter(gms_server=gms_host, token=token) else: emitter = cached_emitter if not dry_run: