83 lines
3.2 KiB
Python
83 lines
3.2 KiB
Python
from flask import Blueprint, render_template, request, redirect, url_for
|
|
from flask_login import login_user, logout_user, login_required, UserMixin
|
|
from werkzeug.security import generate_password_hash, check_password_hash
|
|
from extensions import db, login_manager
|
|
|
|
auth = Blueprint('auth', __name__)
|
|
|
|
class User(UserMixin):
|
|
def __init__(self, id, username, password_hash, created_at):
|
|
self.id = id
|
|
self.username = username
|
|
self.password_hash = password_hash
|
|
self.created_at = created_at
|
|
|
|
@staticmethod
|
|
def get(user_id):
|
|
user_data = db.get_user(int(user_id))
|
|
|
|
if user_data:
|
|
return User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'])
|
|
return None
|
|
|
|
@login_manager.user_loader
|
|
def load_user(user_id):
|
|
user_data = db.get_user(int(user_id))
|
|
if user_data:
|
|
return User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'])
|
|
return None
|
|
|
|
@auth.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
if request.method == 'GET':
|
|
return render_template("login.html")
|
|
|
|
username = request.form.get('username')
|
|
password = request.form.get('password')
|
|
if not username or not password:
|
|
return render_template("login.html", error="Both username and password must be entered")
|
|
|
|
user_data = db.get_user_by_username(username)
|
|
if not user_data:
|
|
return render_template("login.html", error="User does not exist")
|
|
|
|
if not check_password_hash(user_data['password_hash'], password):
|
|
return render_template("login.html", error="Invalid username or password")
|
|
|
|
user = User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'])
|
|
|
|
login_user(user)
|
|
|
|
next = request.args.get('next')
|
|
return redirect(next or url_for('home.index'))
|
|
|
|
@auth.route('/signup', methods=['GET', 'POST'])
|
|
def signup():
|
|
if request.method == 'GET':
|
|
return render_template("signup.html")
|
|
|
|
username = request.form.get('username')
|
|
password = request.form.get('password')
|
|
if not username or not password:
|
|
return render_template("signup.html", error="Both username and password must be entered")
|
|
|
|
if len(username) < 10 or len(password) < 10:
|
|
return render_template("signup.html", error="Both username and password must be at least 10 characters long")
|
|
|
|
user = db.get_user_by_username(username)
|
|
if user:
|
|
return render_template("signup.html", error="User already exists")
|
|
|
|
hashed_password = generate_password_hash(password)
|
|
user_data = db.create_new_user(username, hashed_password)
|
|
|
|
user = User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'])
|
|
login_user(user)
|
|
|
|
return redirect(url_for('home.index'))
|
|
|
|
@auth.route("/logout")
|
|
@login_required
|
|
def logout():
|
|
logout_user()
|
|
return redirect(url_for('landing_page')) |