rich/tools/profile_pretty.py

24 lines
488 B
Python
Raw Normal View History

2020-08-10 15:37:46 +00:00
import json
import io
from time import time
from rich.console import Console
from rich.pretty import Pretty
console = Console(file=io.StringIO(), color_system="truecolor", width=100)
with open("cats.json") as fh:
cats = json.load(fh)
2020-10-20 14:03:28 +00:00
console.begin_capture()
2020-08-10 15:37:46 +00:00
start = time()
pretty = Pretty(cats)
2020-08-19 09:31:11 +00:00
console.print(pretty, overflow="ignore", crop=False)
2020-10-20 14:03:28 +00:00
result = console.end_capture()
2020-08-10 15:37:46 +00:00
taken = (time() - start) * 1000
2020-10-20 14:03:28 +00:00
print(result)
2020-08-10 15:37:46 +00:00
print(console.file.getvalue())
print(f"{taken:.1f}")