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()