[SPARK-2435] Add shutdown hook to pyspark

Author: Matthew Farrellee <[email protected]> Closes #2183 from mattf/SPARK-2435 and squashes the following commits: ee0ee99 [Matthew Farrellee] [SPARK-2435] Add shutdown hook to pyspark
kiszk · Sep 4, 2014 · 7c6e71f · 7c6e71f
1 parent c5cbc49
commit 7c6e71f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/pyspark/shell.py b/python/pyspark/shell.py
@@ -28,6 +28,7 @@
     sys.exit(1)
 
 
+import atexit
 import os
 import platform
 import pyspark
@@ -42,6 +43,7 @@
     SparkContext.setSystemProperty("spark.executor.uri", os.environ["SPARK_EXECUTOR_URI"])
 
 sc = SparkContext(appName="PySparkShell", pyFiles=add_files)
+atexit.register(lambda: sc.stop())
 
 print("""Welcome to
       ____              __