From 274c8fe64703f82f53e9964a39d111e06cd1517e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?David=20Barrag=C3=A1n=20Merino?= Date: Wed, 23 Oct 2013 10:19:47 +0200 Subject: [PATCH] Added the file size to the serializer --- greenmine/projects/serializers.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/greenmine/projects/serializers.py b/greenmine/projects/serializers.py index 59c05894..3add6510 100644 --- a/greenmine/projects/serializers.py +++ b/greenmine/projects/serializers.py @@ -9,15 +9,21 @@ from . import models class AttachmentSerializer(serializers.ModelSerializer): url = serializers.SerializerMethodField("get_url") + size = serializers.SerializerMethodField("get_size") + + class Meta: + model = models.Attachment + fields = ("id", "project", "owner", "attached_file", "size", "created_date", + "modified_date", "object_id", "url") + read_only_fields = ("owner",) def get_url(self, obj): return obj.attached_file.url if obj and obj.attached_file else "" - class Meta: - model = models.Attachment - fields = ("id", "project", "owner", "attached_file", "created_date", - "modified_date", "object_id", "url") - read_only_fields = ("owner",) + def get_size(self, obj): + if obj.attached_file: + return obj.attached_file.size + return None # User Stories common serializers