34 lines
990 B
Python
34 lines
990 B
Python
|
from flask_simplelogin import SimpleLogin
|
||
|
|
from werkzeug.security import check_password_hash, generate_password_hash
|
||
|
|
|
||
|
|
from project_name.ext.database import db
|
||
|
|
from project_name.models import User
|
||
|
|
|
||
|
|
|
||
|
|
def verify_login(user):
|
||
|
|
"""Validates user login"""
|
||
|
|
username = user.get("username")
|
||
|
|
password = user.get("password")
|
||
|
|
if not username or not password:
|
||
|
|
return False
|
||
|
|
existing_user = User.query.filter_by(username=username).first()
|
||
|
|
if not existing_user:
|
||
|
|
return False
|
||
|
|
if check_password_hash(existing_user.password, password):
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def create_user(username, password):
|
||
|
|
"""Creates a new user"""
|
||
|
|
if User.query.filter_by(username=username).first():
|
||
|
|
raise RuntimeError(f"{username} already exists")
|
||
|
|
user = User(username=username, password=generate_password_hash(password))
|
||
|
|
db.session.add(user)
|
||
|
|
db.session.commit()
|
||
|
|
return user
|
||
|
|
|
||
|
|
|
||
|
|
def init_app(app):
|
||
|
|
SimpleLogin(app, login_checker=verify_login)
|