evan/ewconfig
1
Fork 0
Code

Adjust console width for better output

Browse Source 
visualization.
This commit is contained in:
Evan Pratten 2023-11-13 10:19:14 -05:00
parent 39bf9834b4
commit f2ce365c4b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/pytb
View File

@ -96,12 +96,13 @@ def main() -> int:
output = ""
for statement, context in output_lines:
output += f"{statement.ljust(longest_statement)}{context}\n"
longest_line = max(len(line) for line in output.splitlines())
# remove any trailing newlines
output = output.rstrip()
# Pass over to rich to do the syntax highlighting
console = Console(record=True)
console = Console(record=True, width=longest_line+1)
console.print(Syntax(output, "python", background_color="default"))
# Export an image
@ -110,7 +111,7 @@ def main() -> int:
file_name += f" ({args.file.stem})"
file_name += ".svg"
OUTPUT_ROOT.mkdir(parents=True, exist_ok=True)
console.save_svg(OUTPUT_ROOT / file_name, title="Python Traceback Visualizer (a tool by ewpratten)")
console.save_svg(OUTPUT_ROOT / file_name, title="Evan's Python Traceback Visualizer")
return 0