summaryrefslogtreecommitdiffstats
path: root/bin/tests/system/digdelv/yamlget.py
diff options
context:
space:
mode:
Diffstat (limited to 'bin/tests/system/digdelv/yamlget.py')
-rw-r--r--bin/tests/system/digdelv/yamlget.py35
1 files changed, 35 insertions, 0 deletions
diff --git a/bin/tests/system/digdelv/yamlget.py b/bin/tests/system/digdelv/yamlget.py
new file mode 100644
index 0000000..afa582d
--- /dev/null
+++ b/bin/tests/system/digdelv/yamlget.py
@@ -0,0 +1,35 @@
+# Copyright (C) Internet Systems Consortium, Inc. ("ISC")
+#
+# SPDX-License-Identifier: MPL-2.0
+#
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, you can obtain one at https://mozilla.org/MPL/2.0/.
+#
+# See the COPYRIGHT file distributed with this work for additional
+# information regarding copyright ownership.
+
+import sys
+
+try:
+ import yaml
+# pylint: disable=bare-except
+except:
+ print("No python yaml module, skipping")
+ sys.exit(1)
+
+with open(sys.argv[1], "r", encoding="utf-8") as f:
+ for item in yaml.safe_load_all(f):
+ for key in sys.argv[2:]:
+ try:
+ key = int(key)
+ except ValueError:
+ pass
+
+ try:
+ item = item[key]
+ except KeyError:
+ print('Key "' + key + '" not found.')
+ sys.exit(1)
+
+ print(item)