def put(self, hotel_id): dados = Hotel.atributos.parse_args() hotel_encontrado = HotelModel.encontrar_hotel(hotel_id) if hotel_encontrado: hotel_encontrado.atualizar_hotel(**dados) hotel_encontrado.salvar_hotel() return hotel_encontrado.json(), 200 hotel = HotelModel(hotel_id, **dados) try: hotel.salvar_hotel() except: return {'message': 'Ocorreu um erro ao salvar o hotel.'}, 500 return hotel.json(), 201
def post(self, hotel_id): if HotelModel.encontrar_hotel(hotel_id): return {'message': f'Hotel com id {hotel_id} não encontrado.'}, 400 dados = Hotel.atributos.parse_args() hotel = HotelModel(hotel_id, **dados) if not SiteModel.encontrar_por_id(dados['site_id']): return { 'message': 'O hotel precisa estar associado a um site id.' }, 400 try: hotel.salvar_hotel() except: return {'message': 'Ocorreu um erro ao salvar o hotel.'}, 500 return hotel.json(), 201