Fix g:black_fast and g:black_(skip)_string_normalization opts

This commit is contained in:
Rowan Rodrik van der Molen 2020-07-11 21:20:31 +02:00 committed by Łukasz Langa
parent e268cbaf66
commit 5f78401b28

View File

@ -41,8 +41,12 @@ endif
if !exists("g:black_linelength") if !exists("g:black_linelength")
let g:black_linelength = 88 let g:black_linelength = 88
endif endif
if !exists("g:black_skip_string_normalization") if !exists("g:black_string_normalization")
let g:black_skip_string_normalization = 0 if exists("g:black_skip_string_normalization")
let g:black_string_normalization = !g:black_skip_string_normalization
else
let g:black_string_normalization = 1
endif
endif endif
python3 << EndPython3 python3 << EndPython3
@ -50,6 +54,7 @@ import collections
import os import os
import sys import sys
import vim import vim
from distutils.util import strtobool
class Flag(collections.namedtuple("FlagBase", "name, cast")): class Flag(collections.namedtuple("FlagBase", "name, cast")):
@ -62,15 +67,13 @@ class Flag(collections.namedtuple("FlagBase", "name, cast")):
name = self.var_name name = self.var_name
if name == "line_length": if name == "line_length":
name = name.replace("_", "") name = name.replace("_", "")
if name == "string_normalization":
name = "skip_" + name
return "g:black_" + name return "g:black_" + name
FLAGS = [ FLAGS = [
Flag(name="line_length", cast=int), Flag(name="line_length", cast=int),
Flag(name="fast", cast=bool), Flag(name="fast", cast=strtobool),
Flag(name="string_normalization", cast=bool), Flag(name="string_normalization", cast=strtobool),
] ]