From 4a806428640a76a27b5678511d9bbffaefd60549 Mon Sep 17 00:00:00 2001
From: Ernesto Rico Schmidt <ernesto@helsinki.at>
Date: Thu, 8 Jun 2023 16:21:44 -0400
Subject: [PATCH] fix: add note_id to the Timeslot serializer

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

diff --git a/program/serializers.py b/program/serializers.py
index c1c84962..4eb950c3 100644
--- a/program/serializers.py
+++ b/program/serializers.py
@@ -700,6 +700,9 @@ class ScheduleDryRunResponseSerializer(serializers.Serializer):
 
 
 class TimeSlotSerializer(serializers.ModelSerializer):
+    note_id = serializers.PrimaryKeyRelatedField(
+        allow_null=True, queryset=Note.objects.all(), required=False
+    )
     show_id = serializers.PrimaryKeyRelatedField(queryset=Show.objects.all(), required=False)
     schedule_id = serializers.PrimaryKeyRelatedField(
         queryset=Schedule.objects.all(), required=False
@@ -721,6 +724,7 @@ class TimeSlotSerializer(serializers.ModelSerializer):
         )
         fields = (
             "memo",
+            "note_id",
             "playlist_id",
             "repetition_of_id",
         ) + read_only_fields
@@ -730,7 +734,7 @@ class TimeSlotSerializer(serializers.ModelSerializer):
 
         # Only save certain fields
         instance.memo = validated_data.get("memo", instance.memo)
-        instance.repetition_of = validated_data.get("repetition_of", instance.repetition_of_id)
+        instance.repetition_of = validated_data.get("repetition_of_id", instance.repetition_of)
         instance.playlist_id = validated_data.get("playlist_id", instance.playlist_id)
         instance.save()
         return instance
-- 
GitLab