Created
June 7, 2011 21:16
-
-
Save d0ugal/1013188 to your computer and use it in GitHub Desktop.
Quick n dirty decorator for profiling a view.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import hotshot | |
import os | |
import time | |
try: | |
PROFILE_LOG_BASE = settings.PROFILE_LOG_BASE | |
except: | |
PROFILE_LOG_BASE = "/tmp" | |
def profile(log_file): | |
"""Profile some callable. | |
This decorator uses the hotshot profiler to profile some callable (like | |
a view function or method) and dumps the profile data somewhere sensible | |
for later processing and examination. | |
It takes one argument, the profile log name. If it's a relative path, it | |
places it under the PROFILE_LOG_BASE. It also inserts a time stamp into the | |
file name, such that 'my_view.prof' become 'my_view-20100211T170321.prof', | |
where the time stamp is in UTC. This makes it easy to run and compare | |
multiple trials. | |
""" | |
if not os.path.isabs(log_file): | |
log_file = os.path.join(PROFILE_LOG_BASE, log_file) | |
def _outer(f): | |
def _inner(*args, **kwargs): | |
# Add a timestamp to the profile output when the callable | |
# is actually called. | |
(base, ext) = os.path.splitext(log_file) | |
base = base + "-" + time.strftime("%Y%m%dT%H%M%S", time.gmtime()) | |
final_log_file = base + ext | |
prof = hotshot.Profile(final_log_file) | |
try: | |
ret = prof.runcall(f, *args, **kwargs) | |
finally: | |
prof.close() | |
return ret | |
return _inner | |
return _outer |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment