[SNAPPYDATA] Use SnappyContext as default SQLContext on shell (#35)

Author: nthanvi

python/pyspark/shell.py

@@ -30,6 +30,7 @@
 import pyspark
 from pyspark.context import SparkContext
 from pyspark.sql import SparkSession, SQLContext
+from pyspark.sql.snappy import SnappyContext
 from pyspark.storagelevel import StorageLevel
 
 if os.environ.get("SPARK_EXECUTOR_URI"):
@@ -38,6 +39,8 @@
 SparkContext._ensure_initialized()
 
 try:
+    sqlContext = SnappyContext(sc)
+except py4j.protocol.Py4JError:
     # Try to access HiveConf, it will raise exception if Hive is not added
     SparkContext._jvm.org.apache.hadoop.hive.conf.HiveConf()
     spark = SparkSession.builder\