Use the CLoader when present to validate yaml
diff --git a/pre_commit_hooks/check_yaml.py b/pre_commit_hooks/check_yaml.py
index 0085807..293352f 100644
--- a/pre_commit_hooks/check_yaml.py
+++ b/pre_commit_hooks/check_yaml.py
@@ -6,6 +6,12 @@
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):
parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*', help='Yaml filenames to check.')
@@ -14,7 +20,7 @@
retval = 0
for filename in args.filenames:
try:
- yaml.load(open(filename))
+ yaml.load(open(filename), Loader=Loader)
except yaml.YAMLError as exc:
print(exc)
retval = 1
