1 files changed, 57 insertions, 0 deletions
diff --git a/yamlbrowser b/yamlbrowser
new file mode 100755
index 0000000..464f6c6
--- /dev/null
+++ b/yamlbrowser
@@ -0,0 +1,57 @@
+#!/usr/bin/env python3
+
+import sys
+import yaml
+from qtpy import QtGui, QtWidgets, QtCore
+from qtpy.QtWidgets import QApplication, QTreeWidget, QTreeWidgetItem, QLabel
+
+def _make_children (parent:QTreeWidgetItem, d) -> QTreeWidgetItem:
+	"""
+	If the item is a container item, i.e. list or dict,
+	its contents are added as children to the parent.
+
+	If the item is not a container item,
+	it becomes the value to its parent.
+	"""
+	def transform_l (p:QTreeWidgetItem, d:list) -> QTreeWidgetItem:
+		p.addChildren (_make_children (QTreeWidgetItem (), x) for x in d)
+		return p
+	def transform_d (p:QTreeWidgetItem, d:dict) -> QTreeWidgetItem:
+		p.addChildren (_make_children (QTreeWidgetItem ([key]), value) for key, value in d.items ())
+		return p
+	return {
+		list:	lambda p, d : transform_l (p, d)
+	,	dict:	lambda p, d : transform_d (p, d)
+	,	str:	lambda p, d : QTreeWidgetItem ([p.text (0), d])
+	,	int:	lambda p, d : QTreeWidgetItem ([p.text (0), str(d)])
+	,	float:	lambda p, d : QTreeWidgetItem ([p.text (0), str(d)])
+	,	bool:	lambda p, d : QTreeWidgetItem ([p.text (0), str(d)])
+	}[type (d)] (parent, d)
+
+def main ():
+	app = QApplication (sys.argv)
+	# TODO: Check for suitable arguments first.
+	# Parameter may be a Qt option.
+	if len (sys.argv) == 1:
+		print ("Usage:", sys.argv[0], "[file …]", file=sys.stderr)
+		return
+	try:
+		items = []
+		for arg in sys.argv[1:]:
+			with open (arg) as file:
+				data = list (yaml.safe_load_all (file))
+			items.append (_make_children (QTreeWidgetItem ([arg]), data[0] if len (data) == 1 else data))
+
+		tree = QTreeWidget ()
+		tree.setColumnCount (2)
+		tree.setHeaderLabels (["Key", "Value"])
+		tree.insertTopLevelItems(0, items)
+		tree.show ()
+	except Exception as x:
+		print (x, file=sys.stderr)
+		label = QtWidgets.QLabel (str(x))
+		label.show ()	# Opens window.
+	sys.exit (app.exec_())	# Main loop.  Display the label.
+
+if __name__=='__main__':
+	main ()