Borjan Tchakaloff | 340b6f3 | 2018-03-01 15:23:43 +0400 | [diff] [blame] | 1 | from django.utils import timezone |
Dirk Vogt | d134521 | 2016-09-14 14:31:45 +0200 | [diff] [blame] | 2 | from rest_framework import serializers |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 3 | from rest_framework.exceptions import NotFound |
| 4 | from crashreports.models import Crashreport |
| 5 | from crashreports.models import Device |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 6 | from crashreports.models import HeartBeat |
Dirk Vogt | 83107df | 2017-05-02 12:04:19 +0200 | [diff] [blame] | 7 | from crashreports.models import LogFile |
| 8 | |
Dirk Vogt | d134521 | 2016-09-14 14:31:45 +0200 | [diff] [blame] | 9 | from rest_framework import permissions |
Dirk Vogt | 67eb148 | 2016-10-13 12:42:56 +0200 | [diff] [blame] | 10 | from crashreports.permissions import user_is_hiccup_staff |
| 11 | |
| 12 | |
| 13 | class PrivateField(serializers.ReadOnlyField): |
| 14 | |
| 15 | def get_attribute(self, instance): |
| 16 | """ |
| 17 | Given the *outgoing* object instance, return the primitive value |
| 18 | that should be used for this field. |
| 19 | """ |
| 20 | if user_is_hiccup_staff(self.context['request'].user): |
| 21 | return super(PrivateField, self).get_attribute(instance) |
| 22 | return -1 |
Dirk Vogt | d134521 | 2016-09-14 14:31:45 +0200 | [diff] [blame] | 23 | |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 24 | |
Dirk Vogt | d134521 | 2016-09-14 14:31:45 +0200 | [diff] [blame] | 25 | class CrashReportSerializer(serializers.ModelSerializer): |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 26 | permission_classes = (permissions.AllowAny,) |
Dirk Vogt | 83107df | 2017-05-02 12:04:19 +0200 | [diff] [blame] | 27 | logfiles = serializers.HyperlinkedRelatedField( |
| 28 | read_only=True, |
| 29 | many=True, |
| 30 | view_name='api_v1_logfiles_by_id', |
| 31 | ) |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 32 | uuid = serializers.CharField(max_length=64) |
Dirk Vogt | 67eb148 | 2016-10-13 12:42:56 +0200 | [diff] [blame] | 33 | id = PrivateField() |
| 34 | device_local_id = serializers.IntegerField(required=False) |
Borjan Tchakaloff | 340b6f3 | 2018-03-01 15:23:43 +0400 | [diff] [blame] | 35 | date = serializers.DateTimeField(default_timezone=timezone.UTC()) |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 36 | |
Dirk Vogt | d134521 | 2016-09-14 14:31:45 +0200 | [diff] [blame] | 37 | class Meta: |
| 38 | model = Crashreport |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 39 | exclude = ('device',) |
| 40 | |
| 41 | def create(self, validated_data): |
| 42 | try: |
| 43 | device = Device.objects.get(uuid=validated_data['uuid']) |
| 44 | except: |
| 45 | raise NotFound(detail="uuid does not exist") |
| 46 | validated_data.pop('uuid', None) |
| 47 | report = Crashreport(**validated_data) |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 48 | report.device = device |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 49 | report.save() |
| 50 | return report |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 51 | |
| 52 | |
| 53 | class HeartBeatSerializer(serializers.ModelSerializer): |
Dirk Vogt | 3663569 | 2016-10-17 12:19:10 +0200 | [diff] [blame] | 54 | permission_classes = (permissions.IsAuthenticated,) |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 55 | uuid = serializers.CharField(max_length=64) |
Dirk Vogt | 67eb148 | 2016-10-13 12:42:56 +0200 | [diff] [blame] | 56 | id = PrivateField() |
| 57 | device_local_id = serializers.IntegerField(required=False) |
Borjan Tchakaloff | 340b6f3 | 2018-03-01 15:23:43 +0400 | [diff] [blame] | 58 | date = serializers.DateTimeField(default_timezone=timezone.UTC()) |
Dirk Vogt | e178488 | 2016-10-13 16:09:38 +0200 | [diff] [blame] | 59 | |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 60 | class Meta: |
| 61 | model = HeartBeat |
| 62 | exclude = ('device',) |
| 63 | |
| 64 | def create(self, validated_data): |
| 65 | try: |
| 66 | device = Device.objects.get(uuid=validated_data['uuid']) |
| 67 | except: |
| 68 | raise NotFound(detail="uuid does not exist") |
| 69 | validated_data.pop('uuid', None) |
| 70 | heartbeat = HeartBeat(**validated_data) |
| 71 | heartbeat.device = device |
| 72 | heartbeat.save() |
| 73 | return heartbeat |
| 74 | |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 75 | |
Dirk Vogt | 83107df | 2017-05-02 12:04:19 +0200 | [diff] [blame] | 76 | class LogFileSerializer(serializers.ModelSerializer): |
| 77 | permission_classes = (permissions.IsAdminUser,) |
| 78 | |
| 79 | class Meta: |
| 80 | model = LogFile |
| 81 | |
| 82 | |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 83 | class DeviceSerializer(serializers.ModelSerializer): |
| 84 | permission_classes = (permissions.IsAdminUser,) |
Borjan Tchakaloff | 340b6f3 | 2018-03-01 15:23:43 +0400 | [diff] [blame] | 85 | board_date = serializers.DateTimeField(default_timezone=timezone.UTC()) |
| 86 | last_heartbeat = serializers.DateTimeField(default_timezone=timezone.UTC()) |
Dirk Vogt | c9e10ab | 2016-10-12 13:58:15 +0200 | [diff] [blame] | 87 | |
Dirk Vogt | f2a3342 | 2016-10-11 17:17:26 +0200 | [diff] [blame] | 88 | class Meta: |
| 89 | model = Device |
Franz-Xaver Geiger | d994335 | 2018-02-27 14:26:41 +0100 | [diff] [blame] | 90 | |
| 91 | |
| 92 | class DeviceCreateSerializer(DeviceSerializer): |
| 93 | |
| 94 | class Meta: |
| 95 | model = Device |
| 96 | fields = ('board_date', 'chipset') |
| 97 | extra_kwargs = { |
| 98 | 'board_date': {'required': True}, |
| 99 | 'chipset': {'required': True}, |
| 100 | } |