Merge pull request #120 from sanmai-NL/JSON_arbitrary_indentation_separator

Refactor legacy `indent: int` -> `indent: str`
This commit is contained in:
Anthony Sottile 2016-06-12 10:25:59 -07:00 committed by GitHub
commit cf3aabe9b1
3 changed files with 53 additions and 3 deletions

View file

@ -24,6 +24,25 @@ def _autofix(filename, new_contents):
f.write(new_contents)
def parse_indent(s):
# type: (str) -> str
try:
int_indentation_spec = int(s)
if int_indentation_spec >= 0:
return int_indentation_spec * ' '
else:
raise ValueError(
'Negative integer supplied to construct JSON indentation delimiter. ',
)
except ValueError:
if s.strip() == '':
return s
else:
raise ValueError(
'Non-whitespace JSON indentation delimiter supplied. ',
)
def pretty_format_json(argv=None):
parser = argparse.ArgumentParser()
parser.add_argument(
@ -34,9 +53,9 @@ def pretty_format_json(argv=None):
)
parser.add_argument(
'--indent',
type=int,
default=2,
help='Number of indent spaces used to pretty-format files',
type=parse_indent,
default=' ',
help='String used as delimiter for one indentation level',
)
parser.add_argument(
'--no-sort-keys',