From f556e21d511bb5ed8854d5c700117c9f4defe78d Mon Sep 17 00:00:00 2001 From: Tanner Collin Date: Fri, 4 Feb 2022 04:16:35 +0000 Subject: [PATCH] Make sure member_id is included in requests --- apiserver/apiserver/api/serializers.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/apiserver/apiserver/api/serializers.py b/apiserver/apiserver/api/serializers.py index 8a27db9..7ae197c 100644 --- a/apiserver/apiserver/api/serializers.py +++ b/apiserver/apiserver/api/serializers.py @@ -68,6 +68,9 @@ class TransactionSerializer(serializers.ModelSerializer): ] def create(self, validated_data): + if not self.initial_data.get('member_id', None): + raise ValidationError(dict(member_id='This field is required.')) + member = get_object_or_404(models.Member, id=self.initial_data['member_id']) validated_data['user'] = member.user @@ -88,6 +91,9 @@ class TransactionSerializer(serializers.ModelSerializer): return super().create(validated_data) def update(self, instance, validated_data): + if not self.initial_data.get('member_id', None): + raise ValidationError(dict(member_id='This field is required.')) + member = get_object_or_404(models.Member, id=self.initial_data['member_id']) validated_data['user'] = member.user return super().update(instance, validated_data) @@ -395,6 +401,9 @@ class CardSerializer(serializers.ModelSerializer): ] def create(self, validated_data): + if not self.initial_data.get('member_id', None): + raise ValidationError(dict(member_id='This field is required.')) + member = get_object_or_404(models.Member, id=self.initial_data['member_id']) validated_data['user'] = member.user @@ -403,6 +412,9 @@ class CardSerializer(serializers.ModelSerializer): return super().create(validated_data) def update(self, instance, validated_data): + if not self.initial_data.get('member_id', None): + raise ValidationError(dict(member_id='This field is required.')) + member = get_object_or_404(models.Member, id=self.initial_data['member_id']) validated_data['user'] = member.user return super().update(instance, validated_data)