36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from flask import Blueprint,render_template,request,redirect,url_for,session
|
|
from werkzeug.security import check_password_hash
|
|
|
|
from database import WebUser
|
|
|
|
auth = Blueprint('auth',__name__)
|
|
|
|
@auth.route("/")
|
|
@auth.route("/login",methods=["GET"])
|
|
def login_template():
|
|
return render_template("login.html")
|
|
|
|
@auth.route("/login",methods=["POST"])
|
|
def login():
|
|
username = request.form.get("username")
|
|
password = request.form.get("password")
|
|
|
|
if (not username) or (not password):
|
|
return "Enter username/password"
|
|
|
|
# TODO: Write simple frontened (if username is None or password is None)
|
|
if (WebUser.userExists(username)):
|
|
user = WebUser.get(WebUser.username == username)
|
|
|
|
if ( check_password_hash(user.password_hash,password)):
|
|
session["username"] = username
|
|
return redirect(url_for("admin.admin_page"))
|
|
else:
|
|
return "Password not valid"
|
|
return "User not exist"
|
|
|
|
@auth.route("/logout")
|
|
def logout():
|
|
# Logout from account
|
|
return redirect(url_for("auth.login"))
|