flask笔记：6：用户登入登出

╰半橙微兮° 2022-08-22 01:28 150阅读 0赞

用户登入登出需要用到 Flask-Login 插件

初始化

修改配置文件 app/\_\_init\_\_.py

from flask import Flask
    from flask.ext.sqlalchemy import SQLAlchemy
    import os
    from flask.ext.login import LoginManager
    app=Flask(__name__)
    app.config.from_object('config')
    db=SQLAlchemy(app)
    lm = LoginManager()
    lm.init_app(app)
    lm.login_view = 'login'
    from app import views,models
    #导入login模块
    #flask-lonin需要一个LoginManager( )实例
    #init_app(应用名)配置到应用中
    #login_view 那个视图允许登入

重构用户模型

修改模型 app/models.py

from app import db
    class User(db.Model):
        id = db.Column(db.Integer, primary_key=True)
        nickname = db.Column(db.String(64), index=True, unique=True)
        email = db.Column(db.String(120), index=True, unique=True)
        posts = db.relationship('Post', backref='author', lazy='dynamic')
    
        def is_authenticated (self):
            return True
    
        def is_active(self):
            return True
    
        def is_anonymous(self):
            return False
    
        def get_id(self):
            try:
                return unicode(self.id)  # python 2
            except NameError:
                return str(self.id)  # python 3
    
        def __repr__(self):
            return '<User %r>' % (self.nickname)   
    
    class Post(db.Model):
        id = db.Column(db.Integer, primary_key = True)
        body = db.Column(db.String(140))
        timestamp = db.Column(db.DateTime)
        user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    
        def __repr__(self):
            return '<Post %r>' % (self.body)
    # is_authenticated 允许用户验证，只返回True
    # is_active 有效账户都返回True，除非禁止账户
    # is_anonymous 伪造用户之外都是True
    # get_id 返回用户唯一标识符，以unicode格式

登入登出视图

修改视图 app/views.py

from flask import render_template, flash, redirect, session, url_for, request, g
    from flask.ext.login import login_user, logout_user, current_user, login_required
    from app import app, db, lm, models
    from forms import LoginForm
    from models import User
    
    @lm.user_loader
    def load_user(id):
        return User.query.get(int(id))
    
    @app.before_request
    def before_request():
        g.user = current_user
    
    @app.route('/')
    @app.route('/index')
    @login_required
    def index ():
        user=g.user
        posts=[
                {'author':{'nickname':'John'},
                 'body':'Beautiful day in Portland!'},
                {'author':{'nickname':'Susan'},
                 'body':'The Avengers movie was so cool!'}
               ]
        return render_template("index.html",
        title="Home",
        user=user,
        posts=posts)
    
    @app.route('/login', methods = ['GET', 'POST'])
    def login():
        if g.user is not None and g.user.is_authenticated:
            return redirect(url_for('index'))
        form = LoginForm()
        if form.validate_on_submit():
            if models.User.query.filter_by(nickname=form.openid.data).first():
                user = User.query.filter_by(nickname=form.openid.data).first_or_404()
                login_user(user)
                return redirect(url_for('index'))
            else:
                return render_template('login.html',
            title = 'Sign In',
            error='[NO]',
            form = form)
        return render_template('login.html',
            title = 'Sign In',
            form = form)
    
    @app.route('/logout')
    @login_required
    def logout():
        logout_user()
        return redirect(url_for('index'))
    
    # @user_loader回调， 这个回调从会话中存储的用户 ID 重新加载用户对象， 它应该接受一个用户的 unicodeID 作为参数，并且返回相应的用户对象。 如果 ID 无效的话，它应该返回None (而不是抛出异常)。(在这种情况下，ID 会被手动从会话中移除且处理会继续)
    # load_user( )用来返回用户唯一标识
    
    # before_request( )用来储存一个标志，判断用户是否登入
    # current_user 获取当前登陆用户信息
    # 被 @before_request 绑定的函数会在请求收到时执行( 网上说会在请求收到前执行，不太明白)
    # g保存的是当前请求的全局变量，不同的请求会有不同的全局变量
    
    #用户登入登出视图用@login_required， 有些地方（比如修改密码）方法上需要加上fresh_login_required而不是login_required，两者的区别在于前者必须是用户手动登陆，后者还包含了cookie自动登陆的情况，@login_required还用来确保只有登入用户可见
    
    # if g.user is not None and g.user.is_authenticated( ): 的判断用户是否登入，是否允许登入
    # if models.User.query.filter_by(nickname=form.openid.data).first( ):判断输入与数据库里是否一致
    #User.query.filter_by(nickname=form.openid.data).first_or_404( )如果输入与数据库不一致，请求返回404
    #login_user( )是将用户登入(个人理解)
    
    #logout_user( )用户登出，并清除cookie
    #redirect(url_for('视图处理函数名'))跳转页面

修改模板 app/templates/base.html

<html>
        <head>
            {% if title %}
            <title>{
        {title}} - myblog</title>
            {% else %}
            <title>Welcome - myblog</title>
            {% endif %}
        </head>
        <body>
            <div>MyBlog:<a href="{
        {url_for('index')}}">Home</a>
            {% if g.user.is_authenticated %}
            |<a href="{
        {url_for('logout')}}">Logout</a>
            {% endif %}
            </div>
            <hr>
            {% with messages=get_flashed_messages() %}
            {% if messages %}
            <ul>
            {% for message in messages %}
             <li>{
        {message}}</li>
             {% endfor %}
            </ul>
            {% endif %}
            {% endwith %}
            {% block content %}
            {% endblock%}
        </body>
    </html>

\#if g.user.is\_authenticated 判断用户登入就显示登出链接

修改 app/templates/login.html

{% extends "base.html" %}
    {% block content %}
    <h1>Sign In</h1>
    <form action="" method="post" name="login">
        {
        {form.hidden_tag()}}
        <p>
            Please enter your OpenID, or select one of the providers below:<br>
            {
        {form.openid(size=80)}}
            <span style="color: red;">{
        {error}}</span>
            <br>
        </p>
        <p>{
        {form.remember_me}} Remember Me</p>
        <p><input type="submit" value="Sign In"></p>
    </form>
    {% endblock %}

显示：

没有登入，index的URL会自动跳转到登入界面

![Center][]

输入数据库里没有的用户名会显示\[NO\]

![Center 1][]

登入成功会返回index页面，并显示出登出链接

![Center 2][]

登出后跳转到登入界面

![Center 3][]

\#出现 Please log in to access this page. 可能是flask-login 自带一个flash，然后被get\_flashed\_messages( )这个函数捕捉到了( 猜的 )

\#本来应该跳转到 index界面，只是index有 @login\_required，只有登入用户可见，所以指向了登入界面

[Center]: /images/20220724/c62628bfb4004bb99d680d79ff1f7792.png
[Center 1]: /images/20220724/6bdf452705bb4acb898e6e224aaa9e5f.png
[Center 2]: /images/20220724/d4e447411a3d49359b16d1381dce1f37.png
[Center 3]: /images/20220724/a9df67d6e92f4ad9b7f6ab8a75928327.png