From 2be0e8bfa8e1efa77961cfcbefb4423172d6f556 Mon Sep 17 00:00:00 2001
From: Bryan Culbertson <bryan.culbertson@gmail.com>
Date: Wed, 22 Feb 2017 22:09:35 -0800
Subject: [PATCH] Pass tags to ignore to yaml parser (#1)

---
 pre_commit_hooks/check_yaml.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pre_commit_hooks/check_yaml.py b/pre_commit_hooks/check_yaml.py
index cc9a614..36fd128 100644
--- a/pre_commit_hooks/check_yaml.py
+++ b/pre_commit_hooks/check_yaml.py
@@ -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: