Utils.py

import sys
import os

# Define our XML namespaces for easy translation
ns = {  'gnc': 'http://www.gnucash.org/XML/gnc',
	'act': 'http://www.gnucash.org/XML/act',
	'book': 'http://www.gnucash.org/XML/book',
	'cd': 'http://www.gnucash.org/XML/cd',
	'cmdty': 'http://www.gnucash.org/XML/cmdty',
	'price': 'http://www.gnucash.org/XML/price',
	'slot': 'http://www.gnucash.org/XML/slot',
	'split': 'http://www.gnucash.org/XML/split',
	'sx': 'http://www.gnucash.org/XML/sx',
	'trn': 'http://www.gnucash.org/XML/trn',
	'ts': 'http://www.gnucash.org/XML/ts',
	'fs': 'http://www.gnucash.org/XML/fs',
	'bgt': 'http://www.gnucash.org/XML/bgt',
	'recurrence': 'http://www.gnucash.org/XML/recurrence',
	'lot': 'http://www.gnucash.org/XML/lot',
	'addr': 'http://www.gnucash.org/XML/addr',
	'billterm': 'http://www.gnucash.org/XML/billterm',
	'bt-days': 'http://www.gnucash.org/XML/bt-days',
	'bt-prox': 'http://www.gnucash.org/XML/bt-prox',
	'cust': 'http://www.gnucash.org/XML/cust',
	'employee': 'http://www.gnucash.org/XML/employee',
	'entry': 'http://www.gnucash.org/XML/entry',
	'invoice': 'http://www.gnucash.org/XML/invoice',
	'job': 'http://www.gnucash.org/XML/job',
	'order': 'http://www.gnucash.org/XML/order',
	'owner': 'http://www.gnucash.org/XML/owner',
	'taxtable': 'http://www.gnucash.org/XML/taxtable',
	'tte': 'http://www.gnucash.org/XML/tte',
	'vendor': 'http://www.gnucash.org/XML/vendor'}

# Wrap print with the ability to output to stderr or a file
def output(*args, error=False, outfile=None, ending=os.linesep):
	# Catch any BrokenPipeErrors if the other end of the output pipe breaks
	try:
		if error:
			print(*args, file=sys.stderr)
		else:
			if outfile:
				temp_file = open(outfile, "a")
				print(*args, file=temp_file, end=ending)
			else:
				print(*args, end=ending)
	except BrokenPipeError:
		pass

# Check to see if an element has a text attribute. If the element does have a text attribute, return it, else return none
def safety_text(element):
	if hasattr(element, 'text'):
		if element.text is None:
			return ""
		else:
			temp = element.text

			# Make sure that the text does not contain braces as these have special meaning in the ledger format
			temp = temp.replace(']', ')')
			temp = temp.replace('[', '(')

			# Also replace all newlines with spaces since they will cause an entry not to balance
			temp = temp.replace('\r\n', ' ')
			temp = temp.replace('\r', ' ')
			temp = temp.replace('\n', ' ')
			return temp
	else:
		return None

# Prints with a certain level of indentation in front of it
def indent_print(*arg):
	prefix = ""
	count = arg[0]

	while count > 0:
		prefix += "│\t"
		count -= 1

	output(prefix, ending="")

	for part in arg[1:]:
		output(part, ending="")

	output()

# Print a very pretty representation of an object
def tree_print(name, thing, indent=0):
	if indent == 0:
		indent_print(indent, name, ":")
	else:
		indent_print(indent - 1, "├───────┬", name, ":")

	left = len(thing.__dict__)

	for key, value in thing.__dict__.items():
		if key == "splits":
			for subelement in value:
				tree_print("split", subelement, indent + 1)
		elif key == "children":
			for subelement in value:
				tree_print("child", subelement, indent + 1)
		else:
			left -= 1
			if left == 1:
				indent_print(indent,"└─", key, ": ", value)
			else:
				indent_print(indent,"├─", key, ": ", value)