def get_appliances_for_logged_in_user(self):
email = self.session['email']
role = self.session['role']
appliances = []
stores = Store.all().filter(role + ' =', email)
for store in stores:
appliances.extend([appliance for appliance in Appliance.all().filter('store =', store).fetch(100)])
return appliances
def does_user_own_appliance(self):
email = self.session['email']
role = self.session['role']
stores = Store.all().filter(role + ' =', email)
appliance = Appliance.get_by_id(long(self['appliance']))
for store in stores:
if appliance.store.key().id() == store.key().id():
return True
return False
def store_mapper(member, role_name):
fields = member['fields']
store_obj = Store.all().filter('airtable_id =', fields['Store ID']).get()
if not store_obj:
store_obj = Store()
store_obj.name = fields['Store Name']
store_obj.location = db.GeoPt(fields['Latitude'],fields['Longitude'])
store_obj.address = fields['Store Address']
store_obj.billing_address = fields['Billing Address']
store_obj.owner = obj_map[fields['Store Owner'][0]].key().name()
store_obj.manager = obj_map[fields['Store Manager'][0]].key().name()
store_obj.airtable_id = fields['Store ID']
store_obj.put()
obj_map[member['id']] = store_obj
def get_work_orders_for_logged_in_user(self):
email = self.session['email']
workorders = []
if is_store_login(self):
appliances = []
stores = Store.all().filter(self.session['role'] + ' =', email)
for store in stores:
appliances.extend([appliance.id for appliance in Appliance.all().filter('store =', store).fetch(100)])
for appliance in appliances:
workorders.extend([wo for wo in WorkOrder.all().filter('appliance =', str(appliance))])
elif is_provider_login(self):
providers = [p for p in Provider.all().fetch(100) if p.owner.key().name() == email]
for provider in providers:
workorders.extend([wo for wo in WorkOrder.all().filter('provider =', str(provider.key().id()))])
return workorders
def get(self):
email = self.session["email"]
role = self.session["role"]
stores = Store.all().filter(role + " =", email).order("name")
if stores.count() > 1:
self.session["bottom_menu_1"] = "Stores"
elif stores.count() == 1:
self.session["bottom_menu_1"] = "Appliances"
store_appliances = []
for store in stores:
appliances = [appliance for appliance in Appliance.all().filter("store =", store).fetch(100)]
store_appliances.append((store, appliances))
path = "appliances.html"
template_values = {"store_appliances": store_appliances, "count": len(store_appliances)}
self.write(self.get_rendered_html(path, template_values), 200)
def index():
# Get a list of all stores from database
stores = Store.all()
# Get admin email address from environment variables
admin_email = os.getenv("ADMIN")
# Prepare a list of stores
for store in stores:
print(store.name)
# Present user with page containing list of stores with admin email in memory
return render_template('stores/index.html',
stores=stores,
admin=admin_email)
