notify: fix os.environ handling

6f8ffbbc · Őry Máté · d24ba1e7 · 6f8ffbbc
Commit 6f8ffbbc authored Jul 27, 2014 by Őry Máté
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

notify.py
+4 -0

No files found.
--- a/notify.py
+++ b/notify.py
@@ -104,7 +104,11 @@ def notify(url):
        if display:
            webbrowser.open(url, new=2, autoraise=True)
    finally:
+        if olddisplay:
            os.environ["DISPLAY"] = olddisplay
+        elif 'DISPLAY' in os.environ:
+            del os.environ["DISPLAY"]
+


 def file_already_exists(name, mode=0644):