Add links to abstractcreativeworks

CenterForOpenScience · Jul 10, 2016 · 5fef1bd · 5fef1bd
1 parent f613d7f
commit 5fef1bd
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/api/urls.py b/api/urls.py
@@ -19,6 +19,7 @@
 router.register(r'funders', views.FunderViewSet, base_name=views.FunderViewSet.serializer_class.Meta.model._meta.model_name)
 router.register(r'awards', views.AwardViewSet, base_name=views.AwardViewSet.serializer_class.Meta.model._meta.model_name)
 router.register(r'tags', views.TagViewSet, base_name=views.TagViewSet.serializer_class.Meta.model._meta.model_name)
+router.register(r'links', views.LinkViewSet, base_name=views.LinkViewSet.serializer_class.Meta.model._meta.model_name)
 router.register(r'creativeworks', views.CreativeWorkViewSet, base_name=views.CreativeWorkViewSet.serializer_class.Meta.model._meta.model_name)
 router.register(r'preprints', views.PreprintViewSet, base_name=views.PreprintViewSet.serializer_class.Meta.model._meta.model_name)
 router.register(r'publications', views.PublicationViewSet, base_name=views.PublicationViewSet.serializer_class.Meta.model._meta.model_name)

diff --git a/api/views/share.py b/api/views/share.py
@@ -114,6 +114,11 @@ class TagViewSet(ShareObjectViewSet):
     queryset = serializer_class.Meta.model.objects.all().select_related('extra')
 
 
+class LinkViewSet(ShareObjectViewSet):
+    serializer_class = serializers.LinkSerializer
+    queryset = serializer_class.Meta.model.objects.all().select_related('extra')
+
+
 class CreativeWorkViewSet(ShareObjectViewSet):
     serializer_class = serializers.CreativeWorkSerializer
     queryset = serializer_class.Meta.model.objects.all().select_related(

diff --git a/share/serializers.py b/share/serializers.py
@@ -43,6 +43,7 @@ def __init__(self, *args, **kwargs):
     class Meta:
         pass
 
+
 class ExtraDataSerializer(BaseShareSerializer):
     class Meta(BaseShareSerializer.Meta):
         model = models.ExtraData
@@ -58,6 +59,11 @@ class Meta(BaseShareSerializer.Meta):
         model = models.Venue
 
 
+class LinkSerializer(BaseShareSerializer):
+    class Meta(BaseShareSerializer.Meta):
+        model = models.Link
+
+
 class OrganizationSerializer(EntitySerializer):
     class Meta(EntitySerializer.Meta):
         model = models.Organization
@@ -133,6 +139,7 @@ class AbstractCreativeWorkSerializer(BaseShareSerializer):
     institutions = InstitutionSerializer(sparse=True, many=True)
     venues = VenueSerializer(sparse=True, many=True)
     awards = AwardSerializer(sparse=True, many=True)
+    links = LinkSerializer(many=True)
     subject = TagSerializer(sparse=True)