X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=.vim%2Fpython%2Fvirtualenv.py;h=ed46739fb6774130b760abc3caef0a0017d711a5;hb=HEAD;hp=078a0f8c54d226df0b28035a36a4bc140e38f930;hpb=6bb64389fb09a9728a84b3d35a27e530e58cabbe;p=dotfiles.git

diff --git a/.vim/python/virtualenv.py b/.vim/python/virtualenv.py
index 078a0f8..ed46739 100644
--- a/.vim/python/virtualenv.py
+++ b/.vim/python/virtualenv.py
@@ -1,10 +1,25 @@
-import sys, os
+import sys, os  # noqa: E401 multiple imports on one line
 
 virtualenv_dir = os.environ.get('VIRTUAL_ENV')
 if virtualenv_dir:
-    sys.path.insert(0, virtualenv_dir)
-    activate_this = os.path.join(virtualenv_dir, 'bin', 'activate_this.py')
-    if sys.version_info[0] == 2:
-        execfile(activate_this, dict(__file__=activate_this))
-    else:
-        exec(open(activate_this, 'r').read(), dict(__file__=activate_this))
+    if (
+        (sys.version_info[0] == 2)
+            and os.path.exists(
+                os.path.join(virtualenv_dir, 'lib', 'python2.7'))
+    ) or (
+        not os.path.exists(
+                os.path.join(virtualenv_dir, 'lib', 'python2.7'))
+    ):
+        for activate_this in [
+            os.path.join(virtualenv_dir, 'bin', 'activate_this.py'),
+            os.path.join(virtualenv_dir, 'Scripts', 'activate_this.py')
+        ]:
+            if not os.path.exists(activate_this):
+                continue
+            if sys.version_info[0] == 2:
+                execfile(activate_this,  # noqa: F821 undefined name 'execfile'
+                         dict(__file__=activate_this))
+            else:
+                exec(open(activate_this, 'r').read(),
+                     dict(__file__=activate_this))
+            break