Using a regex to check DISPLAY

jschueller · jschueller · commit 48e5bb18b9fb · 2013-04-17T19:00:03.000+02:00
diff --git a/lib/matplotlib/backends/backend_qt4.py b/lib/matplotlib/backends/backend_qt4.py
@@ -1,6 +1,7 @@
 from __future__ import division, print_function
 import math
 import os
+import re
 import signal
 import sys
 
@@ -57,7 +58,7 @@ def _create_qApp():
             # check for DISPLAY env variable on X11 build of Qt
             if hasattr(QtGui, "QX11Info"):
                 display = os.environ.get('DISPLAY')
-                if (display is None) or (not ':' in display):
+                if display is None or not re.search(':\d', display):
                     raise RuntimeError('Invalid DISPLAY variable')
         
             qApp = QtGui.QApplication( [" "] )
