def post(self): data = Hotel.arguments.parse_args() hotel = HotelModel(**data) if HotelModel.hotel_find_by_name(hotel.name): return {'message': 'Hotel already exists'}, 400 if SiteModel.find_site_by_id(hotel.site_id) is None: return {'message': 'Site not exists'}, 400 try: hotel.save_hotel() except: return { 'message': 'An internal error ocurred trying to save hotel' }, 500 return hotel.json(), 201
def get(self, id): site = SiteModel.find_site_by_id(id) if site: return site.json() return {'message': 'Site not found'}, 404
def delete(self, id): site = SiteModel.find_site_by_id(id) if site: site.delete_site() return {'message': 'Site deleted'}, 204 return {'message': 'Site not found'}, 404