Support python26 - os.initgroups()

os.initgroups() is new in python27 see python-diamond#638
jhannus · May 16, 2017 · ad3862d · ad3862d
1 parent 56d0cd1
commit ad3862d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/bin/diamond b/bin/diamond
@@ -201,7 +201,14 @@ def main():
             try:
                 if gid != -1 and uid != -1:
                     # Manually set the groups since they aren't set by default
-                    os.initgroups(pwd.getpwuid(uid).pw_name, gid)
+
+                    # Python 2.7+
+                    if hasattr(os, 'initgroups'):
+                        os.initgroups(user, gid)
+                    # Python 2.6
+                    else:
+                        os.setgroups([e.gr_gid for e in grp.getgrall()
+                                      if user in e.gr_mem] + [gid])
 
                 if gid != -1 and os.getgid() != gid:
                     # Set GID