Limit the number of functions displayed by --profile

2016-10-22 11:02:35 +05:30 · 2016-10-22 11:02:35 +05:30 · a04cc3a97b
commit a04cc3a97b
parent 1244ad0aa1
1 changed files with 6 additions and 1 deletions
--- a/kitty/main.py
+++ b/kitty/main.py
@ -126,11 +126,16 @@ def main():
    w.show()
    if args.profile:
        import cProfile
+        import pstats
        pr = cProfile.Profile()
        pr.enable()
        ret = run_app()
-        pr.print_stats('tottime')
        pr.disable()
+        pr.create_stats()
+        s = pstats.Stats(pr)
+        s.strip_dirs()
+        s.sort_stats('time', 'name')
+        s.print_stats(30)
    else:
        ret = run_app()
    app.installEventFilter(None)