From 1dbb64a38eccb3483aaeca8d99f7114116f26101 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bodor=20M=C3=A1t=C3=A9?= <bodor.mate@kszk.bme.hu>
Date: Fri, 4 Jan 2019 20:41:21 +0100
Subject: [PATCH] add visitor_number field to the serializer

---
 src/stats/serializers.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/stats/serializers.py b/src/stats/serializers.py
index 963445d..6841d4e 100644
--- a/src/stats/serializers.py
+++ b/src/stats/serializers.py
@@ -6,15 +6,19 @@ from . import models
 class EventSerializer(serializers.ModelSerializer):
     created_by = serializers.HiddenField(default=CurrentUserProfileDefault())
     created_by_name = serializers.SerializerMethodField()
+    visitor_number = serializers.SerializerMethodField()
 
     class Meta:
         model = models.Event
         fields = '__all__'
         read_only_fields = ('created_at', 'update_at', 'created_by')
 
-    def get_creted_by_name(self, obj):
+    def get_created_by_name(self, obj):
         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())
-- 
GitLab