Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 1 | from rest_framework import generics |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 2 | from crashreports.models import Crashreport |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 3 | from django.shortcuts import get_object_or_404 |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 4 | from crashreports.permissions import HasRightsOrIsDeviceOwnerDeviceCreation |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 5 | from crashreports.serializers import CrashReportSerializer |
Dirk Vogt | eda80d3 | 2016-11-21 11:45:50 +0100 | [diff] [blame] | 6 | from rest_framework import status |
| 7 | from rest_framework.response import Response |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 8 | |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 9 | |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 10 | class ListCreateView(generics.ListCreateAPIView): |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 11 | queryset = Crashreport.objects.all() |
| 12 | paginate_by = 20 |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 13 | permission_classes = (HasRightsOrIsDeviceOwnerDeviceCreation, ) |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 14 | serializer_class = CrashReportSerializer |
Dirk Vogt | 83107df | 2017-05-02 12:04:19 +0200 | [diff] [blame^] | 15 | filter_fields = ('device','build_fingerprint') |
| 16 | |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 17 | pass |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 18 | |
| 19 | def dispatch(self, *args, **kwargs): |
| 20 | if 'uuid' in kwargs: |
| 21 | self.queryset = Crashreport.objects.filter( |
| 22 | device__uuid=kwargs['uuid']) |
| 23 | return generics.ListCreateAPIView.dispatch(self, *args, **kwargs) |
| 24 | |
Dirk Vogt | eda80d3 | 2016-11-21 11:45:50 +0100 | [diff] [blame] | 25 | def perform_create(self, serializer): |
| 26 | serializer.save() |
| 27 | return Response( |
| 28 | { |
| 29 | 'device_local_id': serializer.data['device_local_id'] |
| 30 | }, status.HTTP_200_OK) |
| 31 | |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 32 | |
| 33 | class RetrieveUpdateDestroyView(generics.RetrieveUpdateDestroyAPIView): |
| 34 | queryset = Crashreport.objects.all() |
| 35 | permission_classes = (HasRightsOrIsDeviceOwnerDeviceCreation, ) |
| 36 | serializer_class = CrashReportSerializer |
Dirk Vogt | 0d9d5d2 | 2016-10-13 16:17:57 +0200 | [diff] [blame] | 37 | multiple_lookup_fields = {'id', 'device__uuid', 'device_local_id'} |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 38 | |
| 39 | def get_object(self): |
| 40 | queryset = self.get_queryset() |
| 41 | filter = {} |
| 42 | for field in self.multiple_lookup_fields: |
| 43 | if field in self.kwargs: |
| 44 | filter[field] = self.kwargs[field] |
| 45 | obj = get_object_or_404(queryset, **filter) |
| 46 | self.check_object_permissions(self.request, obj) |
| 47 | return obj |