Fix tests on windows (#159)

[etc/vim.git] / vim / plugin / black.vim

diff --git a/vim/plugin/black.vim b/vim/plugin/black.vim
index cd80266a01b111dfb2f9aff59206919645d8f44f..9cf5f2dc69bb9fa8261a941591c04362a9058e5a 100644 (file)
--- a/vim/plugin/black.vim
+++ b/vim/plugin/black.vim
@@ -35,20 +35,20 @@ python3 << endpython3
 import sys
 import vim
 
 import sys
 import vim
 

-def _find_python_binary(exec_prefix):
+def _get_python_binary(exec_prefix):

   if sys.platform[:3] == "win":
     return exec_prefix / 'python.exe'
   return exec_prefix / 'bin' / 'python3'
 
   if sys.platform[:3] == "win":
     return exec_prefix / 'python.exe'
   return exec_prefix / 'bin' / 'python3'
 

-def _find_pip(venv_path):
+def _get_pip(venv_path):

   if sys.platform[:3] == "win":
     return venv_path / 'Scripts' / 'pip.exe'
   return venv_path / 'bin' / 'pip'
 
   if sys.platform[:3] == "win":
     return venv_path / 'Scripts' / 'pip.exe'
   return venv_path / 'bin' / 'pip'
 

-def _find_virtualenv_site_packages(venv_path):
+def _get_virtualenv_site_packages(venv_path, pyver):

   if sys.platform[:3] == "win":
     return venv_path / 'Lib' / 'site-packages'
   if sys.platform[:3] == "win":
     return venv_path / 'Lib' / 'site-packages'

-  return venv_path / 'Lib' / f'python{pyver[0]}.{pyver[1]}' / 'site-packages'
+  return venv_path / 'lib' / f'python{pyver[0]}.{pyver[1]}' / 'site-packages'

 
 def _initialize_black_env(upgrade=False):
   pyver = sys.version_info[:2]
 
 def _initialize_black_env(upgrade=False):
   pyver = sys.version_info[:2]


@@ -60,13 +60,13 @@ def _initialize_black_env(upgrade=False):
   import subprocess
   import venv
   virtualenv_path = Path(vim.eval("g:black_virtualenv")).expanduser()
   import subprocess
   import venv
   virtualenv_path = Path(vim.eval("g:black_virtualenv")).expanduser()

-  virtualenv_site_packages = str(_find_virtualenv_site_packages(virtualenv_path))
+  virtualenv_site_packages = str(_get_virtualenv_site_packages(virtualenv_path, pyver))

   first_install = False
   if not virtualenv_path.is_dir():
     print('Please wait, one time setup for Black.')
     _executable = sys.executable
     try:
   first_install = False
   if not virtualenv_path.is_dir():
     print('Please wait, one time setup for Black.')
     _executable = sys.executable
     try:

-      sys.executable = str(_find_python_binary(Path(sys.exec_prefix)))
+      sys.executable = str(_get_python_binary(Path(sys.exec_prefix)))

       print(f'Creating a virtualenv in {virtualenv_path}...')
       print('(this path can be customized in .vimrc by setting g:black_virtualenv)')
       venv.create(virtualenv_path, with_pip=True)
       print(f'Creating a virtualenv in {virtualenv_path}...')
       print('(this path can be customized in .vimrc by setting g:black_virtualenv)')
       venv.create(virtualenv_path, with_pip=True)


@@ -78,7 +78,7 @@ def _initialize_black_env(upgrade=False):
   if upgrade:
     print('Upgrading Black with pip...')
   if first_install or upgrade:
   if upgrade:
     print('Upgrading Black with pip...')
   if first_install or upgrade:

-    subprocess.run([str(_find_pip(virtualenv_path)), 'install', '-U', 'black'])
+    subprocess.run([str(_get_pip(virtualenv_path)), 'install', '-U', 'black'])

     print('DONE! You are all set, thanks for waiting ✨ 🍰 ✨')
   if first_install:
     print('Pro-tip: to upgrade Black in the future, use the :BlackUpgrade command and restart Vim.\n')
     print('DONE! You are all set, thanks for waiting ✨ 🍰 ✨')
   if first_install:
     print('Pro-tip: to upgrade Black in the future, use the :BlackUpgrade command and restart Vim.\n')