Add Friendship, FriendshipRequest Serializer, ViewSet, routeURL

JoinTheCityREST/JoinTheCityREST/urls.py

@@ -8,11 +8,14 @@ router = routers.DefaultRouter()
 from app.API import views as api_views
 from app.Discussion import views as discussion_views
 from app.Event import views as event_views
+from app.Friendship import views as friendship_views
 
 router.register(r'api', api_views.APIViewSet)
 router.register(r'discussion', discussion_views.DiscussionViewSet)
 router.register(r'message', discussion_views.MessageViewSet)
 router.register(r'event', event_views.EventViewSet)
+router.register(r'friendship', friendship_views.FriendshipViewSet)
+router.register(r'friendship_request', friendship_views.FriendshipRequestViewSet)
 
 
 urlpatterns = patterns('',

JoinTheCityREST/app/Friendship/serializers.py

+from rest_framework import serializers
+from app.Friendship.models import Friendship, FriendshipRequest
+
+
+class FriendshipSerializer(serializers.HyperlinkedModelSerializer):
+    class Meta:
+        model = Friendship
+        fields = ()
+
+class FriendshipRequestSerializer(serializers.HyperlinkedModelSerializer):
+    class Meta:
+        model = FriendshipRequest
+        fields = ()
\ No newline at end of file

JoinTheCityREST/app/Friendship/views.py

-from django.shortcuts import render
-
 # Create your views here.
+from rest_framework import viewsets, permissions
+from app.Friendship.models import Friendship, FriendshipRequest
+from app.Friendship.serializers import FriendshipSerializer, FriendshipRequestSerializer
+
+
+class FriendshipViewSet(viewsets.ModelViewSet):
+    """
+    API for articles
+    """
+    queryset = Friendship.objects.all()
+    serializer_class = FriendshipSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
+
+class FriendshipRequestViewSet(viewsets.ModelViewSet):
+    """
+    API for articles
+    """
+    queryset = FriendshipRequest.objects.all()
+    serializer_class = FriendshipRequestSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
\ No newline at end of file