tanner/spaceport
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Make sure member_id is included in requests

Browse Source
This commit is contained in:
Tanner Collin 2022-02-04 04:16:35 +00:00
parent 0ab495f8c5
commit f556e21d51
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
apiserver/apiserver/api/serializers.py
View File

@ -68,6 +68,9 @@ class TransactionSerializer(serializers.ModelSerializer):
] ]
def create(self, validated_data): 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']) member = get_object_or_404(models.Member, id=self.initial_data['member_id'])
validated_data['user'] = member.user validated_data['user'] = member.user
@ -88,6 +91,9 @@ class TransactionSerializer(serializers.ModelSerializer):
return super().create(validated_data) return super().create(validated_data)
def update(self, instance, 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']) member = get_object_or_404(models.Member, id=self.initial_data['member_id'])
validated_data['user'] = member.user validated_data['user'] = member.user
return super().update(instance, validated_data) return super().update(instance, validated_data)
@ -395,6 +401,9 @@ class CardSerializer(serializers.ModelSerializer):
] ]
def create(self, validated_data): 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']) member = get_object_or_404(models.Member, id=self.initial_data['member_id'])
validated_data['user'] = member.user validated_data['user'] = member.user
@ -403,6 +412,9 @@ class CardSerializer(serializers.ModelSerializer):
return super().create(validated_data) return super().create(validated_data)
def update(self, instance, 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']) member = get_object_or_404(models.Member, id=self.initial_data['member_id'])
validated_data['user'] = member.user validated_data['user'] = member.user
return super().update(instance, validated_data) return super().update(instance, validated_data)