Skip to content
Snippets Groups Projects
serializers.py 1.05 KiB
Newer Older
from rest_framework import serializers
from common.serializers import CurrentUserProfileDefault
from . import models


class EventSerializer(serializers.ModelSerializer):
    created_by = serializers.HiddenField(default=CurrentUserProfileDefault())
Bodor Máté's avatar
Bodor Máté committed
    created_by_name = serializers.SerializerMethodField()
    visitor_number = serializers.SerializerMethodField()
        fields = '__all__'
        read_only_fields = ('created_at', 'update_at', 'created_by')

    def get_created_by_name(self, obj):
Bodor Máté's avatar
Bodor Máté committed
        return obj.created_by.full_name

    def get_visitor_number(self, obj):
        return obj.visitors.all().count()


class NoteSerializer(serializers.ModelSerializer):
    created_by = serializers.HiddenField(default=CurrentUserProfileDefault())
Bodor Máté's avatar
Bodor Máté committed
    created_by_name = serializers.SerializerMethodField()

    class Meta:
        model = models.Note
        fields = '__all__'
        read_only_fields = ('created_at', 'update_at', 'created_by')
Bodor Máté's avatar
Bodor Máté committed

    def get_creted_by_name(self, obj):
        return obj.created_by.full_name