Feature: match cluster submit exit code in cli (#478)

Azure · Apr 5, 2018 · 8889059 · 8889059
1 parent a59fe8b
commit 8889059
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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)