consolerobot.py

from termcolor import colored
import xml.etree.ElementTree as ET

#TODO: 'output.xml' is the default filename
# robot writes. How about give it as a cmd line option?
root = ET.parse('output.xml').getroot()

#Indent of output
indentstr="  "
colours = {
	'PASS': 'green',
	'FAIL': 'red',
	'Arg' : 'cyan',
	'Doc' : 'yellow',
	'Msg' : 'cyan'
}

def process_kw(keyword, indent):
	status = keyword.find("status")
	statusstr = status.attrib.get("status")
	if( statusstr == "FAIL" ):
		colour = colours['FAIL']
	else:
		colour = colours['PASS']
	print(indent + colored(statusstr, colour) +": " + keyword.attrib.get("name"))

	argstr=""
	for arg in keyword.findall("arguments"):
		for a in arg.findall("arg"):
			argstr = argstr + a.text + colored("|", colours['Arg'])
	if argstr:
		print(indent + indentstr + colored("Args", colours['Arg']) + ": " + argstr )

	for doc in keyword.findall("doc"):
		print(indent+indentstr + colored("Doc", colours['Doc']) + ": " + doc.text)

	for msg in keyword.findall("msg"):
		if msg.text:
			txt = msg.text.replace("\n", "\n" + indent+indentstr)
			print(indent+indentstr + colored("Msg", colours['Msg']) + ": " + txt)

	for subkw in keyword.findall("kw"):
		process_kw(subkw, indent + indentstr)

def process_test(test):
	print(indentstr + test.attrib.get("name"))
	for keyword in test.findall("kw"):
		process_kw(keyword, indentstr+indentstr)
	print()

def process_suite(suite):
	for subsuite in suite.findall('suite'):
		print("Suite:" + subsuite.attrib.get("name"))
		process_suite(subsuite)
	for test in suite.findall("test"):
		process_test(test)


# iterate over all suites
for suite in root.findall('suite'):
	process_suite(suite)

	for kw in suite.findall("kw"):
		if( kw.attrib.get("type") == "teardown" ):
			print(indentstr + "Suite teardown")
			process_kw(kw, indentstr)

# vim: set noexpandtab: