Simple profiling context manager

profile.py

import cProfile
import contextlib
import os


@contextlib.contextmanager
def profile(filename='~/python.profile', *args, **kwargs):
    profile = cProfile.Profile(*args, **kwargs)

    profile.enable()
    yield
    profile.disable()

    if filename:
        profile.dump_stats(os.path.expanduser(filename))
    else:
        profile.print_stats()