带有 MongoDB 的 Flask-login usermixin 类

Question

我正在努力尝试构建登录方法一段时间。我正在运行 Flask 应用程序并且运行良好。这一切都在我的机器上本地运行。目前，我正在使用pymongo并MongoClient连接到数据库。这一切都运行良好，如果可能的话，我不想改变这一点。

我试图用Flask-Login建立一个users使用类usermixin。这是我非常失败的地方。我尝试了一些不同的事情，我的问题是如何从我的数据库中提取数据。我以前使用 SQL DB 完成了此操作，但对于此项目，我明确想使用 MongoDB。这是我试图遵循的教程，但我很难理解所有内容，因为没有很好地解释每一行在做什么。

https://medium.com/@dmitryrastorguev/basic-user-authentication-login-for-flask-using-mongoengine-and-wtforms-922e64ef87fe

这是我与数据库的连接： client = MongoClient('mongodb://localhost:27017')

这是我目前没有工作的用户类，我需要帮助的地方。

class User(UserMixin):

  def __init__(self, username, password_hash):
    self.username = username
    self.password_hash = password_hash

  def check_password(self, password):
    return check_password_hash(self.password_hash, password)

  def get_id(self):
    return self.username

@login_manager.user_loader
def load_user(user_id):
    return User.objects(pk=user_id).first()

然后我的最后一部分是我的登录表单：

@app.route('/login', methods=["GET" , "POST"])
def login():
  if request.method == "GET":
    return render_template("login.html", error=False)
  if request.method == "POST":
    check_user = request.form["username"]
    if check_user:
      if check_password_hash(check_user['password'], request.form["password"]):
        login_user(check_user)
        return redirect(url_for('index'))

我知道本教程使用MongoEngine了我没有使用的，或者还没有使用的，但是这里有一些帮助，无论是如何让上面的代码工作或如何调整它都会很棒。当我运行此代码时，我没有收到任何错误，它只是不起作用。我的测试是我尝试登录，然后尝试转到加载了以下代码的注销页面：

@app.route("/logout")
@login_required
def logout():
  logout_user()
  return redirect(url_for('index'))

当我这样做时，它不会加载页面，并且会收到“未经授权的页面”通知。因此我知道我的代码不起作用。最后，我在静态文件位置拥有所有模板。

在此先感谢您的帮助，如果有任何不清楚的地方请询问，我会尝试添加更多详细信息。越具体越好，我将能够提供帮助。

更新：

我意识到描述我的数据库的结构以确保我正确访问它可能也很重要，因为这是我遇到问题的一个主要问题。我有一个名为Users的数据库，它的结构是每个文档都是不同的用户记录，如下所示：

{
    "_id" : 1,
    "Reset" : false,
    "FirstName" : "John",
    "LastName" : "Doe",
    "Email" : "JohnDoe@gmail.com",
    "Username" : "",
    "admin" : false,
    "Pass" : "[hashed_password]"
}
{
    "_id" : 2,
    "Reset" : true,
    "FirstName" : "Jane",
    "LastName" : "Smith",
    "Email" : "JaneSmith@hotmail.com",
    "Username" : "Jane",
    "admin" : false,
    "Pass" : "[hashed_password]"
}
{
    "_id" : 3,
    "Reset" : true,
    "FirstName" : "Gary",
    "LastName" : "Bettman",
    "Email" : "GBettman@yahoo.com",
    "Username" : "HockeyGuy",
    "admin" : false,
    "Pass" : "[hashed_password]"
}

Answer 1

您需要了解的有关 Flask-login 的信息：此扩展与应用程序的用户模型一起工作，并期望在其中实现某些属性和方法。（来源：https : //blog.miguelgrinberg.com/post/the-flask-mega-tutorial-part-v-user-logins）。

下面列出了四个必需的项目：

• is_authenticated: 如果用户具有有效凭据，则为 True，否则为 False。

• is_active：如果用户的帐户处于活动状态，则为 True，否则为 False。

• is_anonymous：对于普通用户为 False 的属性，对于特殊的匿名用户为 True 的属性。

• get_id(): 一种以字符串形式返回用户唯一标识符的方法

不幸的是，官方文档和Miguel Grinberg优秀博客中的所有示例都使用 SQLAlchemy。好消息，可以用 Pymongo 实现它......

解决方案

1. 路线.py

   from flask import Flask
   from flask_pymongo import PyMongo
   from flask_login import LoginManager
   from flask import render_template, url_for, request, flash
   from app.forms import Login
   from flask import request
   from werkzeug.urls import url_parse
   from werkzeug.security import generate_password_hash, check_password_hash
   from flask_login import current_user, login_user, logout_user, login_required
   
   mongo = PyMongo(app)
   login = LoginManager(app)
   login.login_view = 'login'
   
   class User:
       def __init__(self, username):
           self.username = username
   
       @staticmethod
       def is_authenticated():
           return True
   
       @staticmethod
       def is_active():
           return True
   
       @staticmethod
       def is_anonymous():
           return False
   
       def get_id(self):
           return self.username
   
       @staticmethod
       def check_password(password_hash, password):
           return check_password_hash(password_hash, password)
   
   
       @login.user_loader
       def load_user(username):
           u = mongo.db.Users.find_one({"Name": username})
           if not u:
               return None
           return User(username=u['Name'])
   
   
       @app.route('/login', methods=['GET', 'POST'])
       def login():
           if current_user.is_authenticated:
               return redirect(url_for('index'))
           form = Login()
           if form.validate_on_submit():
               user = mongo.db.Users.find_one({"Name": form.name.data})
               if user and User.check_password(user['Password'], form.password.data):
                   user_obj = User(username=user['Name'])
                   login_user(user_obj)
                   next_page = request.args.get('next')
                   if not next_page or url_parse(next_page).netloc != '':
                       next_page = url_for('index')
                   return redirect(next_page)
               else:
                   flash("Invalid username or password")
           return render_template('login.html', title='Sign In', form=form)
   
   
       @app.route('/logout')
       def logout():
           logout_user()
           return redirect(url_for('login'))
   

2. 表单.py

   from flask_wtf import FlaskForm
   from wtforms import StringField, SubmitField, PasswordField
   from wtforms.validators import DataRequired
   
   
   class Login(FlaskForm):
       name = StringField('name' validators=[DataRequired()])
       password = PasswordField('Password', validators=[DataRequired()])
       login = SubmitField('Login')
   

假设我们在 Mongodb 一侧有一个包含一些登录信息的集合 ( Users )。例如：

{
  Name: [username],
  Password: [hashed_password]
} 

有关每行代码的作用的进一步说明，我建议您参考以下链接：

• https://boh717.github.io/post/flask-login-and-mongodb/
• https://flask-login.readthedocs.io/en/latest/

Answer 2

我发现以下作品与flask-login, UserMixin,pymongo

这是用户模型

import datetime
import uuid
from depo import bcrypt, login_manager
from flask import session, flash
from depo.common.database import Database
from depo.models.blog import Blog
from flask_login import UserMixin

class User(UserMixin):

    def __init__(self, username, email, password, _id=None):

        self.username = username
        self.email = email
        self.password = password
        self._id = uuid.uuid4().hex if _id is None else _id

    def is_authenticated(self):
        return True
    def is_active(self):
        return True
    def is_anonymous(self):
        return False
    def get_id(self):
        return self._id

    @classmethod
    def get_by_username(cls, username):
        data = Database.find_one("users", {"username": username})
        if data is not None:
            return cls(**data)

    @classmethod
    def get_by_email(cls, email):
        data = Database.find_one("users", {"email": email})
        if data is not None:
            return cls(**data)

    @classmethod
    def get_by_id(cls, _id):
        data = Database.find_one("users", {"_id": _id})
        if data is not None:
            return cls(**data)

    @staticmethod
    def login_valid(email, password):
        verify_user = User.get_by_email(email)
        if verify_user is not None:
            return bcrypt.check_password_hash(verify_user.password, password)
        return False

    @classmethod
    def register(cls, username, email, password):
        user = cls.get_by_email(email)
        if user is None:
            new_user = cls( username, email, password)
            new_user.save_to_mongo()
            session['email'] = email
            return True
        else:
            return False

    def json(self):
        return {
            "username": self.username,
            "email": self.email,
            "_id": self._id,
            "password": self.password
        }

    def save_to_mongo(self):
        Database.insert("users", self.json())

这里是路线

from flask import flash, render_template, request, session, make_response,  redirect, url_for
from depo import app, bcrypt, login_manager
from depo.models.blog import Blog
from depo.models.post import Post
from depo.models.user import User
from depo.common.database import Database
from depo.usercon.forms import RegistrationForm, LoginForm
from flask_login import login_user


@app.before_first_request
def initialize_database():
    Database.initialize()

@app.route("/register", methods=['GET', 'POST'])
def register():
    form = RegistrationForm()
    if form.validate_on_submit():

        if request.method == 'POST':
            username = request.form["username"]
            email = request.form["email"]
            password = bcrypt.generate_password_hash(request.form["password"])
                       .decode('utf-8')
            find_user =  User.get_by_email(email)
            if find_user is None:
                User.register(username, email, password)
                flash(f'Account created for {form.username.data}!', 'success')
                return redirect(url_for('home'))
            else:
                flash(f'Account already exists for {form.username.data}!', 'success')
    return render_template('register.html', title='Register', form=form)
@app.route("/login", methods=['GET', 'POST'])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        email = request.form["email"]
        password = request.form["password"]
        find_user = Database.find_one("users", {"email": email})
        if User.login_valid(email, password):
            loguser = User(find_user["_id"], find_user["email"], find_user["password"])
            login_user(loguser, remember=form.remember.data)
            flash('You have been logged in!', 'success')
            return redirect(url_for('home'))
        else:
            flash('Login Unsuccessful. Please check email and password', 'danger')
    return render_template('login.html', title='Login', form=form)

@login_manager.user_loader
def load_user(user_id):
    user =User.get_by_id(user_id)
    if user is not None:
        return User(user["_id"])
    else:
        return None

app如果尚未初始化，可能还需要一些代码

from flask_login import LoginManager
login_manager = LoginManager(app)