Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def create(self, request):
- request.data['owner'] = request.user.id
- request.data['created'] = datetime.now()
- serializer = self.get_serializer(data=request.data)
- if serializer.is_valid():
- d = serializer.validated_data
- existing = Objective.objects.filter(owner__id=self.request.user.id,
- year=d['year'],
- quarter=d['quarter'])
- num_existing = existing.count()
- # make sure the user doesn't already have 5 objectives
- if (num_existing >= 5):
- return Response(
- 'User already has 5 objectives this quarter',
- status.HTTP_400_BAD_REQUEST
- )
- # set the Objective number as the next in line
- d['number'] = num_existing + 1
- e = serializer.save()
- return Response(serializers.ObjectiveSerializer(e).data)
- else:
- return Response(serializer.errors)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement