class SavedPlacesApi(generics.ListCreateAPIView): """API for saving places""" queryset_hotels = Hotel.get_all_hotels() queryset_cafes = Cafe.get_all_cafes() hotels_serializer_class = HotelSerializer cafe_serializer_class = CafeSerializer def post(self, request): username = request.data['user_name'] place_name = request.data['place_name'] if SavedPlaces.save_place(username, place_name): return HttpResponse('Place was saved') else: return HttpResponse(status=status.HTTP_400_BAD_REQUEST)
class HotelApi(generics.ListCreateAPIView): """API to get hotels and save comments""" queryset = Hotel.get_all_hotels() serializer_class = HotelSerializer permission_classes = (permissions.AllowAny, ) def post(self, request): self.save_comment(request.data) return HttpResponse("OK") def save_comment(self, data): try: hotel = Hotel().get_object_by_name(data['hotel']) comment = Comment(hotel=hotel, author=data['author'], text=data['text']) comment.save() except Exception as exc: print("save_comment EXCEPTION: ", repr(exc))