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))