Fix an infinite loop when using `# fmt: on/off` ... (#3158)

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

diff --git a/autoload/black.vim b/autoload/black.vim
index 6c3bbfea81d873e5fd2f55ea8243fc416ef4def2..6c381b431a36ebb0d03f6d7792b1efe4937652dd 100644 (file)
--- a/autoload/black.vim
+++ b/autoload/black.vim
@@ -5,9 +5,9 @@ import sys
 import vim
 
 def strtobool(text):
-  if text.lower() in ['y', 'yes', 't', 'true' 'on', '1']:
+  if text.lower() in ['y', 'yes', 't', 'true', 'on', '1']:
     return True
-  if text.lower() in ['n', 'no', 'f', 'false' 'off', '0']:
+  if text.lower() in ['n', 'no', 'f', 'false', 'off', '0']:
     return False
   raise ValueError(f"{text} is not convertable to boolean")
 
@@ -29,6 +29,7 @@ FLAGS = [
   Flag(name="fast", cast=strtobool),
   Flag(name="skip_string_normalization", cast=strtobool),
   Flag(name="quiet", cast=strtobool),
+  Flag(name="skip_magic_trailing_comma", cast=strtobool),
 ]
 
 
@@ -143,6 +144,7 @@ def Black(**kwargs):
     line_length=configs["line_length"],
     string_normalization=not configs["skip_string_normalization"],
     is_pyi=vim.current.buffer.name.endswith('.pyi'),
+    magic_trailing_comma=not configs["skip_magic_trailing_comma"],
     **black_kwargs,
   )
   quiet = configs["quiet"]