Created
January 6, 2014 12:21
-
-
Save pankaj28843/8282072 to your computer and use it in GitHub Desktop.
Django view profiler - a decorator function to profile your code
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 | |
from django.conf import settings | |
import tempfile | |
try: | |
PROFILE_LOG_BASE = settings.PROFILE_LOG_BASE | |
except: | |
PROFILE_LOG_BASE = tempfile.gettempdir() | |
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