def add_review(id):
"""get the id from the url"""
business_found = Business.find_business_id(id)
description = str(request.data.get('description', ''))
if not business_found:
response = jsonify({
"message": "cannot add review to business that does not exist",
"status_code": 404
})
response.status_code = 404
return response
if description:
"""create review object"""
new_review = Review.save_review(description, id)
response = jsonify({"review": new_review, "status_code": 201})
response.status_code = 201
return response
else:
response = jsonify({
"message": "enter all details",
"status_code": 400
})
response.status_code = 400
return response
def test_find_business_by_id(self):
self.business.save_business(self.business.name,
self.business.description,
self.business.location,
self.business.contact)
found_business = Business.find_business_id(self.business.id)
self.assertEqual(found_business[0]["name"], self.business.name)
def get_reviews(id):
business_found = Business.find_business_id(id)
reviews = Review.business_reviews(id)
if not business_found:
response = jsonify({
"message": "cannot add review to business that does not exist",
"status_code": 404
})
response.status_code = 404
return response
response = jsonify({"reviews": reviews})
response.status_code = 201
return response
def business_manipulation(id):
"""gets the id of the business"""
"""uses the id to get a single business"""
business_found = Business.find_business_id(id)
if request.method == "GET":
if business_found:
response = jsonify({"business": business_found})
response.status_code = 200
return response
else:
response = jsonify({
"message": "business does not exist",
"status": 404
})
response.status_code = 404
return response
elif request.method == "PUT":
if business_found:
name = str(request.data.get('name', ''))
description = str(request.data.get('description', ''))
location = str(request.data.get('location', ''))
contact = str(request.data.get('contact', ''))
business_found[0]["name"] = name
business_found[0]["description"] = description
business_found[0]["location"] = location
business_found[0]["contact"] = contact
response = jsonify({"business": business_found})
response.status_code = 200
return response
else:
response = jsonify({
"message": "Cannot update business that does not exist",
"status": 404
})
response.status_code = 404
return response
else:
if business_found:
businesses = Business.get_all_businesses()
businesses.remove(business_found[0])
response = jsonify({
"business": "business successfully deleted",
"status": 200
})
response.status_code = 200
return response
else:
response = jsonify({
"message": "Cannot delete business that does not exist",
"status": 404
})
response.status_code = 404
return response