def get_restaurant_page(request):
"""retrieve restaurant by id"""
_id = request.GET.get('_id')
restaurant = Restaurant.get(_id)
if restaurant:
return JsonResponse(model_to_json(restaurant))
else:
return JsonResponse({})
def edit_restaurant_page(request):
"""Update restaurant data"""
validate(instance=request.body, schema=restaurant_schema)
body = json.loads(request.body)
invalid = Restaurant.field_validate(body)
if invalid: # exit if invalid body
return JsonResponse(invalid)
restaurant = Restaurant.get(body["restaurant_id"])
edit_model(restaurant, body, restaurant_editable)
if address_changed(body):
update_model_geo(restaurant, body['address'])
restaurant = save_and_clean(restaurant)
return JsonResponse(model_to_json(restaurant))
def get(self, request, rest_id):
""" Retrieve approved restaurant by id """
restaurant = Restaurant.get(rest_id)
restaurant = model_to_json(restaurant)
return JsonResponse(restaurant)
