From df03816b8e2ddec3942b7576865e3b1b96a539ce Mon Sep 17 00:00:00 2001 From: JensDiemer Date: Fri, 24 Jul 2015 17:04:39 +0200 Subject: [PATCH] Quick fix for #7 TODO: move the command into seperate .py file (IMHO) --- setup.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index d9ccded0..fa5a2db5 100644 --- a/setup.py +++ b/setup.py @@ -1,8 +1,15 @@ -import os, glob, shutil, sys, pwd, grp +import os, glob, shutil, sys from setuptools import setup from distutils.core import setup, Command +try: + import pwd, grp +except ImportError: + # e.g. under windows + pwd = grp = None + + def read(fname): """Read file contents.""" return open(os.path.join(os.path.dirname(__file__), fname)).read() @@ -97,10 +104,10 @@ def run(self): shutil.copy(template, drqueue_etc) # set to user-supplied user / group - if self.owner != None: + if self.owner != None and pwd is not None: uid = pwd.getpwnam(self.owner)[2] recursive_chown(drqueue_root, uid, -1) - if self.group != None: + if self.group != None and grp is not None: gid = grp.getgrnam(self.group)[2] recursive_chown(drqueue_root, -1, gid)