Make custom traceback print_list compatible with standard tracebacks

This commit is contained in:
David Vierra 2015-09-05 10:42:54 -10:00
parent ad301d77a8
commit 0cb359f11c

View File

@ -84,11 +84,16 @@ def print_list(extracted_list, file=None):
""" """
if file is None: if file is None:
file = sys.stderr file = sys.stderr
for filename, lineno, name, line, selfstr in extracted_list: for entry in extracted_list:
print(file, if len(entry) > 4:
' File "%s", line %d, in %s %s' % (filename, lineno, name, selfstr)) filename, lineno, name, line, selfstr = entry
else:
filename, lineno, name, line = entry
selfstr = ""
print(' File "%s", line %d, in %s %s' % (filename, lineno, name, selfstr),
file=file)
if line: if line:
print(file, ' %s' % line.strip()) print(' %s' % line.strip(), file=file)
def print_tb(tb, limit=None, file=None): def print_tb(tb, limit=None, file=None):