def handle_amenity_for_place(place_id, amenity_id): '''Add the amenity with `amenity_id` to the place with `place_id` with a POST request. Delete the amenity with the id of `amenity_id` with a DELETE request. Keyword arguments: place_id -- The id of the place. amenity_id -- The id of the amenity. ''' try: Amenity.select().where(Amenity.id == amenity_id).get() except Amenity.DoesNotExist: return jsonify(msg="Amenity does not exist."), 404 try: Place.select().where(Place.id == place_id).get() except Place.DoesNotExist: return jsonify(msg="Place does not exist."), 404 if request.method == 'POST': '''Save the connection in the ReviewPlace table.''' PlaceAmenities().create(place=place_id, amenity=amenity_id) return jsonify(msg="Amenity added to place successfully."), 201 elif request.method == 'DELETE': (PlaceAmenities.delete().where((PlaceAmenities.place == place_id) & ( PlaceAmenities.amenity == amenity_id)).execute()) Amenity.delete().where(Amenity.id == amenity_id).execute() return jsonify(msg="Amenity deleted successfully."), 200
def handle_amenity_for_place(place_id, amenity_id): '''Add the amenity with `amenity_id` to the place with `place_id` with a POST request. Delete the amenity with the id of `amenity_id` with a DELETE request. Keyword arguments: place_id -- The id of the place. amenity_id -- The id of the amenity. ''' try: Amenity.select().where(Amenity.id == amenity_id).get() except Amenity.DoesNotExist: return jsonify(msg="Amenity does not exist."), 404 try: Place.select().where(Place.id == place_id).get() except Place.DoesNotExist: return jsonify(msg="Place does not exist."), 404 if request.method == 'POST': '''Save the connection in the ReviewPlace table.''' PlaceAmenities().create(place=place_id, amenity=amenity_id) return jsonify(msg="Amenity added to place successfully."), 201 elif request.method == 'DELETE': (PlaceAmenities .delete() .where((PlaceAmenities.place == place_id) & (PlaceAmenities.amenity == amenity_id)) .execute()) Amenity.delete().where(Amenity.id == amenity_id).execute() return jsonify(msg="Amenity deleted successfully."), 200
def delete_amenity(amenity_id): """ Delete the given amenity Deletes the given amenity in the database. --- tags: - Amenity parameters: - in: path name: amenity_id type: string required: True description: ID of the amenity responses: 200: description: Amenity deleted successfully schema: id: delete_200 required: - code - msg properties: code: type: integer description: Response code from the API default: 200 msg: type: string description: Message about record deletion default: "deleted successfully" 404: description: Amenity was not found 500: description: Request could not be processed """ try: ''' Check if amenity exists ''' query = Amenity.select().where(Amenity.id == amenity_id) if not query.exists(): raise LookupError('amenity_id') ''' Delete the amenity ''' amenity = Amenity.delete().where(Amenity.id == amenity_id) amenity.execute() res = {} res['code'] = 200 res['msg'] = "Amenity was deleted successfully" return res, 200 except LookupError as e: abort(404) except Exception as e: abort(500)
def delete_amenity(amenity_id): """ Delete amenity with id as amenity_id """ try: amenity = Amenity.get(Amenity.id == amenity_id) except Exception: return {'code': 404, 'msg': 'Amenity not found'}, 404 amenity = Amenity.delete().where(Amenity.id == amenity_id) amenity.execute() res = {} res['code'] = 201 res['msg'] = "Amenity was deleted successfully" return res, 201
def handle_amenity_id(amenity_id): '''Returns a JSON object of the amenity with the id passed as parameter with a GET request method. Removes an amenity with DELETE request method. Keyword arguments: amenity_id: The id of the amenity. ''' try: amenity = Amenity.select().where(Amenity.id == amenity_id).get() except Amenity.DoesNotExist: raise Exception("There is no amenity with this id.") if request.method == 'GET': return jsonify(amenity.to_dict()), 200 elif request.method == 'DELETE': amenity = Amenity.delete().where(Amenity.id == amenity_id) amenity.execute() return jsonify(msg="Amenity deleted successfully."), 200
def handle_amenity_id(amenity_id): '''Returns a JSON object of the amenity with the id passed as parameter with a GET request method. Removes an amenity with DELETE request method. Keyword arguments: amenity_id: The id of the amenity. ''' try: amenity = Amenity.select().where(Amenity.id == amenity_id).get() except Amenity.DoesNotExist: raise Exception("There is no amenity with this id.") if request.method == 'GET': return jsonify(amenity.to_dict()), 200 elif request.method == 'DELETE': amenity = Amenity.delete().where(Amenity.id == amenity_id) amenity.execute() return jsonify(msg="Amenity deleted successfully."), 200
def setUp(self): #connect to db and delete everyone in the amenities table Amenity.delete().execute()
def tearDown(self): #delete all from amenities Amenity.delete().execute()