From 8889059aad1b3e2cf55971075b818eac2133e6ef Mon Sep 17 00:00:00 2001 From: Jacob Freck Date: Thu, 5 Apr 2018 11:54:25 -0700 Subject: [PATCH] Feature: match cluster submit exit code in cli (#478) --- aztk_cli/spark/endpoints/cluster/cluster_submit.py | 5 ++++- aztk_cli/utils.py | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/aztk_cli/spark/endpoints/cluster/cluster_submit.py b/aztk_cli/spark/endpoints/cluster/cluster_submit.py index 00f14bef..b8d7177b 100644 --- a/aztk_cli/spark/endpoints/cluster/cluster_submit.py +++ b/aztk_cli/spark/endpoints/cluster/cluster_submit.py @@ -1,4 +1,5 @@ import argparse +import sys import typing from aztk_cli import utils, config, log import aztk.spark @@ -140,4 +141,6 @@ def execute(args: typing.NamedTuple): ) if args.wait: - utils.stream_logs(client=spark_client, cluster_id=args.cluster_id, application_name=args.name) + exit_code = utils.stream_logs(client=spark_client, cluster_id=args.cluster_id, application_name=args.name) + sys.exit(exit_code) + diff --git a/aztk_cli/utils.py b/aztk_cli/utils.py index af6e2d19..f39d837e 100644 --- a/aztk_cli/utils.py +++ b/aztk_cli/utils.py @@ -121,7 +121,7 @@ def stream_logs(client, cluster_id, application_name): current_bytes=current_bytes) print(app_logs.log, end="") if app_logs.application_state == 'completed': - break + return app_logs.exit_code current_bytes = app_logs.total_bytes time.sleep(3)