update error messages

sdpython · Aug 5, 2016 · 75b7bfd · 75b7bfd
1 parent 5e5c43a
commit 75b7bfd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/pymyinstall/cli/pymy_install.py b/src/pymyinstall/cli/pymy_install.py
@@ -115,7 +115,11 @@ def do_main(temp_folder="build/update_modules",
         pfolder = os.path.normpath(os.path.join(
             os.path.abspath(os.path.dirname(__file__)), "..", ".."))
         sys.path.append(pfolder)
-        from pymyinstall import is_travis_or_appveyor
+        try:
+            from pymyinstall import is_travis_or_appveyor
+        except ImportError as e:
+            mes = "Unable to import is_travis_or_appveyor from '{0}' after adding '{1}'.".format(__file__, pfolder)
+            raise ImportError(mes)
     if is_travis_or_appveyor() and source is None:
         source = "2"
 

diff --git a/src/pymyinstall/cli/pymy_update.py b/src/pymyinstall/cli/pymy_update.py
@@ -70,7 +70,11 @@ def do_main(temp_folder="build/update_modules",
         pfolder = os.path.normpath(os.path.join(
             os.path.abspath(os.path.dirname(__file__)), "..", ".."))
         sys.path.append(pfolder)
-        from pymyinstall import is_travis_or_appveyor
+        try:
+            from pymyinstall import is_travis_or_appveyor
+        except ImportError as e:
+            mes = "Unable to import is_travis_or_appveyor from '{0}' after adding '{1}'.".format(__file__, pfolder)
+            raise ImportError(mes)
     if is_travis_or_appveyor() and source is None:
         source = "2"