simple-forum/login.py

# This is only a module to make the login process faster and easier. Don't care about it unless you also wants a simple login for your app
import hashlib as hl
from flask_login import LoginManager, UserMixin, login_user, logout_user, login_required, current_user

def initLogin(app, db):
    # Create the User clasis
	class User(UserMixin, db.Model):
	    id = db.Column(db.Integer, primary_key=True)
	    username = db.Column(db.Text, unique=True)
	    password = db.Column(db.Text)
	    def __init__(self, username, password):
	        self.username = username
	        self.password = password

    # Configure login
	login_manager = LoginManager()
	login_manager.init_app(app)
	@login_manager.user_loader
	def load_user(user_id):
		return User.query.get(int(user_id))

	return User
	

def loginUser(username, password, User):
	# Hash the username and the password
	#username = hl.md5(bytes(username, 'utf-8')).hexdigest()
	password = hl.md5(bytes(password, 'utf-8')).hexdigest()
	
	# Check if it exists
	user = User.query.filter_by(username=username, password=password).first_or_404()
	login_user(user)
	return True

def createUser(username, password, db, User):
	# hash the username and the password
	#username = hl.md5(bytes(username, 'utf-8')).hexdigest() # Comment this is you want a clear username
	password = hl.md5(bytes(password, 'utf-8')).hexdigest()
	
	# Send them to db
	user = User(username, password)
	db.session.add(user)
	db.session.commit()

	# Login the user
	login_user(user)

	# return success
	return True

def getUsername():
    return current_user.username

# To restrict a page to a user just add @login_required
# To logout just do logout_user()
# To get the current username do current_user.username