def get_actions(self): actions = {} if can_vote_choice(self.request): actions['vote'] = Action(method='POST', attributes=None) return actions
def post(self, request, *args, **kwargs): if not can_vote_choice(self.request): return self.http_method_not_allowed(request) self.get_object().vote() response = self.get(request) response.status_code = 201 return response
def post(self, request, *args, **kwargs): if not can_vote_choice(self.request): return self.http_method_not_allowed(request) try: choice = self.get_object() except self.model.DoesNotExist: raise Http404('Choice does not exist') choice.vote() response = self.get(request) response.status_code = 201 return response