movie-night-api/movie_manager/serializers.py

23 lines
595 B
Python
Raw Normal View History

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())