flask-login2的简单使用

 1 #coding:utf8
 2 from flask import Flask, render_template, request, redirect, url_for, flash, abort
 3 from flask_login import (LoginManager, UserMixin, login_user, logout_user,
 4                             current_user, login_required, fresh_login_required)
 5 
 6 app = Flask(__name__)
 7 login_manager = LoginManager(app)
 8 # 设置登录视图的名称，如果一个未登录用户请求一个只有登录用户才能访问的视图，
 9 # 则闪现一条错误消息，并重定向到这里设置的登录视图。
10 # 如果未设置登录视图，则直接返回401错误。
11 login_manager.login_view = 'login'
12 # 设置当未登录用户请求一个只有登录用户才能访问的视图时，闪现的错误消息的内容，
13 # 默认的错误消息是：Please log in to access this page.。
14 login_manager.login_message = 'Unauthorized User'
15 # 设置闪现的错误消息的类别
16 login_manager.login_message_category = "info"
17 
18 users = [
19     {'username': 'Tom', 'password': '111111'},
20     {'username': 'Michael', 'password': '123456'}
21 ]
22 
23 class User(UserMixin):
24     pass
25 
26 # 通过用户名，获取用户记录，如果不存在，则返回None
27 def query_user(username):
28     for user in users:
29         if user['username'] == username:
30             return user
31 
32 # 如果用户名存在则构建一个新的用户类对象，并使用用户名作为ID
33 # 如果不存在，必须返回None
34 @login_manager.user_loader
35 def load_user(username):
36     if query_user(username) is not None:
37         curr_user = User()
38         curr_user.id = username
39         return curr_user
40     # Must return None if username not found
41 
42 # 从请求参数中获取Token，如果Token所对应的用户存在则构建一个新的用户类对象
43 # 并使用用户名作为ID，如果不存在，必须返回None
44 # @login_manager.request_loader
45 #     username = request.args.get('token')
46 #     user = query_user(username)
47 #     if user is not None:
48 #         curr_user = User()
49 #         curr_user.id = username
50 #         return curr_user
51 #     # Must return None if username not found
52 
53 @login_manager.unauthorized_handler
54 def unauthorized_handler():
55     return 'Unauthorized'
56 
57 @app.route('/')
58 @login_required
59 def index():
60     return render_template('hello.html')
61 
62 @app.route('/home')
63 @fresh_login_required
64 def home():
65     return 'Logged in as: %s' % current_user.get_id()
66 
67 @app.route('/login', methods=['GET', 'POST'])
68 def login():
69     if request.method == 'POST':
70         username = request.form.get('username')
71         user = query_user(username)
72         # 验证表单中提交的用户名和密码
73         if user is not None and request.form['password'] == user['password']:
74             curr_user = User()
75             curr_user.id = username
76 
77             # 通过Flask-Login的login_user方法登录用户
78             login_user(curr_user, remember=True)
79 
80             # 如果请求中有next参数，则重定向到其指定的地址，
81             # 没有next参数，则重定向到"index"视图
82             next = request.args.get('next')
83             return redirect(next or url_for('index'))
84 
85         flash('Wrong username or password!')
86     # GET 请求
87     return render_template('login.html')
88 
89 @app.route('/logout')
90 @login_required
91 def logout():
92     logout_user()
93     return 'Logged out successfully!'
94 
95 app.secret_key = '1234567'
96 
97 if __name__ == '__main__':
98     app.run(host='0.0.0.0', debug=True)
http://www.bjhee.com/flask-ext8.html
中文文档：http://docs.jinkan.org/docs/flask-login/