X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/4ccc8c86f5081990000c4dc09868d0bf040fb484..7fe19fac5bccc08dbcf21c293ab50f1fcda7716f:/plugin/black.vim diff --git a/plugin/black.vim b/plugin/black.vim index 4af044e..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), ]