Better message

Author: HyukjinKwon

python/pyspark/find_spark_home.py

@@ -40,7 +40,7 @@ def is_spark_home(path):
     paths = ["../", os.path.dirname(os.path.realpath(__file__))]
 
     # Add the path of the PySpark module if it exists
-    is_error_raised = False
+    import_error_raised = False
     if sys.version < "3":
         import imp
         try:
@@ -50,7 +50,7 @@ def is_spark_home(path):
             paths.append(os.path.join(module_home, "../../"))
         except ImportError:
             # Not pip installed no worries
-            is_error_raised = True
+            import_error_raised = True
     else:
         from importlib.util import find_spec
         try:
@@ -60,7 +60,7 @@ def is_spark_home(path):
             paths.append(os.path.join(module_home, "../../"))
         except ImportError:
             # Not pip installed no worries
-            is_error_raised = True
+            import_error_raised = True
 
     # Normalize the paths
     paths = [os.path.abspath(p) for p in paths]
@@ -69,15 +69,18 @@ def is_spark_home(path):
         return next(path for path in paths if is_spark_home(path))
     except StopIteration:
         print("Could not find valid SPARK_HOME while searching {0}".format(paths), file=sys.stderr)
-        if is_error_raised:
+        if import_error_raised:
             print(
-                "Did you install PySpark via a package manager such as PIP or Conda? If so, your "
-                "Python executable does not have the PySpark installed. It is possible your "
-                "Python executable does not point out your PIP. Please check your default "
-                "'python' and if you set PYSPARK_PYTHON and/or PYSPARK_DRIVER_PYTHON environment "
-                "variables, and see if you can import PySpark. If you cannot import, you can "
-                "install by using Python executable directly, for example, 'python -m pip "
-                "install pyspark'", file=sys.stderr)
+                "\nDid you install PySpark via a package manager such as PIP or Conda? If so,\n"
+                "your PySpark was not found in your Python executable. It is possible your\n"
+                "Python executable does not properly bind with your package manager.\n"
+                "\nPlease check your default 'python' and if you set PYSPARK_PYTHON and/or\n"
+                "PYSPARK_DRIVER_PYTHON environment variables, and see if you can import PySpark.\n"
+                "\nIf you cannot import, you can install by using Python executable directly,\n"
+                "for example, 'python -m pip install pyspark'. Otherwise, you can also\n"
+                "explicitly set your Python executable, that has PySpark installed, to\n"
+                "PYSPARK_PYTHON or PYSPARK_DRIVER_PYTHON environment variables, for example,\n"
+                "'PYSPARK_PYTHON=python3 pyspark'.\n", file=sys.stderr)
         sys.exit(-1)
 
 if __name__ == "__main__":