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

时间：2016-05-12 14:29:55 阅读：262 评论：0 收藏：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会自动跳转到登入界面