diff --git a/gui/graphFrame.py b/gui/graphFrame.py index 4eceaba89..8321514fe 100644 --- a/gui/graphFrame.py +++ b/gui/graphFrame.py @@ -19,6 +19,7 @@ import os import logging +import imp import wx @@ -30,27 +31,18 @@ from gui.graph import Graph from gui.bitmapLoader import BitmapLoader from config import parsePath +# Don't actually import the thing, since it takes for fucking ever try: - import matplotlib as mpl - if mpl.__version__[0] >= "2": - mpl.use('wxagg') - mplImported = True - from matplotlib.patches import Patch - else: - mplImported = False - from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as Canvas - from matplotlib.figure import Figure + imp.find_module('matplotlib') graphFrame_enabled = True + mplImported = True except ImportError: - mpl = None - Canvas = None - Figure = None - mplImported = False graphFrame_enabled = False + mplImported = False + + logger = logging.getLogger(__name__) -if not graphFrame_enabled: - logger.info("Problems importing matplotlib; continuing without graphs") class GraphFrame(wx.Frame): @@ -59,8 +51,27 @@ class GraphFrame(wx.Frame): global graphFrame_enabled global mplImported + try: + import matplotlib as mpl + if mpl.__version__[0] >= "2": + mpl.use('wxagg') + mplImported = True + from matplotlib.patches import Patch + else: + mplImported = False + from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as Canvas + from matplotlib.figure import Figure + graphFrame_enabled = True + except ImportError: + mpl = None + Canvas = None + Figure = None + mplImported = False + graphFrame_enabled = False + self.legendFix = False if not graphFrame_enabled: + logger.info("Problems importing matplotlib; continuing without graphs") return try: