diff --git a/backend/api/posts/serializers.py b/backend/api/posts/serializers.py index 251c40c..7216aaa 100644 --- a/backend/api/posts/serializers.py +++ b/backend/api/posts/serializers.py @@ -10,6 +10,7 @@ class PostPreviewSerializer(serializers.ModelSerializer): description = serializers.SerializerMethodField() date = serializers.SerializerMethodField() media = serializers.SerializerMethodField() + media_count = serializers.SerializerMethodField() source_site = serializers.SerializerMethodField() category = serializers.SerializerMethodField() creator = serializers.SerializerMethodField() @@ -27,6 +28,7 @@ class PostPreviewSerializer(serializers.ModelSerializer): "creator", "date", "media", + "media_count", "tags", ] @@ -72,6 +74,9 @@ class PostPreviewSerializer(serializers.ModelSerializer): data.append({"type": f_type, "mimetype": i.mimetype, "src": i.file.url}) return data + def get_media_count(self, obj: PostModel) -> int: + return obj.files.count() + def get_tags(self, obj: PostModel) -> List[str]: return [tag.slug for tag in obj.tags.all()]