Fix formatting examples/drop_privileges.py

tuomasjjrasanen · tuomasjjrasanen · commit 9250c0fe2cb4 · 2016-04-04T22:03:22.000+03:00
diff --git a/examples/drop_privileges.py b/examples/drop_privileges.py
@@ -1,21 +1,26 @@
+import grp
+import os
+import pwd
+
 import uinput
 
 def drop_privileges(uid_name='nobody', gid_name='nogroup'):
-    #https://stackoverflow.com/questions/2699907/dropping-root-permissions-in-python
-    import os, pwd, grp
+    # https://stackoverflow.com/questions/2699907/dropping-root-permissions-in-python
+
     if os.getuid() != 0:
         return
+
     running_uid = pwd.getpwnam(uid_name).pw_uid
     running_gid = grp.getgrnam(gid_name).gr_gid
+
     os.setgroups([])
     os.setgid(running_gid)
     os.setuid(running_uid)
     old_umask = os.umask(077)
 
-
 def main():
-    uinput_fd= uinput.Device.create_uinput_fd()
-    drop_privileges() #no need to be root beyond this line
+    uinput_fd = uinput.fdopen()
+    drop_privileges() # No need to be root beyond this line.
 
     events = (uinput.KEY_A,)
     with uinput.Device(events, fd=uinput_fd) as device:
