mirror of
https://github.com/antonioCoco/SharPyShell
synced 2026-06-08 13:11:44 +00:00
NuHarborMartin
39e931ff27
Fixing the string/bytes conversion needed to work with python3 All modules tested (except lateral) and responding as they did with 2.7 on an IIS10 box
84 lines
2.2 KiB
Python
84 lines
2.2 KiB
Python
import prettytable
|
|
import types
|
|
|
|
|
|
def tablify(data, table_border=True):
|
|
# TODO: Check that is prettytable-0.7.2 that supports the
|
|
# dynamic table columns number setting. Version 0.5 does not.
|
|
|
|
output = ''
|
|
|
|
# Empty outputs. False is probably a good output value
|
|
if data and not data:
|
|
output = ''
|
|
else:
|
|
|
|
table = prettytable.PrettyTable()
|
|
|
|
# List outputs.
|
|
if isinstance(data, (list, tuple)):
|
|
|
|
if len(data) > 0:
|
|
|
|
columns_num = 1
|
|
if isinstance(data[0], (list, tuple)):
|
|
columns_num = len(data[0])
|
|
|
|
for row in data:
|
|
if not row:
|
|
continue
|
|
|
|
if isinstance(row, (list, tuple)):
|
|
table.add_row(row)
|
|
else:
|
|
table.add_row([row])
|
|
|
|
# Dict outputs are display as tables
|
|
elif isinstance(data, types.DictType) and data:
|
|
|
|
# Populate the rows
|
|
randomitem = next(data.itervalues())
|
|
if isinstance(randomitem, (list, tuple)):
|
|
for field in data:
|
|
table.add_row([field] + data[field])
|
|
else:
|
|
for field in data:
|
|
table.add_row([field, str(data[field])])
|
|
|
|
# Else, try to stringify
|
|
else:
|
|
output = str(data)
|
|
|
|
if not output:
|
|
table.header = False
|
|
table.align = 'l'
|
|
table.border = table_border
|
|
output = table.get_string()
|
|
|
|
return output
|
|
|
|
|
|
def shorten(body, keep_header=0, keep_trailer=0):
|
|
"""
|
|
Smartly shorten a given string.
|
|
"""
|
|
|
|
# Cut header
|
|
if (keep_header
|
|
and not keep_trailer
|
|
and len(body) > keep_header):
|
|
return '..%s' % body[:keep_header]
|
|
|
|
# Cut footer
|
|
if (keep_trailer
|
|
and not keep_header
|
|
and len(body) > keep_trailer):
|
|
return '..%s' % body[-keep_header:]
|
|
|
|
if (keep_header
|
|
and keep_trailer
|
|
and len(body) > keep_header + keep_trailer):
|
|
return '%s .. %s' % (body[:keep_header], body[-keep_trailer:])
|
|
|
|
return body
|