Mae/groupsvoteroute

api/optimeet/groups/models.py

@@ -32,3 +32,8 @@ class Recommendations(models.Model):
     place_url = models.CharField(max_length=100)
     times = models.JSONField()
 
+class Votes(models.Model):
+    group_id = models.ForeignKey(Group, related_name='group_votes', on_delete=models.CASCADE)
+    user_id = models.CharField(max_length=64, blank=True)
+    results_of_voted = models.CharField(max_length=64, blank=True)
+

api/optimeet/groups/serializers.py

@@ -1,6 +1,6 @@
 from rest_framework import serializers
 from . import models
-from .models import UserGroup
+from .models import UserGroup, Votes
 from random import randint
 
 def rnd_id(): return randint(1000000000,9999999999)
@@ -30,3 +30,7 @@ class Meta:
         model = models.Recommendations
         fields = '__all__'
 
+class VotesSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Votes
+        fields = '__all__'

api/optimeet/groups/urls.py

@@ -10,6 +10,8 @@
     path('<str:group_id>/users/', views.add_users_to_group, name='add_users_to_group'),
     #path('usergroups/', views.UserGroupListAPIView.as_view(), name='usergroup-list'),#get all usergroup
     #path('usergroups/<str:user_id>/', views.UserGroupListAPIView.as_view(), name='usergroup-byuser'),#get usergroup by user_id
-    path("<int:group_id>/recs", views.get_recommendation)
+    path("<int:group_id>/recs", views.get_recommendation),
+    path("<int:group_id>/votes", views.votes, name='votes'),
+    path("<int:group_id>/votes", views.create_vote, name='create_vote'),
 
 ]

api/optimeet/groups/views.py

@@ -63,6 +63,29 @@ def add_users_to_group(request, group_id):
 
     return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
 
+@csrf_exempt
+@api_view(['GET'])  
+def votes(request, group_id):
+        votes = models.Votes.objects.filter(group_id=group_id)
+        serializer = serializers.VotesSerializer(votes, many=True)
+        return Response(serializer.data)
+
+@csrf_exempt
+@api_view(['POST'])
+def create_vote(request, group_id):
+
+    try:
+        group = models.Group.objects.get(group_id=group_id)
+    except models.Group.DoesNotExist:
+        raise Http404  
+
+    request.data['group_id'] = group_id
+    serializer = serializers.VotesSerializer(data=request.data, context={'group_id': group_id})
+    if serializer.is_valid():
+        serializer.save()
+        return Response(status=status.HTTP_201_CREATED)
+
+    return Response(status=status.HTTP_400_BAD_REQUEST)    
 # #for testing
 # class UserGroupListAPIView(APIView):
 #     def get(request, self):