def post(self): argumentos = reqparse.RequestParser() argumentos.add_argument( 'nome', type=str, required=True, help="O campo 'nome' não pode ser deixado em branco") argumentos.add_argument( 'estrelas', type=float, required=True, help="O campo 'estrelas' não pode ser deixado em branco") argumentos.add_argument( 'diaria', type=float, required=True, help="O campo 'diaria' não pode ser deixado em branco") argumentos.add_argument( 'cidade', type=str, required=True, help="O campo 'cidade' não pode ser deixado em branco") dados = argumentos.parse_args() obj = HotelModel(**dados) hotel = obj.toJson() returnObj = HotelDao.save(hotel) return "Hotel salvo com id: " + str(returnObj.inserted_id), 201
def put(self, id): dados = Hotel.argumentos.parse_args() novoHotelObj = HotelModel(**dados) novoHotel = novoHotelObj.toJson() try: hotel = HotelDao.findById(id) except: return None, 404 if hotel: hotel.update(novoHotel) print(hotel) HotelDao.update(hotel) return novoHotel, 200 return None, 404
def post(self, hotel_id): if HotelModel.find_hotel(hotel_id): return {"message": f"Hotel ID ({hotel_id}) already exists!"} dados = Hotel.arguments.parse_args() if not SiteModel.find_by_id(dados.get('site_id')): return {"message": "Site ID not found!"}, 404 obj_hotel = HotelModel(hotel_id, **dados) try: obj_hotel.save_hotel() except: return { "message": "An internal error ocurred trying to save hotel." }, 500 return obj_hotel.toJson(), 200
def put(self, hotel_id): dados = Hotel.arguments.parse_args() obj_hotel = HotelModel.find_hotel(hotel_id) if obj_hotel: obj_hotel.update_hotel(**dados) try: obj_hotel.save_hotel() except: return { "message": "An internal error ocurred trying to save hotel." }, 500 return obj_hotel.toJson(), 200 obj_hotel = HotelModel(hotel_id, **dados) try: obj_hotel.save_hotel() except: return { "message": "An internal error ocurred trying to save hotel." }, 500 return obj_hotel.toJson(), 201