Fix serializer

CSSAInformationDepartment · Nov 20, 2021 · 2e53052 · 2e53052
1 parent cdf7f86
commit 2e53052
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/EventAPI/serializers.py b/src/EventAPI/serializers.py
@@ -1,8 +1,11 @@
-from rest_framework import serializers
+from rest_framework import fields, serializers
 from EventAPI.models import Event as eventModels
 
 class EventsSerializer(serializers.ModelSerializer):
 
+    eventTaker = fields.SerializerMethodField()
+    eventType = fields.SerializerMethodField()
+
     def get_eventTaker(self, instance):
         return instance.eventBy.eventTakerName if instance.eventBy else None
 
@@ -12,5 +15,8 @@ def get_eventType(self, instance):
     class Meta:
       model = eventModels
       fields = ('eventID','eventName','eventInfo','mainVisual','address','venue','isFree','price',
-                'pastEventLink', 'pastEventPoster', 'recentEventLink', 'recentEventPoster',)
+                'pastEventLink', 'pastEventPoster', 'recentEventLink', 'recentEventPoster',
+                # extra fields
+                'eventTaker', 'eventType',
+                )