Merge pull request capstone-engine#406 from achernet/python-virtualen…

…v-fix-master

[master] Virtualenv support for Python binding

.gitignore

@@ -22,6 +22,7 @@
 
 # python
 bindings/python/build/
+bindings/python/capstone.egg-info/
 *.pyc
 
 # java

bindings/python/setup.py

@@ -6,7 +6,6 @@
 import stat
 import sys
 
-from site import getusersitepackages
 from distutils import log
 from distutils import dir_util
 from distutils.command.build_clib import build_clib
@@ -26,10 +25,14 @@
 VERSION = '3.0.3'
 SYSTEM = sys.platform
 
+# virtualenv breaks import, but get_python_lib() will work.
+SITE_PACKAGES = os.path.join(get_python_lib(), "capstone")
 if "--user" in sys.argv:
-    SITE_PACKAGES = os.path.join(getusersitepackages(), "capstone")
-else:
-    SITE_PACKAGES = os.path.join(get_python_lib(), "capstone")
+    try:
+        from site import getusersitepackages
+        SITE_PACKAGES = os.path.join(getusersitepackages(), "capstone")
+    except ImportError:
+        pass
 
 
 SETUP_DATA_FILES = []