Add LinkSerializer and modify HostSerializer to handle nested links

program/serializers.py

@@ -26,7 +26,7 @@ from rest_framework import serializers
 from profile.models import Profile
 from profile.serializers import ProfileSerializer
 from program.models import Show, Schedule, TimeSlot, Category, FundingCategory, Host, Topic, MusicFocus, Note, Type, Language, \
-    RRule
+    RRule, Link
 from steering.settings import THUMBNAIL_SIZES
 
 
@@ -111,7 +111,14 @@ class CategorySerializer(serializers.ModelSerializer):
         return instance
 
 
+class LinkSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Link
+        fields = ('description', 'url')
+
+
 class HostSerializer(serializers.ModelSerializer):
+    links = LinkSerializer(many=True)
     thumbnails = serializers.SerializerMethodField()  # Read-only
 
     @staticmethod
@@ -129,6 +136,15 @@ class HostSerializer(serializers.ModelSerializer):
         model = Host
         fields = '__all__'
 
+    def create(self, validated_data):
+        links_data = validated_data.pop('links')
+        host = Host.objects.create(**validated_data)
+
+        for link_data in links_data:
+            Link.objects.create(host=host, **link_data)
+        host.save()
+        return host
+
     def update(self, instance, validated_data):
         """
         Update and return an existing Host instance, given the validated data.
@@ -142,7 +158,14 @@ class HostSerializer(serializers.ModelSerializer):
         instance.image = validated_data.get('image', instance.image)
         instance.ppoi = validated_data.get('ppoi', instance.ppoi)
 
+        for link in instance.links.all():
+            link.delete(keep_parents=True)
+
+        for link_data in validated_data.get('links'):
+            Link.objects.create(host=instance, **link_data)
+
         instance.save()
+
         return instance