From 72dc4ded1e4812d3c895662aef9e39f50407e546 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bodor=20M=C3=A1t=C3=A9?= <bmate711@gmail.com>
Date: Wed, 5 Feb 2020 20:57:18 +0100
Subject: [PATCH] Change file value in document serializer
---
src/document/serializers.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/src/document/serializers.py b/src/document/serializers.py
index 937ecac..f1c209f 100644
--- a/src/document/serializers.py
+++ b/src/document/serializers.py
@@ -8,6 +8,7 @@ _max_count = 1
class DocumentSerializer(serializers.ModelSerializer):
uploaded_by = serializers.HiddenField(default=CurrentUserProfileDefault())
uploaded_by_name = serializers.SerializerMethodField()
+ file = serializers.SerializerMethodField()
class Meta:
model = models.Document
@@ -30,6 +31,9 @@ class DocumentSerializer(serializers.ModelSerializer):
def get_uploaded_by_name(self, obj):
return obj.uploaded_by.full_name
+ def get_file(self, obj):
+ return f"/api/v1/documents/{obj.id}/download/"
+
def validate_solution(self, value):
profile = self.context['request'].user.profile
if value not in profile.solution.all():
@@ -38,7 +42,5 @@ class DocumentSerializer(serializers.ModelSerializer):
uploaded_by=profile, solution=value).count()
if count >= _max_count:
raise serializers.ValidationError(
- 'You cant upload more than ' +
- str(_max_count) +
- ' document to one solution!')
+ f'You cant upload more than {max_count} document to one solution!')
return value
--
GitLab