From 783bc86aaf086ef4f2e23a247f1e157e27018364 Mon Sep 17 00:00:00 2001 From: rubenwardy Date: Sun, 19 Jan 2020 02:46:07 +0000 Subject: Update dependencies --- app/models.py | 9 ++++----- app/utils.py | 19 ++----------------- 2 files changed, 6 insertions(+), 22 deletions(-) (limited to 'app') diff --git a/app/models.py b/app/models.py index 732f38d..22a7d25 100644 --- a/app/models.py +++ b/app/models.py @@ -23,7 +23,7 @@ from urllib.parse import urlparse from flask import Flask, url_for from flask_sqlalchemy import SQLAlchemy, BaseQuery from flask_migrate import Migrate -from flask_user import login_required, UserManager, UserMixin, SQLAlchemyAdapter +from flask_user import login_required, UserManager, UserMixin from sqlalchemy import func, CheckConstraint from sqlalchemy_searchable import SearchQueryMixin from sqlalchemy_utils.types import TSVectorType @@ -125,7 +125,7 @@ class User(db.Model, UserMixin): # User email information email = db.Column(db.String(255), nullable=True, unique=True) - confirmed_at = db.Column(db.DateTime()) + email_confirmed_at = db.Column(db.DateTime()) # User information profile_pic = db.Column(db.String(255), nullable=True, server_default=None) @@ -148,7 +148,7 @@ class User(db.Model, UserMixin): def __init__(self, username, active=False, email=None, password=None): self.username = username - self.confirmed_at = datetime.datetime.now() - datetime.timedelta(days=6000) + self.email_confirmed_at = datetime.datetime.now() - datetime.timedelta(days=6000) self.display_name = username self.active = active self.email = email @@ -1044,5 +1044,4 @@ class ForumTopic(db.Model): # Setup Flask-User -db_adapter = SQLAlchemyAdapter(db, User) # Register the User model -user_manager = UserManager(db_adapter, app) # Initialize Flask-User +user_manager = UserManager(app, db, User) diff --git a/app/utils.py b/app/utils.py index fa423cb..4a6d0a5 100644 --- a/app/utils.py +++ b/app/utils.py @@ -102,7 +102,7 @@ def make_flask_user_password(plaintext_str): else: return password.decode("UTF-8") -def _do_login_user(user, remember_me=False): +def loginUser(user): def _call_or_get(v): if callable(v): return v() @@ -128,29 +128,14 @@ def _do_login_user(user, remember_me=False): flash("Your account has not been enabled.", "error") return False - # Check if user has a confirmed email address - user_manager = current_app.user_manager - if user_manager.enable_email and user_manager.enable_confirm_email \ - and not current_app.user_manager.enable_login_without_confirm_email \ - and not user.has_confirmed_email(): - url = url_for("user.resend_confirm_email") - flash("Your email address has not yet been confirmed", "error") - return False - # Use Flask-Login to sign in user - login_user(user, remember=remember_me) + login_user(user, remember=True) signals.user_logged_in.send(current_app._get_current_object(), user=user) flash("You have signed in successfully.", "success") return True -def loginUser(user): - user_mixin = None - if user_manager.enable_username: - user_mixin = user_manager.find_user_by_username(user.username) - - return _do_login_user(user_mixin, True) def rank_required(rank): def decorator(f): -- cgit v1.2.3