Merge pull request #57 from maxnoe/fix_qt_import

Fix qt import, fixes #55

Author: tacaswell

setup.py

@@ -27,6 +27,6 @@
         "Programming Language :: Python :: 3",
         ],
     packages=find_packages(),
-    install_requires=["numpy", "matplotlib", "colorspacious"],
+    install_requires=["numpy", "matplotlib", "colorspacious", "scipy"],
     package_data={'viscm': ['examples/*']},
 )

viscm/gui.py

@@ -16,14 +16,14 @@
 # Do this first before any other matplotlib imports, to force matplotlib to
 # use a Qt backend
 from matplotlib.backends.qt_compat import QtWidgets, QtCore, QtGui, _getSaveFileName
-from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas4
-from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas5
 
-def FigureCanvas(fig):
+try:
+    from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg as FigureCanvas
+except ImportError:
     try:
-        return FigureCanvas5(fig)
-    except Exception:
-        return FigureCanvas4(fig)
+        from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
+    except ImportError:
+        from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
 
 import matplotlib
 import matplotlib.pyplot as plt