add support of pip --user option

gem-pasteur · Nov 8, 2016 · 7d1bf91 · 7d1bf91
1 parent 25a8881
commit 7d1bf91
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -85,13 +85,17 @@ def __init__(self, attrs=None):
 
 def get_install_data_dir(inst):
     """
-    :param inst: iinstallation option
+    :param inst: installation option
     :type inst: dict
     :return: the prefix where to install data
     :rtype: string
     """
+
     if 'VIRTUAL_ENV' in os.environ:
         inst['prefix'] = ('environment', os.environ['VIRTUAL_ENV'])
+    elif 'user' in inst:
+        import site
+        inst['prefix'] = ('command line', site.USER_BASE)
 
     if 'install_data' in inst:
         install_dir = inst['install_data'][1]