autoload/ale/fixers/yapf.vim

   1 " Author: w0rp <devw0rp@gmail.com>
   2 " Description: Fixing Python files with yapf.
   3
   4 call ale#Set('python_yapf_executable', 'yapf')
   5 call ale#Set('python_yapf_use_global', get(g:, 'ale_use_global_executables', 0))
   6 call ale#Set('python_yapf_auto_pipenv', 0)
   7 call ale#Set('python_yapf_auto_poetry', 0)
   8 call ale#Set('python_yapf_auto_uv', 0)
   9
  10 function! ale#fixers#yapf#GetExecutable(buffer) abort
  11     if (ale#Var(a:buffer, 'python_auto_pipenv') || ale#Var(a:buffer, 'python_yapf_auto_pipenv'))
  12     \ && ale#python#PipenvPresent(a:buffer)
  13         return 'pipenv'
  14     endif
  15
  16     if (ale#Var(a:buffer, 'python_auto_poetry') || ale#Var(a:buffer, 'python_yapf_auto_poetry'))
  17     \ && ale#python#PoetryPresent(a:buffer)
  18         return 'poetry'
  19     endif
  20
  21     if (ale#Var(a:buffer, 'python_auto_uv') || ale#Var(a:buffer, 'python_yapf_auto_uv'))
  22     \ && ale#python#UvPresent(a:buffer)
  23         return 'uv'
  24     endif
  25
  26     return ale#python#FindExecutable(a:buffer, 'python_yapf', ['yapf'])
  27 endfunction
  28
  29 function! ale#fixers#yapf#Fix(buffer) abort
  30     let l:executable = ale#fixers#yapf#GetExecutable(a:buffer)
  31
  32     let l:exec_args = l:executable =~? '\(pipenv\|poetry\|uv\)$'
  33     \   ? ' run yapf'
  34     \   : ''
  35
  36     let l:config = ale#path#FindNearestFile(a:buffer, '.style.yapf')
  37     let l:config_options = !empty(l:config)
  38     \   ? ' --no-local-style --style ' . ale#Escape(l:config)
  39     \   : ''
  40
  41     return {
  42     \   'command': ale#Escape(l:executable) . l:exec_args . l:config_options,
  43     \}
  44 endfunction