2025-04-07 00:30:11 -05:00
|
|
|
from itertools import count
|
|
|
|
|
|
|
|
from rest_framework import serializers
|
|
|
|
from movie_manager.models import Movie, MovieList
|
|
|
|
|
|
|
|
class MovieSerializer(serializers.ModelSerializer):
|
|
|
|
class Meta:
|
|
|
|
model = Movie
|
|
|
|
fields = '__all__'
|
|
|
|
|
|
|
|
|
|
|
|
class MovieListSerializer(serializers.ModelSerializer):
|
|
|
|
movie_count = serializers.SerializerMethodField()
|
2025-04-07 21:05:05 -05:00
|
|
|
movies = MovieSerializer(read_only=True, many=True)
|
2025-04-07 00:30:11 -05:00
|
|
|
|
|
|
|
class Meta:
|
|
|
|
model = MovieList
|
2025-04-07 21:05:05 -05:00
|
|
|
fields = ["id","name","owner","public", "movies", "movie_count"]
|
2025-04-07 00:30:11 -05:00
|
|
|
|
|
|
|
|
|
|
|
def get_movie_count(self, obj):
|
|
|
|
return len(obj.movies.all())
|