def register():
if request.method == 'GET':
return render_template("register.html")
username = request.form['username']
password = request.form['password']
new_account = UserAccounts(user_name=username, password=password, mugshot="default.jpg")
db.session.add(new_account)
db.session.commit()
return redirect(url_for("index"))
def register():
isLogin = False
if request.method == 'GET':
return render_template("register.html")
username = request.form['username']
password = request.form['password']
role = True if request.form['role'] == "host" else False
mugshot = "default_{}.png".format(str(random.randint(1,9)))
new_account = UserAccounts(user_name=username, password=password, mugshot=mugshot, role=role)
db.session.add(new_account)
db.session.commit()
return redirect(url_for("home"))
def login():
user_id = session.get('user_id')
if request.method == 'GET':
return render_template("login.html")
if current_user.is_authenticated and query_user(user_id):
return redirect(url_for('index'))
username = request.form['username']
user = UserAccounts.query.filter_by(UserName=username).first()
if not user:
return render_template("login.html", error="username or password error")
pw_form = UserAccounts.psw_to_md5(request.form['password'])
pw_db = user.Password
if pw_form == pw_db:
user = User()
user.id = username
login_user(user, remember=True)
flash('Logged in successfully')
return redirect(url_for('index'))
return render_template("login.html", error="username or password error")