X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=.vim%2Fpython%2Fvirtualenv.py;h=62316009e94bfe707b5cdc5cf6b465283375ab03;hb=refs%2Fheads%2Fmaster;hp=3b76fcd46adb0d3bb3b0f3ab2b1c68bc7cd321fe;hpb=c1fb9100f55005a4d854b99cee25d4ef056569ce;p=dotfiles.git

diff --git a/.vim/python/virtualenv.py b/.vim/python/virtualenv.py
index 3b76fcd..6231600 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, 'rU').read(), dict(__file__=activate_this))
+    if (
+        os.path.exists(
+            os.path.join(
+                virtualenv_dir, 'lib',
+                'python%d.%d' % (sys.version_info[0], sys.version_info[1])
+            )
+        )
+    ):
+        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