Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 1 | from rest_framework import generics |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 2 | from django.shortcuts import get_object_or_404 |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 3 | from crashreports.models import HeartBeat |
| 4 | from crashreports.serializers import HeartBeatSerializer |
| 5 | from crashreports.permissions import HasRightsOrIsDeviceOwnerDeviceCreation |
| 6 | |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 7 | # TODO: There is quite some code duplciation between here and the corresponding |
| 8 | # crashreport code. We should revisit this later. |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 9 | |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 10 | |
| 11 | class ListCreateView(generics.ListCreateAPIView): |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 12 | queryset = HeartBeat.objects.all() |
| 13 | paginate_by = 20 |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 14 | permission_classes = (HasRightsOrIsDeviceOwnerDeviceCreation,) |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 15 | serializer_class = HeartBeatSerializer |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 16 | filter_fields = ("device", "build_fingerprint", "radio_version") |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 17 | |
| 18 | def get(self, *args, **kwargs): |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 19 | if "uuid" in kwargs: |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 20 | self.queryset = HeartBeat.objects.filter( |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 21 | device__uuid=kwargs["uuid"] |
| 22 | ) |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 23 | return generics.ListCreateAPIView.get(self, *args, **kwargs) |
| 24 | |
| 25 | |
| 26 | class RetrieveUpdateDestroyView(generics.RetrieveUpdateDestroyAPIView): |
| 27 | queryset = HeartBeat.objects.all() |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 28 | permission_classes = (HasRightsOrIsDeviceOwnerDeviceCreation,) |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 29 | serializer_class = HeartBeatSerializer |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 30 | multiple_lookup_fields = {"id", "device__uuid", "device_local_id"} |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 31 | |
| 32 | def get_object(self): |
| 33 | queryset = self.get_queryset() |
| 34 | filter = {} |
| 35 | for field in self.multiple_lookup_fields: |
| 36 | if field in self.kwargs: |
| 37 | filter[field] = self.kwargs[field] |
| 38 | obj = get_object_or_404(queryset, **filter) |
| 39 | self.check_object_permissions(self.request, obj) |
| 40 | return obj |