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

Use the CLoader when present to validate yaml

Browse source
This commit is contained in:
Anthony Sottile 2015-01-15 09:44:48 -08:00
parent 5541a6a046
commit e1c091a4e3
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
pre_commit_hooks/check_yaml.py
View file

@ -6,6 +6,12 @@ import sys
import yaml import yaml
try:
from yaml.cyaml import CLoader as Loader
except ImportError: # pragma: no cover (no libyaml-dev / pypy)
Loader = yaml.Loader
def check_yaml(argv=None): def check_yaml(argv=None):
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*', help='Yaml filenames to check.') parser.add_argument('filenames', nargs='*', help='Yaml filenames to check.')
@ -14,7 +20,7 @@ def check_yaml(argv=None):
retval = 0 retval = 0
for filename in args.filenames: for filename in args.filenames:
try: try:
yaml.load(open(filename)) yaml.load(open(filename), Loader=Loader)
except yaml.YAMLError as exc: except yaml.YAMLError as exc:
print(exc) print(exc)
retval = 1 retval = 1