Merge pull request python-diamond#650 from bhepple/patch-1

Support python26 - os.initgroups()
jhannus · May 16, 2017 · f2bece4 · f2bece4
2 parents 71e683c + ad3862d
commit f2bece4
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