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