mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-03-13 07:41:19 +08:00
Use the CLoader when present to validate yaml
This commit is contained in:
@@ -6,6 +6,12 @@ import sys
|
||||
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 @@ def check_yaml(argv=None):
|
||||
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
|
||||
|
||||
Reference in New Issue
Block a user