Linux: better display management

BoboTiG · BoboTiG · commit 51a02223f8ff · 2016-10-07T13:52:13.000+02:00
diff --git a/mss/linux.py b/mss/linux.py
@@ -9,7 +9,7 @@
     c_uint, c_uint32, c_ulong, c_ushort, c_void_p, cast, cdll)
 from ctypes.util import find_library
 from os import environ
-from sys import maxsize, version
+from sys import maxsize
 
 from .base import MSSBase
 from .exception import ScreenshotError
@@ -101,14 +101,11 @@ def __init__(self, display=None):
 
         if not display:
             try:
-                if version > '3':
-                    display = bytes(environ['DISPLAY'], 'utf-8')
-                else:
-                    display = environ['DISPLAY']
+                display = environ['DISPLAY']
             except KeyError:
                 err = '$DISPLAY not set. Stopping to prevent segfault.'
                 raise ScreenshotError(err)
-        elif not isinstance(display, bytes):
+        if not isinstance(display, bytes):
             display = bytes(display, 'utf-8')
 
         x11 = find_library('X11')
