notify: fix os.environ handling

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

Showing with 5 additions and 1 deletions

notify.py
+5 -1

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:
-        os.environ["DISPLAY"] = olddisplay
+        if olddisplay:
+            os.environ["DISPLAY"] = olddisplay
+        elif 'DISPLAY' in os.environ:
+            del os.environ["DISPLAY"]
+


 def file_already_exists(name, mode=0644):