$DISPLAY now defaults to ":0", useful for eg Xming

VIDA-NYU · Jan 20, 2015 · b187603 · b187603
1 parent 0214acd
commit b187603
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/reprounzip-vagrant/reprounzip/unpackers/vagrant/__init__.py b/reprounzip-vagrant/reprounzip/unpackers/vagrant/__init__.py
@@ -432,7 +432,9 @@ def __init__(self, enabled, display=None):
                       self.display)
 
         # Locate local display
-        local_addr, local_display = os.environ['DISPLAY'].rsplit(':', 1)
+        # We default to ":0", Xming for instance doesn't set $DISPLAY
+        local_addr, local_display = os.environ.get(
+                'DISPLAY', ':0').rsplit(':', 1)
         local_display = int(local_display.split('.', 1)[0])
         if ('XAUTHORITY' in os.environ and
                 Path(os.environ['XAUTHORITY']).is_file()):