diff --git a/program/serializers.py b/program/serializers.py index c1c8496269044f7bba3dea400431517c2b23ca99..4eb950c3d91bc7f778075f59efb441d1a6a4cfb7 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