🍻🍻🍻 Added a field to the serializer of attachments to show the file name 🍻🍻🍻
parent
6cb055abb7
commit
a76b811bfd
|
@ -6,17 +6,25 @@ from greenmine.base.serializers import PickleField
|
||||||
|
|
||||||
from . import models
|
from . import models
|
||||||
|
|
||||||
|
from os import path
|
||||||
|
|
||||||
|
|
||||||
class AttachmentSerializer(serializers.ModelSerializer):
|
class AttachmentSerializer(serializers.ModelSerializer):
|
||||||
|
name = serializers.SerializerMethodField("get_name")
|
||||||
url = serializers.SerializerMethodField("get_url")
|
url = serializers.SerializerMethodField("get_url")
|
||||||
size = serializers.SerializerMethodField("get_size")
|
size = serializers.SerializerMethodField("get_size")
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
model = models.Attachment
|
model = models.Attachment
|
||||||
fields = ("id", "project", "owner", "attached_file", "size", "created_date",
|
fields = ("id", "project", "owner", "name", "attached_file", "size",
|
||||||
"modified_date", "object_id", "url")
|
"created_date", "modified_date", "object_id", "url")
|
||||||
read_only_fields = ("owner",)
|
read_only_fields = ("owner",)
|
||||||
|
|
||||||
|
def get_name(self, obj):
|
||||||
|
if obj.attached_file:
|
||||||
|
return path.basename(obj.attached_file.path)
|
||||||
|
return ""
|
||||||
|
|
||||||
def get_url(self, obj):
|
def get_url(self, obj):
|
||||||
return obj.attached_file.url if obj and obj.attached_file else ""
|
return obj.attached_file.url if obj and obj.attached_file else ""
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue