Update script.py

This should fix #27
bilelmoussaoui · Jun 9, 2015 · c478f4d · c478f4d
1 parent 74507ba
commit c478f4d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/script.py b/script.py
@@ -14,7 +14,10 @@
 script_folder = "scripts"
 db_ext = ".txt"
 userhome = os.path.expanduser("~")
-username = pwd.getpwuid( os.getuid() )[ 0 ]
+if os.getuid() != 0:
+    username = pwd.getpwuid( os.getuid() )[ 0 ]
+else:
+    username = os.getenv("SUDO_USER")
 useros = platform.linux_distribution()
 useros = useros[0].strip('"')
 theme = Gtk.IconTheme.get_default()