X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/61f04e658bcc4af228f24bd42a716804ccaaae27..7bca930ca3d84bbd01e98937b6b8a493d0254c7c:/plugin/black.vim diff --git a/plugin/black.vim b/plugin/black.vim index a0de953..3dd3f21 100644 --- a/plugin/black.vim +++ b/plugin/black.vim @@ -41,8 +41,12 @@ endif if !exists("g:black_linelength") let g:black_linelength = 88 endif -if !exists("g:black_skip_string_normalization") - let g:black_skip_string_normalization = 0 +if !exists("g:black_string_normalization") + 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 python3 << EndPython3 @@ -50,6 +54,7 @@ import collections import os import sys import vim +from distutils.util import strtobool class Flag(collections.namedtuple("FlagBase", "name, cast")): @@ -62,15 +67,13 @@ class Flag(collections.namedtuple("FlagBase", "name, cast")): name = self.var_name if name == "line_length": name = name.replace("_", "") - if name == "string_normalization": - name = "skip_" + name return "g:black_" + name FLAGS = [ Flag(name="line_length", cast=int), - Flag(name="fast", cast=bool), - Flag(name="string_normalization", cast=bool), + Flag(name="fast", cast=strtobool), + Flag(name="string_normalization", cast=strtobool), ] @@ -190,7 +193,7 @@ def get_configs(): toml_config = {} return { - flag.var_name: toml_config.get(flag.name, flag.cast(vim.eval(flag.vim_rc_name))) + flag.var_name: flag.cast(toml_config.get(flag.name, vim.eval(flag.vim_rc_name))) for flag in FLAGS }