mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-05 11:36:54 +00:00
127 lines
3.9 KiB
Python
127 lines
3.9 KiB
Python
from __future__ import print_function
|
|
|
|
import argparse
|
|
import sys
|
|
|
|
import yaml
|
|
|
|
|
|
def _get_pretty_format(content, **kwargs):
|
|
return yaml.safe_dump(yaml.safe_load(content), **kwargs)
|
|
|
|
|
|
def _autofix(filename, new_contents):
|
|
print("Fixing file {0}".format(filename))
|
|
with open(filename, 'w') as f:
|
|
f.write(new_contents)
|
|
|
|
|
|
def filter_argument(accepted_values):
|
|
def _validate(s):
|
|
try:
|
|
return {str(v): v for v in accepted_values}[s]
|
|
except:
|
|
raise argparse.ArgumentTypeError('Accepted values are: {}'.format(accepted_values))
|
|
return _validate
|
|
|
|
|
|
def none_or_boolean_argument(s):
|
|
return filter_argument([None, True, False])(s)
|
|
|
|
|
|
def pretty_format_yaml(argv=None):
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument(
|
|
'--autofix',
|
|
action='store_true',
|
|
dest='autofix',
|
|
help='Automatically fixes encountered not-pretty-formatted files',
|
|
)
|
|
|
|
parser.add_argument(
|
|
'--default_style',
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--default_flow_style', type=none_or_boolean_argument,
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--canonical', type=none_or_boolean_argument,
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--indent', type=int,
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--width', type=int,
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--allow_unicode', type=none_or_boolean_argument,
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--line_break', type=none_or_boolean_argument,
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--encoding',
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--explicit_start', type=none_or_boolean_argument,
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--explicit_end', type=none_or_boolean_argument,
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--version',
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument(
|
|
'--tags',
|
|
help='PyYAML dump parameter. More info on http://pyyaml.org/wiki/PyYAMLDocumentation#Dumper'
|
|
)
|
|
parser.add_argument('filenames', nargs='*', help='Filenames to fix')
|
|
args = parser.parse_args(argv)
|
|
|
|
pyyaml_kwargs = {
|
|
key: value
|
|
for key, value in args._get_kwargs()
|
|
if key != 'autofix' and key != 'filenames'
|
|
}
|
|
|
|
status = 0
|
|
|
|
for yaml_file in args.filenames:
|
|
with open(yaml_file) as f:
|
|
contents = f.read()
|
|
|
|
try:
|
|
pretty_contents = _get_pretty_format(contents, **pyyaml_kwargs)
|
|
|
|
if contents != pretty_contents:
|
|
print("File {0} is not pretty-formatted".format(yaml_file))
|
|
|
|
if args.autofix:
|
|
_autofix(yaml_file, pretty_contents)
|
|
|
|
status = 1
|
|
|
|
except yaml.YAMLError:
|
|
print(
|
|
"Input File {0} is not a valid YAML, consider using check-yaml"
|
|
.format(yaml_file)
|
|
)
|
|
return 1
|
|
|
|
return status
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(pretty_format_yaml(sys.argv[1:]))
|