pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-04-02 02:56:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add an --allow-multiple-documents option to check-yaml

Browse source
This commit is contained in:
Anthony Sottile 2017-10-12 15:47:20 -07:00
parent e09278e806
commit e87b81afd9
3 changed files with 32 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
pre_commit_hooks/check_yaml.py
View file

@ -11,15 +11,24 @@ except ImportError: # pragma: no cover (no libyaml-dev / pypy)
Loader = yaml.SafeLoader
def _load_all(*args, **kwargs):
# need to exhaust the generator
return tuple(yaml.load_all(*args, **kwargs))
def check_yaml(argv=None):
parser = argparse.ArgumentParser()
parser.add_argument(
'-m', '--allow-multiple-documents', dest='yaml_load_fn',
action='store_const', const=_load_all, default=yaml.load,
)
parser.add_argument('filenames', nargs='*', help='Yaml filenames to check.')
args = parser.parse_args(argv)
retval = 0
for filename in args.filenames:
try:
yaml.load(open(filename), Loader=Loader)
args.yaml_load_fn(open(filename), Loader=Loader)
except yaml.YAMLError as exc:
print(exc)
retval = 1