def get(self, request): """ Generate a new favorite if the search in db return nothing. If the db return something this favorite is deleted. """ username = request.GET["username"] line_number = request.GET["line_number"] user = BusinemeUser.objects.get(username=username) busline = Busline.objects.get(line_number=line_number) try: favorite = Favorite.objects.get(user=user, busline=busline) favorite.delete() json_data = serialize(favorite) except Favorite.DoesNotExist: message = ReturnMessage() new_favorite = Favorite() new_favorite.user = user new_favorite.busline = busline new_favorite.save() json_data = message.return_message(STATUS_NOT_FOUND) return JsonResponse(json_data, content_type="application/json")
def get_user(self, user_id): """Return a user given a id""" try: user = BusinemeUser.objects.get(pk=user_id) json_data = serialize(user) except BusinemeUser.DoesNotExist: message = ReturnMessage() json_data = message.return_message(STATUS_NOT_FOUND) return JsonResponse(json_data, content_type='application/json')
def get_busline(self, busline_id): """ Obtains the required busline based in line's number. """ try: busline = Busline.objects.get(pk=busline_id) json_data = serialize(busline) except Busline.DoesNotExist: message = ReturnMessage() json_data = message.return_message(STATUS_NOT_FOUND) return JsonResponse(json_data, content_type="application/json")
def get_post(self, post_id): """ Obtains the required terminal based in line's number. """ try: post = Post.objects.get(pk=post_id) json_data = serialize(post) except Post.DoesNotExist: message = ReturnMessage() json_data = message.return_message(STATUS_NOT_FOUND) return JsonResponse(json_data, content_type="application/json")