Merge pull request #193 from ericsagnes/fix/build-setup
[rainbowstream.git] / setup.py
index 31300ea..2b7b15e 100644 (file)
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@ else:
     from shlex import quote
 
 # Bumped version
-version = '1.3.4'
+version = '1.3.5'
 
 # Require
 install_requires = [
@@ -19,19 +19,21 @@ install_requires = [
     "pyfiglet",
     "twitter",
     "Pillow",
-    "PySocks"
+    "PySocks",
+    "pocket"
 ]
 
 # Default user (considers non virtualenv method)
-user = os.environ.get('SUDO_USER', os.environ['USER'])
+user = os.environ.get('SUDO_USER', os.environ.get('USER', None))
 
 # Copy default config if not exists
 default = os.path.expanduser("~") + os.sep + '.rainbow_config.json'
 if not os.path.isfile(default):
     cmd = 'cp rainbowstream/colorset/config ' + default
     os.system(cmd)
-    cmd = 'chown ' + quote(user) + ' ' + default
-    os.system(cmd)
+    if user:
+        cmd = 'chown ' + quote(user) + ' ' + default
+        os.system(cmd)
     cmd = 'chmod 777 ' + default
     os.system(cmd)