def dispatch_response_handler(r):
if isinstance(r, HttpResponseBase):
return r
elif isinstance(r, Part):
if not r._is_bound:
r = r.bind(request=request)
return HttpResponse(render_root(part=r, **kwargs))
else:
return HttpResponse(json.dumps(r), content_type='application/json')
def dispatch_response_handler(r):
if isinstance(r, HttpResponseBase):
return r
elif isinstance(r, Part):
# We can't do r.bind(...).render_to_response() because then we recurse in here
# r also has to be bound already
return HttpResponse(render_root(part=r, **kwargs))
else:
return HttpResponse(json.dumps(r),
content_type='application/json')
def render_to_response(self, **kwargs):
request = self.get_request()
req_data = request_data(request)
def dispatch_response_handler(r):
if isinstance(r, HttpResponseBase):
return r
elif isinstance(r, Part):
if not r._is_bound:
r = r.bind(request=request)
return HttpResponse(render_root(part=r, **kwargs))
else:
return HttpResponse(json.dumps(r),
content_type='application/json')
if request.method == 'GET':
dispatch_prefix = DISPATCH_PATH_SEPARATOR
dispatcher = perform_ajax_dispatch
dispatch_error = 'Invalid endpoint path'
elif request.method == 'POST':
dispatch_prefix = '-'
dispatcher = perform_post_dispatch
dispatch_error = 'Invalid post path'
else: # pragma: no cover
assert False # This has already been checked in request_data()
dispatch_commands = {
key: value
for key, value in items(req_data)
if key.startswith(dispatch_prefix)
}
assert len(dispatch_commands) in (
0, 1), 'You can only have one or no dispatch commands'
if dispatch_commands:
dispatch_target, value = next(iter(dispatch_commands.items()))
try:
result = dispatcher(root=self,
path=dispatch_target,
value=value)
except InvalidEndpointPathException:
if settings.DEBUG:
raise
result = dict(error=dispatch_error)
if result is not None:
return dispatch_response_handler(result)
else:
if request.method == 'POST':
assert False, 'This request was a POST, but there was no dispatch command present.'
response = HttpResponse(render_root(part=self, **kwargs))
response.iommi_part = self
return response
def __call__(self, request):
# Disable profiling early on /media requests since touching request.user will add a
# "Vary: Cookie" header to the response.
request.profiler_disabled = False
for prefix in MEDIA_PREFIXES:
if request.path.startswith(prefix):
request.profiler_disabled = True
break
if should_profile(request):
prof = cProfile.Profile()
prof.enable()
else:
prof = None
response = self.get_response(request)
if prof is not None:
response = HttpResponse()
prof.disable()
import pstats
s = StringIO()
ps = pstats.Stats(prof, stream=s).sort_stats(
request.GET.get('_iommi_prof') or 'cumulative')
ps.print_stats()
stats_str = s.getvalue()
if 'graph' in request.GET:
with NamedTemporaryFile() as stats_dump:
ps.stream = stats_dump
ps.dump_stats(stats_dump.name)
gprof2dot_path = Path(sys.executable).parent / 'gprof2dot'
if not gprof2dot_path.exists():
raise Exception(
'gprof2dot not found. Please install it to use the graph feature.'
)
gprof2dot = subprocess.Popen(
(sys.executable, gprof2dot_path, '-f', 'pstats',
stats_dump.name),
stdout=subprocess.PIPE)
response['Content-Type'] = 'image/svg+xml'
dot_path = get_dot_path()
if dot_path:
response.content = subprocess.check_output(
(dot_path, '-Tsvg'), stdin=gprof2dot.stdout)
else:
response['Content-Type'] = 'text/plain'
response[
'Content-Disposition'] = "attachment; filename=gprof2dot-graph.txt"
response.content = subprocess.check_output(
'tee', stdin=gprof2dot.stdout)
else:
limit = 280
result = []
def strip_extra_path(s, token):
if token not in s:
return s
pre, _, post = s.rpartition(' ')
post = post[post.rindex(token) + len(token):]
return f'{pre} {post}'
base_dir = str(settings.BASE_DIR)
for line in stats_str.split("\n")[:limit]:
should_bold = base_dir in line and '/site-packages/' not in line
line = line.replace(base_dir, '')
line = strip_extra_path(line, '/site-packages')
line = strip_extra_path(line, '/Python.framework/Versions')
if should_bold:
line = f'<b>{line}</b>'
line = line.replace(' ', ' ')
result.append(line)
start_html = '''
<style>
html {
font-family: monospace;
white-space: nowrap;
}
@media (prefers-color-scheme: dark) {
html {
background-color: black;
color: #bbb;
}
b {
color: white;
}
}
</style>
<div>'''
lines_html = "<br />\n".join(result)
end_html = '</div>'
response.content = start_html + lines_html + end_html
response['Content-Type'] = 'text/html'
return response
def csrf_exempt_view(request):
return HttpResponse('hello!') # pragma: no cover
def view(request):
return HttpResponse('hello!') # pragma: no cover
def __call__(self, request):
# Disable profiling early on /media requests since touching request.user will add a
# "Vary: Cookie" header to the response.
request.profiler_disabled = False
for prefix in MEDIA_PREFIXES:
if request.path.startswith(prefix):
request.profiler_disabled = True
break
if not request.profiler_disabled and (
settings.DEBUG
or request.user.is_staff) and 'prof' in request.GET:
self.prof = cProfile.Profile()
self.prof.enable()
response = self.get_response(request)
disabled = getattr(request, 'profiler_disabled', True)
is_staff = hasattr(request, 'user') and request.user.is_staff
if 'prof' in request.GET and not disabled and is_staff:
response = HttpResponse()
self.prof.disable()
import pstats
s = StringIO()
ps = pstats.Stats(self.prof, stream=s).sort_stats(
request.GET.get('prof') or 'cumulative')
ps.print_stats()
stats_str = s.getvalue()
if 'graph' in request.GET:
with NamedTemporaryFile() as stats_dump:
ps.stream = stats_dump
ps.dump_stats(stats_dump.name)
gprof2dot_path = os.path.join(
os.path.dirname(os.path.abspath(__file__)), 'bin',
'profiling', 'gprof2dot.py')
gprof2dot = subprocess.Popen(
('python', gprof2dot_path, '-f', 'pstats',
stats_dump.name),
stdout=subprocess.PIPE)
response['Content-Type'] = 'image/svg+xml'
if os.path.exists('/usr/bin/dot'):
response.content = subprocess.check_output(
('/usr/bin/dot', '-Tsvg'), stdin=gprof2dot.stdout)
elif os.path.exists('/usr/local/bin/dot'):
response.content = subprocess.check_output(
('/usr/local/bin/dot', '-Tsvg'),
stdin=gprof2dot.stdout)
else:
response['Content-Type'] = 'text/plain'
response[
'Content-Disposition'] = "attachment; filename=gprof2dot-graph.txt"
response.content = subprocess.check_output(
'tee', stdin=gprof2dot.stdout)
else:
limit = 280
result = []
def strip_extra_path(s, token):
if token not in s:
return s
pre, _, post = s.rpartition(' ')
post = post[post.rindex(token) + len(token):]
return f'{pre} {post}'
for line in stats_str.split("\n")[:limit]:
should_bold = settings.BASE_DIR in line and '/site-packages/' not in line or '/tri/' in line
line = line.replace(settings.BASE_DIR, '')
line = strip_extra_path(line, '/site-packages')
line = strip_extra_path(line, '/Python.framework/Versions')
if should_bold:
line = f'<b>{line}</b>'
line = line.replace(' ', ' ')
result.append(line)
response.content = '<div style="font-family: monospace; white-space: nowrap">%s</div' % "<br />\n".join(
result)
response['Content-Type'] = 'text/html'
return response
