Refactor ActSectionSerializer to use SerializerMethodField for act an…

…nouncement date
miskibin · Nov 2, 2024 · 6ffd0c3 · 6ffd0c3
1 parent 0b73424
commit 6ffd0c3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/api/views/vector_search.py b/src/api/views/vector_search.py
@@ -1,3 +1,4 @@
+from datetime import datetime
 from django.db.models import F
 from django.core.cache import cache
 from rest_framework.views import APIView
@@ -16,7 +17,7 @@
 class ActSectionSerializer(serializers.ModelSerializer):
     act_url = serializers.SerializerMethodField()
     act_title = serializers.CharField(source="act.title")
-    act_announcement_date = serializers.DateTimeField(source="act.announcementDate")
+    act_announcement_date = serializers.SerializerMethodField()
     similarity_score = serializers.FloatField()
 
     class Meta:
@@ -34,6 +35,9 @@ class Meta:
     def get_act_url(self, obj: ActSection):
         return obj.act.url
 
+    def get_act_announcement_date(self, obj: ActSection):
+        return str(obj.act.announcementDate)
+
 
 class VectorSearchView(APIView):
     CACHE_TIMEOUT = 3600