117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
|
|
||
|
|
from __future__ import print_function
|
||
|
|
from __future__ import division
|
||
|
|
import dsz
|
||
|
|
import dsz.ui
|
||
|
|
import ops.data
|
||
|
|
ALIGN_LEFT = '<'
|
||
|
|
ALIGN_CENTER = '_'
|
||
|
|
ALIGN_RIGHT = '>'
|
||
|
|
|
||
|
|
def pprint(data, header=None, dictorder=None, echocodes=None, align=None, print_handler=print):
|
||
|
|
if ((data is None) or (len(data) == 0)):
|
||
|
|
return
|
||
|
|
if ((dict is type(data[0])) and (dictorder is None)):
|
||
|
|
dictorder = data[0].keys()
|
||
|
|
if ((dict is type(data[0])) and (header is None)):
|
||
|
|
header = dictorder
|
||
|
|
if isinstance(data[0], ops.data.OpsObject):
|
||
|
|
newdata = list()
|
||
|
|
for item in data:
|
||
|
|
newdata.append(item.__dict__)
|
||
|
|
data = newdata
|
||
|
|
if (dictorder is None):
|
||
|
|
raise Exception('You must specify a dictorder (set of keys) when pprinting an ops.data object')
|
||
|
|
if (header is None):
|
||
|
|
header = dictorder
|
||
|
|
(sdata, align) = makeStrings(data, dictorder, align)
|
||
|
|
(widths, percents) = calcSize(sdata, header)
|
||
|
|
output = ''
|
||
|
|
if header:
|
||
|
|
for i in range(len(header)):
|
||
|
|
output += ((('|' + (' ' * (((widths[i] - len(header[i])) // 2) + 1))) + header[i]) + (' ' * (((widths[i] - len(header[i])) // 2) + 1)))
|
||
|
|
if ((widths[i] - len(header[i])) % 2):
|
||
|
|
output += ' '
|
||
|
|
if percents[i]:
|
||
|
|
output += (' ' * (percents[i] - header[i].count('%')))
|
||
|
|
output += '|'
|
||
|
|
if echocodes:
|
||
|
|
dsz.ui.Echo(output)
|
||
|
|
output = ''
|
||
|
|
else:
|
||
|
|
output += '\n'
|
||
|
|
for i in range(len(widths)):
|
||
|
|
output += ('+-' + ('-' * ((widths[i] + 1) + percents[i])))
|
||
|
|
output += '+'
|
||
|
|
if echocodes:
|
||
|
|
dsz.ui.Echo(output)
|
||
|
|
output = ''
|
||
|
|
else:
|
||
|
|
output += '\n'
|
||
|
|
for j in range(len(sdata)):
|
||
|
|
d = sdata[j]
|
||
|
|
a = align[j]
|
||
|
|
for i in range(len(d)):
|
||
|
|
if (a[i] == ALIGN_RIGHT):
|
||
|
|
output += ((('|' + (' ' * ((widths[i] - len(d[i])) + 1))) + d[i]) + ' ')
|
||
|
|
elif (a[i] == ALIGN_CENTER):
|
||
|
|
output += ((('|' + (' ' * (((widths[i] - len(d[i])) // 2) + 1))) + d[i]) + (' ' * (((widths[i] - len(d[i])) // 2) + 1)))
|
||
|
|
if ((widths[i] - len(d[i])) % 2):
|
||
|
|
output += ' '
|
||
|
|
else:
|
||
|
|
output += (('| ' + d[i]) + (' ' * ((widths[i] - len(d[i])) + 1)))
|
||
|
|
if percents[i]:
|
||
|
|
output += (' ' * (percents[i] - d[i].count('%')))
|
||
|
|
output += '|'
|
||
|
|
if echocodes:
|
||
|
|
dsz.ui.Echo((output.encode('utf8') if isinstance(output, unicode) else output), echocodes[j])
|
||
|
|
output = ''
|
||
|
|
else:
|
||
|
|
output += '\n'
|
||
|
|
if (echocodes is None):
|
||
|
|
print_handler(output, end='')
|
||
|
|
|
||
|
|
def makeStrings(data, dictOrder, align):
|
||
|
|
r = []
|
||
|
|
a = ([] if (align is None) else None)
|
||
|
|
for i in data:
|
||
|
|
c = []
|
||
|
|
ac = []
|
||
|
|
if dictOrder:
|
||
|
|
for k in dictOrder:
|
||
|
|
c += ([i[k]] if (unicode is type(i[k])) else [(str(i[k]) if (i[k] is not None) else '')])
|
||
|
|
if (a is not None):
|
||
|
|
ac += ([ALIGN_RIGHT] if ((int is type(i[k])) or (float is type(i[k])) or (long is type(i[k]))) else [ALIGN_LEFT])
|
||
|
|
else:
|
||
|
|
for k in i:
|
||
|
|
c += ([k] if (unicode is type(k)) else [(str(k) if (k is not None) else '')])
|
||
|
|
if (a is not None):
|
||
|
|
ac += ([ALIGN_RIGHT] if ((int is type(k)) or (float is type(k)) or (long is type(k))) else [ALIGN_LEFT])
|
||
|
|
r += [c]
|
||
|
|
if (a is not None):
|
||
|
|
a += [ac]
|
||
|
|
return (r, (a if (a is not None) else align))
|
||
|
|
|
||
|
|
def calcSize(data, header):
|
||
|
|
widths = range(len(data[0]))
|
||
|
|
percents = range(len(data[0]))
|
||
|
|
for i in widths:
|
||
|
|
widths[i] = 0
|
||
|
|
percents[i] = 0
|
||
|
|
if header:
|
||
|
|
for i in range(len(header)):
|
||
|
|
r = len(header[i])
|
||
|
|
if (r > widths[i]):
|
||
|
|
widths[i] = r
|
||
|
|
r = header[i].count('%')
|
||
|
|
if (r > percents[i]):
|
||
|
|
percents[i] = r
|
||
|
|
for d in data:
|
||
|
|
for i in range(len(d)):
|
||
|
|
r = len(d[i])
|
||
|
|
if (r > widths[i]):
|
||
|
|
widths[i] = r
|
||
|
|
r = d[i].count('%')
|
||
|
|
if (r > percents[i]):
|
||
|
|
percents[i] = r
|
||
|
|
return (widths, percents)
|