def getRestaurants(self):
out = {}
r = RestaurantModel()
for r in RestaurantModel.all() :
if r.active :
out[r.key().id()] = r.name
return out
def add(self,
name,
city,
phone,
address,
state):
rm = RestaurantModel()
rm.author = users.get_current_user()
# date is inserted auto
rm.name = name
rm.city = city
rm.phone = phone
rm.address = address
rm.state = state
rm.active = False
rm.put()
def getRestaurantFromID(self,id):
rm = RestaurantModel()
for r in RestaurantModel.all() :
if r.key().id() == id :
rm = r
return rm
def getRestaurantsInProfile(self):
r = {}
p = UserProfileModel()
current_profile = p.get_by_key_name(users.get_current_user().user_id())
restaurant = RestaurantModel.gql("WHERE profiles = :1", current_profile.key())
for x in restaurant:
r[x.key().id()] = x.name
return r
def getRestaurantName(self):
r = {}
p = UserProfileModel()
current_profile = p.get_by_key_name(users.get_current_user().user_id())
restaurant = RestaurantModel.gql('WHERE profiles = :1', current_profile.key())
for x in restaurant:
r[x.key().id()] = x.name
return sorted(r.iteritems())
def getLatestRestaurants(self):
out = {}
#r = RestaurantModel()
for r in RestaurantModel.all() :
if r.active :
details = r.city + ' , ' + r.state
out[r.name.title()] = details.title()
return out
def addRestaurantNameId(self,id):
r = RestaurantModel.get_by_id(int(id))
r_name = r.name
rem = memcache.get("rem")
if rem is not None :
rem.current_model_edit_id = id
rem.current_model_name = r_name
if not memcache.add("rem",rem) :
memcache.replace("mc_error",True)
m = RestaurantEditModel.get_by_key_name(users.get_current_user().user_id())
m.current_model_edit_id = id
m.current_model_name = r_name
m.put()
def add(self,
name,
country,
city,
phone,
address,
state):
#success = True
default_restaurant = False
restaurant_id=0
try :
rm = RestaurantModel()
# if this is first restaurant added - make default
if not Restaurant.getRestaurantName(self) :
default_restaurant = True
rm.author = users.get_current_user()
# date is inserted auto
rm.name = name
rm.country = country
rm.city = city.strip().lower()
rm.phone = phone
rm.address = address
rm.state = state.strip().lower()
rm.active = False
rm.default_restaurant = default_restaurant
# add profile relationship
p = UserProfileModel()
current_profile = p.get_by_key_name(users.get_current_user().user_id())
admin_profile1 = p.get_by_key_name('114115181577730413318')
if current_profile.key() not in rm.profiles:
rm.profiles.append(current_profile.key())
if users.get_current_user().user_id() != '114115181577730413318' :
#add wiley.snyder
if(admin_profile1):
rm.profiles.append(admin_profile1.key())
rm.put()
restaurant_id = rm.key().id()
#create invite key
if users.is_current_user_admin():
try :
im = InvitationModel()
im.invitation_code = Restaurant.createInvitationKey(self,rm.key().id())
im.restaurant_key = str(rm.key())
im.put()
except :
#success = False
x = RestaurantModel.get_by_id(rm.key().id())
x.delete()
raise
except :
#success = False
raise
return restaurant_id
def getRestaurant(self,id):
return RestaurantModel.get_by_id(int(id))
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()
