37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from __future__ import absolute_import
|
|
|
|
import logging
|
|
|
|
from rest_framework import serializers
|
|
from rest_framework.reverse import reverse
|
|
|
|
from .models import StagingFolder
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class SerializerStagingFolderFile(serializers.Serializer):
|
|
url = serializers.SerializerMethodField('get_url')
|
|
image_url = serializers.SerializerMethodField('get_image_url')
|
|
filename = serializers.CharField(max_length=255)
|
|
|
|
def get_url(self, obj):
|
|
return reverse('stagingfolderfile-detail', args=[obj.staging_folder.pk, obj.encoded_filename], request=self.context.get('request'))
|
|
|
|
def get_image_url(self, obj):
|
|
return reverse('stagingfolderfile-image-view', args=[obj.staging_folder.pk, obj.encoded_filename], request=self.context.get('request'))
|
|
|
|
|
|
class SerializerStagingFolder(serializers.HyperlinkedModelSerializer):
|
|
files = serializers.SerializerMethodField('get_files')
|
|
|
|
def get_files(self, obj):
|
|
try:
|
|
return [SerializerStagingFolderFile(entry, context=self.context).data for entry in obj.get_files()]
|
|
except Exception as exception:
|
|
logger.error('unhandled exception: %s' % exception)
|
|
return []
|
|
|
|
class Meta:
|
|
model = StagingFolder
|