Isolate command line tests for notebooks from user-level config (#2854)

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

diff --git a/plugin/black.vim b/plugin/black.vim
index 3bdca62e6ad990650d3f8576948c0e39e9c6d1dd..dbe236b5f34208e436c92540eb964eedb3579652 100644 (file)
--- a/plugin/black.vim
+++ b/plugin/black.vim
@@ -50,11 +50,31 @@ if !exists("g:black_skip_string_normalization")
     let g:black_skip_string_normalization = 0
   endif
 endif
     let g:black_skip_string_normalization = 0
   endif
 endif

+if !exists("g:black_skip_magic_trailing_comma")
+  if exists("g:black_magic_trailing_comma")
+    let g:black_skip_magic_trailing_comma = !g:black_magic_trailing_comma
+  else
+    let g:black_skip_magic_trailing_comma = 0
+  endif
+endif

 if !exists("g:black_quiet")
   let g:black_quiet = 0
 endif
 if !exists("g:black_quiet")
   let g:black_quiet = 0
 endif

+if !exists("g:black_target_version")
+  let g:black_target_version = ""
+endif

 
 

+function BlackComplete(ArgLead, CmdLine, CursorPos)
+  return [
+\    'target_version=py27',
+\    'target_version=py36',
+\    'target_version=py37',
+\    'target_version=py38',
+\    'target_version=py39',
+\    'target_version=py310',
+\  ]
+endfunction

 
 

-command! Black :call black#Black()
+command! -nargs=* -complete=customlist,BlackComplete Black :call black#Black(<f-args>)

 command! BlackUpgrade :call black#BlackUpgrade()
 command! BlackVersion :call black#BlackVersion()
 command! BlackUpgrade :call black#BlackUpgrade()
 command! BlackVersion :call black#BlackVersion()