Re-worked exception handling to avoid masking shared library dependency

jaytmiller · jaytmiller · commit 32b1d0bd7cc6 · 2004-10-19T14:46:36.000Z
exceptions.

svn path=/trunk/matplotlib/; revision=600
diff --git a/lib/matplotlib/_image.py b/lib/matplotlib/_image.py
@@ -4,15 +4,11 @@
     try:
         from matplotlib._na_image import *
     except ImportError:
-        print "numarray wasn't available at compile time so the numarray"
-        print "extensions were not built.  Please make sure you have a working"
-        print "numarray installation and then rebuild matplotlib."
-        sys.exit(1)
+        numerix._import_fail_message("_image", "_na")
+        raise
 else:
     try:
         from matplotlib._nc_image import *
     except ImportError:
-        print "Numeric wasn't available at compile time so the Numeric"
-        print "extensions were not built.  Please make sure you have a working"
-        print "Numeric installation and then rebuild matplotlib."
-        sys.exit(1)
+        numerix._import_fail_message("_image", "_nc")
+        raise
diff --git a/lib/matplotlib/_transforms.py b/lib/matplotlib/_transforms.py
@@ -4,15 +4,11 @@
     try:
         from matplotlib._na_transforms import *
     except ImportError:
-        print "numarray wasn't available at compile time so the numarray"
-        print "extensions were not built.  Please make sure you have a working"
-        print "numarray installation and then rebuild matplotlib."
-        sys.exit(1)
+        numerix._import_fail_message("_transforms", "_na")
+        raise
 else:
     try:
         from matplotlib._nc_transforms import *
     except ImportError:
-        print "Numeric wasn't available at compile time so the Numeric"
-        print "extensions were not built.  Please make sure you have a working"
-        print "Numeric installation and then rebuild matplotlib."
-        sys.exit(1)
+        numerix._import_fail_message("_transforms", "_nc")
+        raise
diff --git a/lib/matplotlib/numerix.py b/lib/matplotlib/numerix.py
@@ -58,3 +58,14 @@
 verbose.report('numerix %s'%version)
 # a bug fix for blas numeric suggested by Fernando Perez
 matrixmultiply=dot
+
+
+def _import_fail_message(module, version):
+    """Prints a message when the array package specific version of an extension
+    fails to import correctly.
+    """
+    _dict = { "which" : which[0],
+              "module" : module,
+              "specific" : version + module
+              }
+    print """\nThe import of the %(which)s version of the %(module)s module, %(specific)s, failed.\nThis is either because %(which)s was unavailable when matplotlib was compiled,\nor because a dependency of %(specific)s could not be satisfied.\nIf it appears that %(specific)s was not built,  make sure you have a working copy of\n%(which)s and then re-install matplotlib. Otherwise, the following traceback gives more details:\n""" % _dict
