84 lines
3.0 KiB
Python
84 lines
3.0 KiB
Python
|
import hashlib
|
||
|
|
|
||
|
|
from rest_framework import serializers
|
||
|
|
|
||
|
|
from apps.accounts.models import User
|
||
|
|
from apps.accounts.serializers import UserSerializer
|
||
|
|
|
||
|
|
from .models import Album, AlbumCollaborator, AlbumItem, LibraryItem, MediaRendition
|
||
|
|
|
||
|
|
|
||
|
|
class MediaRenditionSerializer(serializers.ModelSerializer):
|
||
|
|
class Meta:
|
||
|
|
model = MediaRendition
|
||
|
|
fields = ["kind", "file", "width", "height"]
|
||
|
|
|
||
|
|
|
||
|
|
class LibraryItemSerializer(serializers.ModelSerializer):
|
||
|
|
renditions = MediaRenditionSerializer(many=True, read_only=True)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = LibraryItem
|
||
|
|
fields = [
|
||
|
|
"id", "file", "media_type", "original_filename", "file_size",
|
||
|
|
"width", "height", "duration_seconds", "captured_at",
|
||
|
|
"latitude", "longitude", "camera_make", "camera_model",
|
||
|
|
"processing_status", "renditions", "created_at",
|
||
|
|
]
|
||
|
|
read_only_fields = [
|
||
|
|
"id", "width", "height", "duration_seconds", "captured_at",
|
||
|
|
"camera_make", "camera_model", "processing_status", "renditions", "created_at",
|
||
|
|
]
|
||
|
|
|
||
|
|
def create(self, validated_data):
|
||
|
|
upload = validated_data["file"]
|
||
|
|
checksum = hashlib.sha256(upload.read()).hexdigest()
|
||
|
|
upload.seek(0)
|
||
|
|
validated_data["checksum"] = checksum
|
||
|
|
validated_data["file_size"] = upload.size
|
||
|
|
validated_data.setdefault("original_filename", upload.name)
|
||
|
|
validated_data["owner"] = self.context["request"].user
|
||
|
|
library_item, _created = LibraryItem.objects.get_or_create(
|
||
|
|
owner=validated_data["owner"],
|
||
|
|
checksum=checksum,
|
||
|
|
defaults=validated_data,
|
||
|
|
)
|
||
|
|
return library_item
|
||
|
|
|
||
|
|
|
||
|
|
class AlbumItemSerializer(serializers.ModelSerializer):
|
||
|
|
library_item = LibraryItemSerializer(read_only=True)
|
||
|
|
library_item_id = serializers.PrimaryKeyRelatedField(
|
||
|
|
queryset=LibraryItem.objects.all(), source="library_item", write_only=True
|
||
|
|
)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = AlbumItem
|
||
|
|
fields = ["id", "album", "library_item", "library_item_id", "added_by", "added_at"]
|
||
|
|
read_only_fields = ["id", "album", "added_by", "added_at"]
|
||
|
|
|
||
|
|
|
||
|
|
class AlbumCollaboratorSerializer(serializers.ModelSerializer):
|
||
|
|
user = UserSerializer(read_only=True)
|
||
|
|
user_id = serializers.PrimaryKeyRelatedField(
|
||
|
|
queryset=User.objects.all(), source="user", write_only=True
|
||
|
|
)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = AlbumCollaborator
|
||
|
|
fields = ["id", "album", "user", "user_id", "role", "status", "invited_by"]
|
||
|
|
read_only_fields = ["id", "album", "status", "invited_by"]
|
||
|
|
|
||
|
|
|
||
|
|
class AlbumSerializer(serializers.ModelSerializer):
|
||
|
|
items = AlbumItemSerializer(many=True, read_only=True)
|
||
|
|
collaborators = AlbumCollaboratorSerializer(many=True, read_only=True)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = Album
|
||
|
|
fields = [
|
||
|
|
"id", "owner", "name", "description", "cover_item", "audience",
|
||
|
|
"items", "collaborators", "created_at", "updated_at",
|
||
|
|
]
|
||
|
|
read_only_fields = ["id", "owner", "items", "collaborators", "created_at", "updated_at"]
|