Allow passing custom tags to ignore to yaml parser

This commit is contained in:
Bryan Culbertson 2017-02-22 18:38:50 -08:00
parent e626cd5709
commit 529d30c847

View file

@ -13,9 +13,15 @@ except ImportError: # pragma: no cover (no libyaml-dev / pypy)
def check_yaml(argv=None):
parser = argparse.ArgumentParser()
parser.add_argument('--ignore-tags', type=lambda s: s.split(','), default=[],
help='Custom tags to ignore.')
parser.add_argument('filenames', nargs='*', help='Yaml filenames to check.')
args = parser.parse_args(argv)
# Ignore custom tags by returning None
for tag in args.ignore_tags:
Loader.add_constructor(tag, lambda *a, **k: None)
retval = 0
for filename in args.filenames:
try: