Some snippets for profiling in Python

profiling.py

# paste these in
import contextlib

@contextlib.contextmanager
def profile(result_file_name=None):
  import cProfile
  import pstats
  import sys

  pr = cProfile.Profile()
  pr.enable()
  try:
    yield
  finally:
    pr.disable()
    ps = pstats.Stats(pr, stream=sys.stdout)
    ps.sort_stats('cumulative')
    if result_file_name:
      ps.dump_stats(result_file_name)
    else:
      ps.print_stats()

# in code
with profile('profiling.stats'):
  ret = something(blah)

# view output with snakeviz
# https://jiffyclub.github.io/snakeviz/