- ret = function(*args, **kwargs)
- except:
- caught = sys.exc_info()[0]
- pr.disable()
- ps = pstats.Stats(pr, stream=outfile)
- ps.print_stats()
- if caught:
- raise caught
- return ret
+ return function(*args, **kwargs)
+ finally:
+ pr.disable()
+ ps = pstats.Stats(pr, stream=outfile)
+ ps.sort_stats('time').print_stats()