diff options
| author | rubenwardy <rw@rubenwardy.com> | 2020-01-19 02:46:07 +0000 |
|---|---|---|
| committer | rubenwardy <rw@rubenwardy.com> | 2020-01-19 02:46:07 +0000 |
| commit | 783bc86aaf086ef4f2e23a247f1e157e27018364 (patch) | |
| tree | d1b84318ca1fc507f92fdf590cc8c7b985b1e2ae /app | |
| parent | 6e626c0f8949c8a705973015387100507a8b1ab9 (diff) | |
| download | cheatdb-783bc86aaf086ef4f2e23a247f1e157e27018364.tar.xz | |
Update dependencies
Diffstat (limited to 'app')
| -rw-r--r-- | app/models.py | 9 | ||||
| -rw-r--r-- | app/utils.py | 19 |
2 files changed, 6 insertions, 22 deletions
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): |
