def session_detail(request, profile_id, session_id, template_name="spider/session_detail.html"):
profile = get_object_or_404(SpiderProfile, pk=profile_id)
session = get_object_or_404(profile.sessions.all(), pk=session_id)
queryset = session.results.all()
max_id = queryset.aggregate(max_id=Max("id"))["max_id"] or 0
if request.is_ajax():
max_id = int(request.GET.get("max_id", max_id))
context = {"max_id": max_id, "complete": session.complete, "results": []}
queryset = queryset.order_by("-pk")
results = queryset.filter(pk__gt=max_id).order_by("-id")
if results:
context["max_id"] = results[0].pk
context["results"] = [render_to_string("spider/includes/result_list.html", {"object": r}) for r in results]
return json_response(context)
else:
ordering = request.GET.get("ordering", "").lstrip("-")
if ordering in ("response_time", "content_length", "url", "response_status"):
queryset = queryset.order_by(request.GET["ordering"])
else:
queryset = queryset.order_by("-response_status", "url")
return object_list(
request,
queryset=queryset,
template_name=template_name,
paginate_by=50,
page=int(request.GET.get("page", 1)),
extra_context={"session": session, "max_id": max_id},
)
def url_result_detail(request, profile_id, session_id, url_result_id, template_name="spider/url_result_detail.html"):
profile = get_object_or_404(SpiderProfile, pk=profile_id)
session = get_object_or_404(profile.sessions.all(), pk=session_id)
if request.is_ajax():
result = get_object_or_404(session.results.all(), pk=url_result_id)
previous_qs = URLResult.objects.filter(
session__spider_profile=profile, url=result.url, created_date__lte=result.created_date
)
response_times = previous_qs.values_list("response_time", flat=True).order_by("created_date")
context = {
"rendered": render_to_string("spider/includes/result_detail.html", {"object": result}),
"response_times": list(enumerate(response_times[:50])),
}
return json_response(context)
else:
return object_detail(
request,
queryset=session.results.all(),
object_id=url_result_id,
template_name=template_name,
extra_context={"profile": profile, "session": session},
)
def url_result_detail(request, profile_id, session_id, url_result_id,
template_name='spider/url_result_detail.html'):
profile = get_object_or_404(SpiderProfile, pk=profile_id)
session = get_object_or_404(profile.sessions.all(), pk=session_id)
if request.is_ajax():
result = get_object_or_404(session.results.all(), pk=url_result_id)
previous_qs = URLResult.objects.filter(
session__spider_profile=profile,
url=result.url,
created_date__lte=result.created_date,
)
response_times = previous_qs.values_list(
'response_time', flat=True
).order_by('created_date')
context = {
'rendered': render_to_string('spider/includes/result_detail.html', {'object': result}),
'response_times': list(enumerate(response_times[:50])),
}
return json_response(context)
else:
return object_detail(
request,
queryset=session.results.all(),
object_id=url_result_id,
template_name=template_name,
extra_context={
'profile': profile,
'session': session,
}
)
def session_create(request, profile_id):
if request.method != "POST":
return HttpResponseNotAllowed()
profile = get_object_or_404(SpiderProfile, pk=profile_id)
new_session = SpiderSession.objects.create(spider_profile=profile)
run_spider(new_session.pk)
if request.is_ajax():
return json_response({"id": new_session.pk})
return redirect(new_session)
def session_create(request, profile_id):
if request.method != 'POST':
return HttpResponseNotAllowed()
profile = get_object_or_404(SpiderProfile, pk=profile_id)
new_session = SpiderSession.objects.create(spider_profile=profile)
run_spider(new_session.pk)
if request.is_ajax():
return json_response({'id': new_session.pk})
return redirect(new_session)
def dashboard_data_endpoint(request, data_type=PANEL_AGGREGATE_MINUTE):
panels = Panel.objects.get_panels()
max_id = int(request.GET.get('max_id') or 0)
limit = int(request.GET.get('limit') or DASHBOARD_DEFAULT_LIMIT)
panels_and_data = {}
for panel in panels:
panels_and_data[panel] = panel.data.filter(aggregate_type=data_type,
pk__gt=max_id)[:limit]
payload = serialize_panel_data(panels_and_data)
return json_response(payload)
def dashboard_data_endpoint(request, data_type=PANEL_AGGREGATE_MINUTE):
panels = Panel.objects.get_panels()
max_id = int(request.GET.get('max_id') or 0)
limit = int(request.GET.get('limit') or DASHBOARD_DEFAULT_LIMIT)
panels_and_data = {}
for panel in panels:
panels_and_data[panel] = panel.data.filter(
aggregate_type=data_type,
pk__gt=max_id
)[:limit]
payload = serialize_panel_data(panels_and_data)
return json_response(payload)
def session_detail(request, profile_id, session_id, template_name='spider/session_detail.html'):
profile = get_object_or_404(SpiderProfile, pk=profile_id)
session = get_object_or_404(profile.sessions.all(), pk=session_id)
queryset = session.results.all()
max_id = queryset.aggregate(max_id=Max('id'))['max_id'] or 0
if request.is_ajax():
max_id = int(request.GET.get('max_id', max_id))
context = {'max_id': max_id, 'complete': session.complete, 'results': []}
queryset = queryset.order_by('-pk')
results = queryset.filter(pk__gt=max_id).order_by('-id')
if results:
context['max_id'] = results[0].pk
context['results'] = [
render_to_string('spider/includes/result_list.html', {'object': r}) \
for r in results
]
return json_response(context)
else:
ordering = request.GET.get('ordering', '').lstrip('-')
if ordering in ('response_time', 'content_length', 'url', 'response_status'):
queryset = queryset.order_by(request.GET['ordering'])
else:
queryset = queryset.order_by('-response_status', 'url')
return object_list(
request,
queryset=queryset,
template_name=template_name,
paginate_by=50,
page=int(request.GET.get('page', 1)),
extra_context={
'session': session,
'max_id': max_id,
}
)
