def login():
"""Log in page"""
name = request.form.get('name')
email = request.form.get('email')
image = request.form.get('image')
user = User.query.filter_by(email=email).first()
if user:
user_id = user.user_id
session['user_id']=user_id
session['name']=name
session['image']=image
return "%s is in session" %(name)
if not user:
User.add_user(email=email, name=name, image=image)
session['user_id']=user_id
session['name']=name
session['image']=image
return "%s is in session" %(name)
else:
return None
def test_add_user(self):
# Ensures that a food object is created in the database
# when the mush form is submitted
User.add_user("*****@*****.**", "test", "test", "test")
assert User.query.filter_by(email="*****@*****.**").one()
def thread(j):
from model import User
try:
User.get_all(i, j)
User.add_user(i, j)
User.get_all(i, j)
except Exception as exc:
logging.error(f'{i}-{j} {type(exc).__name__}:{exc.args}')
def facebook_login():
# Handles the login from the facebook login button
fb_user_id = request.form.get('fbUserId')
fb_fname = request.form.get('fbFname')
fb_lname = request.form.get('fbLname')
fb_email = request.form.get('fbEmail')
current_acces_token = request.form.get('accessToken')
fb_friends = request.form.get('fbFriends')
fb_friends = json.loads(fb_friends)
fb_user = User.query.filter_by(fb_id=fb_user_id).first()
if fb_user:
# User has previously logged into MLM
user_id = fb_user.user_id
session['user_id'] = user_id
session['current_acces_token'] = current_acces_token
# check friends list in friends table. If friendship not there, add it.
if fb_friends:
friends_user_ids = []
# turn the fb_ids into user_ids.
for friend_fb_id in fb_friends:
friend_user_id = db.session.query(User.user_id
).filter_by(
fb_id=friend_fb_id
).first()
friends_user_ids.append(friend_user_id)
friends_user_ids = [x[0] for x in friends_user_ids]
# now see if those friends are in the friendship table.
for friend in friends_user_ids:
friend_exists = db.session.query(Friendship.friend_id
).filter_by(friend_id=friend
).first()
# if they're not, add them in! Yay friendship!
if friend_exists is None:
Friendship.add_friendship(user_id, friend)
flash('Login successful!')
return redirect('/home')
else:
# First time for user logging into MLM
# add the user to the database
User.add_user(email=fb_email, fname=fb_fname,
lname=fb_lname, fb_id=fb_user_id)
# access that user's information, add it to the session
fb_user = User.query.filter_by(fb_id=fb_user_id).first()
user_id = fb_user.user_id
session['user_id'] = user_id
session['current_acces_token'] = current_acces_token
flash('Thanks for creating an account with Make Less Mush')
return redirect('/home')
def signup_portal():
"""Create new user and add the user to the database"""
email = request.form.get('email')
password = request.form.get('password')
age = request.form.get('age')
zipcode = request.form.get('zipcode')
User.add_user(email, password, age, zipcode)
return redirect("/login")
def account_feedback():
email = request.form.get('email')
password = request.form.get('password')
fname = request.form.get('fname')
lname = request.form.get('lname')
user = User.query.filter_by(email=email).first()
if user:
flash("Hi, %s, you already have an account"%fname)
return redirect("/")
else:
User.add_user( email=email,
password=password,
fname=fname,
lname=lname)
flash("Congrats %s! You've successfully created an account!\nYou can now log in."%fname)
return redirect("/")
def signup_portal():
# Handles the signup form
email = request.form.get('email')
password = request.form.get('password')
password = hash(password)
fname = request.form.get('fname')
fname = titlecase(fname)
lname = request.form.get('lname')
lname = titlecase(lname)
User.add_user(email, fname, lname, password)
# automatically sign in user after account creation
user = User.query.filter_by(email=email, password=password).first()
user_id = user.user_id
session['user_id'] = user_id
flash('Account successfully created. Welcome to Make Less Mush!')
return redirect('/home')
def add_user(self):
id_ = View.get_data('id')
login = View.get_data('login')
password = View.get_data('password')
levelOFPrioriti = View.get_data('levelOFPrioriti')
firstname = View.get_data('firstname')
lastname = View.get_data('lastname')
midname = View.get_data('midname')
passportId = View.get_data('passportId')
user = User.add_user(id_, firstname, lastname, midname, passportId,
login, password, levelOFPrioriti)
