from rest_framework import serializers
from .models import CustomUser

class UserSerializer(serializers.ModelSerializer):
  class Meta:
    model = CustomUser
    fields = ["id", "username", "email", "liked_videos", "watched_videos"]
    

class LikedVideosSerializer(serializers.ModelSerializer):
    class Meta:
        model = CustomUser
        fields = ['liked_videos']
        
class WatchedVideosSerializer(serializers.ModelSerializer):
    class Meta:
        model = CustomUser
        fields = ['watched_videos']