def getRestaurantsByCity(self,id):
out = {}
c = CityModel.get_by_id(int(id))
if c :
q = c.restaurants
for r in q:
if r.active :
out[r.key().id()] = r.name
return out
def delete(self,id):
#delete restaurant
menu_delete = []
category_delete = []
item_delete = []
rm = RestaurantModel.get_by_id(int(id))
if rm :
menus = rm.menus
if menus :
for x in menus :
#logging.info('delete menu : ' + str( x.key().id()))
mm = MenuModel.get_by_id(x.key().id())
menu_delete.append(x.key().id())
menucategory = mm.categories
if menucategory :
for y in menucategory :
#logging.info('delete category : ' + str(y.key().id()))
mcm = MenuCategoryModel.get_by_id(y.key().id())
category_delete.append(y.key().id())
items = mcm.items
if items :
for z in items :
#logging.info('delete item : ' + str(z.key().id()))
item_delete.append(z.key().id())
for menu in menu_delete :
#logging.info('delete menu : ' + str(menu))
mm = MenuModel.get_by_id(menu)
mm.delete()
for category in category_delete :
#logging.info('delete category : ' + str(category))
mcm = MenuCategoryModel.get_by_id(category)
mcm.delete()
for item in item_delete :
#logging.info('delete item : ' + str(item))
mim = MenuItemModel.get_by_id(item)
mim.delete()
#invites
rKey = rm.key()
for i in InvitationModel.all():
if str(i.restaurant_key) == str(rKey) :
i.delete()
#country
if rm.active :
cm = CountryModel.get_by_id(int(rm.country_id))
rc = cm.restaurant_count
if not str(rc) == '1' :
cm.restaurant_count = rc - 1
else :
cm.restaurant_count = rc - 1
cm.active = False
cm.put()
#city
if rm.active :
ci = CityModel.get_by_id(int(rm.city_id))
rc = ci.restaurant_count
if not str(rc) == '1' :
ci.restaurant_count = rc - 1
else :
ci.restaurant_count = rc - 1
ci.active = False
ci.put()
#state
if rm.active :
st = StateModel.get_by_id(int(rm.state_id))
rc = st.restaurant_count
if not str(rc) == '1' :
st.restaurant_count = rc - 1
else :
st.restaurant_count = rc - 1
st.active = False
st.put()
rm.delete()
def addCityModel(self,rm_id,cm_id):
rm = RestaurantModel.get_by_id(rm_id)
cm = CityModel.get_by_id(cm_id)
rm.citymodel = cm
rm.put()
