Linux: fix segfault du to double call to _cleanup() which called twice xlib.XCloseDisplay() on an already closed display

BoboTiG · BoboTiG · commit 0587ec8015ba · 2016-04-27T11:48:32.000+02:00
diff --git a/mss/linux.py b/mss/linux.py
@@ -78,11 +78,6 @@ def __del__(self):
 
         self._cleanup()
 
-    def __exit__(self, exc_type, exc_value, traceback):
-        ''' Quit. '''
-
-        self._cleanup()
-
     def __init__(self):
         ''' GNU/Linux initialisations. '''
 
@@ -134,6 +129,7 @@ def _cleanup(self):
         try:
             if self.display:
                 self.xlib.XCloseDisplay(self.display)
+                self.display = None
         except AttributeError:
             pass
 
