def business_add_item(bzName):
bzid = db_get_business_id(bzName)
bzName = db.child("Businesses").child(bzid).child("business").get()
print(bzid)
if request.method == 'POST':
if request.form['submit'] == 'add':
name = request.form['name']
price = request.form['price']
description = request.form['description']
newItem = {
"name": name,
"price": price,
"description": description
}
db.child("Businesses").child(bzid).child("items").push(newItem)
items = db.child("Businesses").child(bzid).child("items").get()
#elif request.form['submit']=='delete':
return redirect(url_for('business_add_item'))
items = db.child("Businesses").child(bzid).child("items").get()
if items.val() == None:
return render_template('itemsDisplay.html',
title='Business Items',
bzName=bzName)
return render_template('itemsDisplay.html',
title='Business Items',
items=items,
bzName=bzName)
def validate_id(business, uid):
if (business):
all_users = db.child("Businesses").get()
else:
all_users = db.child("Users").get()
for user in all_users.each():
if (uid == user.key()):
return True
return False
def validate_email(email, business):
if business:
all_users = db.child("Businesses").get()
else:
all_users = db.child("Users").get()
for user in all_users.each():
if (email == user.val().get("email")):
return False
return True
def cusotmer_view_business():
bid = '-temp' #switch to currently selected business later#####################################
bz = db.child("Businesses").child(bid)
items = bz.child("items").get()
bzName = db.child("Businesses").child("-temp").child("business").get()
return render_template('userViewBZ.html',
title='Viewing Business',
items=items,
bzName=bzName)
def search_add(name):
name_str = str(name)
user_id = current_user.get_id()
businesses_list = db.child("Businesses").get()
for bl in businesses_list:
if (bl.val().get("business") == name_str):
key_value = bl.key()
data = {name_str: key_value}
db.child("Users").child(user_id).child("followingBZ").update(data)
return render_template('searchDisplay.html', after=True)
def validate_password(email, business, password):
if business:
all_users = db.child("Businesses").get()
else:
all_users = db.child("Users").get()
for user in all_users.each():
if (email == user.val().get("email")):
if (bcrypt.check_password_hash(user.val().get("password"),
password)):
return True
return False
def business_posts(businessname):
userType = typeofUser()
user_id = current_user.get_id()
bzPosts = db.child(userType).child(user_id).child("bzPost").get()
businessname = db.child(userType).child(user_id).child("business").get()
if bzPosts.val() == None:
return render_template('businessPost.html',
title='Business Post Feed',
businessname=db_get_business_name(businessname))
return render_template('businessPost.html',
title='Business Post Feed',
bzPosts=bzPosts,
businessname=db_get_business_name(businessname))
def customer_main():
userType = typeofUser()
user_id = current_user.get_id()
userName = db.child(userType).child(user_id).child("username").get()
mainFeed = db.child(userType).child(user_id).child("followingBZ").get()
if mainFeed.val() == None:
return render_template('customerMain.html',
title='Customer Feed',
userName=userName)
posts = []
for p in mainFeed:
currPost = db.child("Businesses").child(p.val()).child("bzPost").get()
if currPost.val() is not None:
for cp in currPost:
posts.append(cp)
return render_template('customerMain.html',
title='Customer Feed',
posts=posts,
userName=userName)
def login():
if (current_user.is_authenticated):
return redirect(url_for('index'))
form = LoginForm()
if form.validate_on_submit():
if (form.business.data):
if (not validate_email(form.email.data, True)
and validate_password(form.email.data, True,
form.password.data)):
all_users = db.child("Businesses").get()
for gc in all_users.each():
if (form.email.data == gc.val().get("email")):
user = Business(uid=gc.key(),
username=gc.val().get('business'),
businessname=gc.val().get('business'),
email=gc.val().get('email'),
business=True)
login_user(user)
print(gc.val().get('businessname'))
flash(f'Account successfully logged in!', 'success')
return redirect(
url_for('business_main',
businessname=gc.val().get('business')))
else:
flash(f'Email or password are wrong', 'danger')
else:
if (not validate_email(form.email.data, False)
and validate_password(form.email.data, False,
form.password.data)):
all_users = db.child("Users").get()
for gc in all_users.each():
if (form.email.data == gc.val().get("email")):
user = User(uid=gc.key(),
username=gc.val().get('username'),
email=gc.val().get('email'),
business=False)
login_user(user)
flash(f'Account successfully logged in! ', 'success')
return customer_main()
else:
flash(f'Email or password are wrong', 'danger')
return render_template('login.html', title='Login', form=form)
def search_food():
business_name = str(request.args.get('search')).lower()
list_business_search = []
business_query = db.child("Businesses").get()
for bq in business_query:
temp = str(bq.val().get("business"))
if (business_name in temp.lower()):
list_business_search.append(temp)
return render_template('searchDisplay.html',
business_list=list_business_search)
def customer_cart():
userType = typeofUser()
user_id = current_user.get_id()
cart_items = db.child(userType).child(user_id).child("cart").get()
if cart_items.val() == None:
return render_template('customerCart.html',
title='Customer',
cart_items=cart_items)
return render_template('customerCart.html',
title='Customer Cart',
cart_items=cart_items)
def register():
if (current_user.is_authenticated):
return redirect(url_for('index'))
form = RegistrationForm()
if form.validate_on_submit():
data = {
"username":
form.username.data,
"email":
form.email.data,
"password":
bcrypt.generate_password_hash(form.password.data).decode('utf-8')
}
if (validate_email(form.email.data, False)):
db.child("Users").push(data)
flash(f'Account successfully created', 'success')
return redirect(url_for('index'))
else:
flash(f'Email already exists!', 'danger')
return render_template('register.html', title='Register', form=form)
def business_main(businessname):
userType = typeofUser()
user_id = current_user.get_id()
analytics = db.child(userType).child(user_id).child("analytics").get()
if (validate_id(True, user_id)):
if analytics.val() == None:
return render_template(
'businessPost.html',
title='Business Post Feed',
businessname=db_get_business_name(businessname))
return render_template('businessMain.html',
title='Business Analytics',
analytics=analytics,
businessname=db_get_business_name(businessname))
def register_business():
if current_user.is_authenticated:
return redirect(url_for('index'))
form = BusinessForm()
if form.validate_on_submit():
data = {
"business":
form.businessname.data,
"email":
form.email.data,
"password":
bcrypt.generate_password_hash(form.password.data).decode('utf-8'),
"bzPost":
""
}
if (validate_email(form.email.data, True)):
db.child("Businesses").push(data)
flash(f'Account successfully created', 'success')
return redirect(url_for('index'))
else:
flash(f'Email already exists!', 'danger')
return render_template('register_business.html',
title='Business Registration',
form=form)
def load_user(user_id):
#c_u = auth.current_user['idToken']
# print(findBusiness(user_id))
current_user_data= db.child(findBusiness(user_id)).order_by_key().equal_to(user_id).limit_to_first(1).get()
# print(findBusiness(user_id))
if findBusiness(user_id) == "Users":
return User(uid = user_id,
username = current_user_data.val()[user_id].get("username"),
email = current_user_data.val()[user_id].get("email"),
business = False)
else:
#print("RETURNED A BUSINESS!")
return Business(uid= user_id,
username = current_user_data.val()[user_id].get("businessname"),
businessname = current_user_data.val()[user_id].get("businessname"),
email = current_user_data.val()[user_id].get("email"),
business = True)
def db_get_business_id(businessname):
all_users = db.child("Businesses").get()
for user in all_users.each():
if(businessname == user.val().get("business")):
return user.key()
def findBusiness(user_id):
userType = ["Users", "Businesses"]
for x in userType:
# print(db.child(x).order_by_key().equal_to(user_id).limit_to_first(1).get().val())
if db.child(x).order_by_key().equal_to(user_id).limit_to_first(1).get().val() is not None:
return x
