|
| 1 | +import logging |
| 2 | +from http import HTTPStatus |
| 3 | + |
1 | 4 | from rest_framework import viewsets |
| 5 | +from rest_framework.decorators import action |
| 6 | +from rest_framework.exceptions import ValidationError |
2 | 7 | from rest_framework.permissions import IsAuthenticated |
| 8 | +from rest_framework.response import Response |
3 | 9 |
|
| 10 | +from api_app.analyzables_manager.filters import AnalyzableFilter |
| 11 | +from api_app.analyzables_manager.models import Analyzable |
4 | 12 | from api_app.analyzables_manager.serializers import AnalyzableSerializer |
| 13 | +from api_app.serializers.job import JobAnalyzableHistorySerializer |
| 14 | +from api_app.user_events_manager.serializers import ( |
| 15 | + UserAnalyzableEventSerializer, |
| 16 | + UserDomainWildCardEventSerializer, |
| 17 | + UserIPWildCardEventSerializer, |
| 18 | +) |
| 19 | + |
| 20 | +logger = logging.getLogger(__name__) |
5 | 21 |
|
6 | 22 |
|
7 | 23 | class AnalyzableViewSet(viewsets.ReadOnlyModelViewSet): |
8 | 24 |
|
9 | 25 | serializer_class = AnalyzableSerializer |
10 | 26 | permission_classes = [IsAuthenticated] |
| 27 | + queryset = Analyzable.objects.all() |
| 28 | + filterset_class = AnalyzableFilter |
11 | 29 |
|
12 | 30 | def get_queryset(self): |
13 | 31 | user = self.request.user |
14 | 32 | return super().get_queryset().visible_for_user(user) |
| 33 | + |
| 34 | + @action(detail=True) |
| 35 | + def history(self, request, pk=None): |
| 36 | + user = request.user |
| 37 | + try: |
| 38 | + analyzable: Analyzable = self.get_queryset().get(pk=pk) |
| 39 | + except Analyzable.DoesNotExist: |
| 40 | + raise ValidationError({"detail": "Requested analyzable does not exist."}) |
| 41 | + |
| 42 | + jobs_queryset = analyzable.jobs.visible_for_user(user).order_by( |
| 43 | + "-finished_analysis_time" |
| 44 | + ) |
| 45 | + user_events_queryset = analyzable.user_events.visible_for_user(user).order_by( |
| 46 | + "-date" |
| 47 | + ) |
| 48 | + user_domain_wildcard_events_queryset = ( |
| 49 | + analyzable.user_domain_wildcard_events.visible_for_user(user).order_by( |
| 50 | + "-date" |
| 51 | + ) |
| 52 | + ) |
| 53 | + user_ip_wildcard_events_queryset = ( |
| 54 | + analyzable.user_ip_wildcard_events.visible_for_user(user).order_by("-date") |
| 55 | + ) |
| 56 | + |
| 57 | + jobs = JobAnalyzableHistorySerializer(jobs_queryset, many=True).data |
| 58 | + user_events = UserAnalyzableEventSerializer( |
| 59 | + user_events_queryset, many=True |
| 60 | + ).data |
| 61 | + user_domain_wildcard_events = UserDomainWildCardEventSerializer( |
| 62 | + user_domain_wildcard_events_queryset, many=True |
| 63 | + ).data |
| 64 | + user_ip_wildcard_events = UserIPWildCardEventSerializer( |
| 65 | + user_ip_wildcard_events_queryset, many=True |
| 66 | + ).data |
| 67 | + |
| 68 | + return Response( |
| 69 | + status=HTTPStatus.OK.value, |
| 70 | + data={ |
| 71 | + "jobs": jobs, |
| 72 | + "user_events": user_events, |
| 73 | + "user_domain_wildcard_events": user_domain_wildcard_events, |
| 74 | + "user_ip_wildcard_events": user_ip_wildcard_events, |
| 75 | + }, |
| 76 | + ) |
0 commit comments