Flask框架的学习指南之用户登录管理


Posted in Python onNovember 20, 2016

继续flask的学习之旅。今天介绍flask的登陆管理模块,还记得上一篇中的blog小项目么,登录是咱们自己写的验证代码,大概有以下几个步骤:

1、在登录框中输入用户名和密码

2、flask view函数获取用户密码,然后到数据库中查询该用户信息,进行匹配

3、如果成功,就写入session中,重定向到首页

4、如果对于特定视图,必须要登录才能访问,那么需要在每个视图函数验证session里是否存在该用户。

今天继续改造blog项目,介绍的flask-login模块就是替我们来搞定这些业务相关度较少的公用功能,它可以帮助我们:

在会话中存储当前活跃的用户 ID,让你能够自由地登入和登出。
让你限制登入(或者登出)用户可以访问的视图。
处理让人棘手的 “记住我” 功能。
帮助你保护用户会话免遭 cookie 被盗的牵连。
可以与以后可能使用的 Flask-Principal 或其它认证扩展集成。

一、如何使用flask-login模块?

1、安装flask-login

E:\workdir\dct-server-5055187\src>pip install flask-login

2、使用flask-login

 2.1)  在/blog2/__init__.py添加:

#引用包
from flask.ext.login import LoginManager

#登陆管理
#声明login对象
login_manager = LoginManager()
#初始化绑定到应用
login_manager.init_app(app)

#声明默认视图函数为login,当我们进行@require_login时,如果没登陆会自动跳到该视图函数处理
login_manager.login_view = "login"

#当登陆成功后,该函数会自动从会话中存储的用户 ID 重新加载用户对象。它应该接受一个用户的 unicode ID 作为参数,并且返回相应的用户对象。
@login_manager.user_loader
def load_user(userid):
return User.query.get(int(userid))

2.2) 修改User模型(红色部分为新增代码)

from flask.ext.login import UserMixin

from blog2 import db

class User(db.Model, UserMixin):
  __tablename__ = 'b_user'
  id = db.Column(db.Integer,primary_key=True)
  username = db.Column(db.String(10),unique=True)
  password = db.Column(db.String(16))

  def __init__(self,username,password):
    self.username = username
    self.password = password
  def __repr__(self):
    return '<User %r>' % self.username

该用户类必须实现以下几个方法:

is_authenticated

当用户通过验证时,也即提供有效证明时返回 True(只有通过验证的用户会满足 login_required 的条件。)

is_active

如果这是一个活动用户且通过验证,账户也已激活,未被停用,也不符合任何你 的应用拒绝一个账号的条件,返回 True 。不活动的账号可能不会登入(当然, 是在没被强制的情况下)。

is_anonymous

如果是一个匿名用户,返回 True 。(真实用户应返回 False 。)

get_id()

返回一个能唯一识别用户的,并能用于从 user_loader 回调中加载用户的 unicode 。注意着 必须 是一个 unicode —— 如果 ID 原本是 一个 int 或其它类型,你需要把它转换为 unicode 。
要简便地实现用户类,你可以从 UserMixin 继承,它提供了对所有这些方法的默认实现。我们这里使用UserMixin实现。

2.3) 修改view函数(红色部分为新增)

from flask.ext.login import login_required, login_user, logout_user

from blog2.model.User import User
from blog2.model.Category import Category
import os

from blog2 import app,db
from flask import request,render_template,flash,abort,url_for,redirect,session,Flask,g

@app.route('/')
@login_required
def show_entries():
  categorys = Category.query.all()
  return render_template('show_entries.html',entries=categorys)

@app.route('/add',methods=['POST'])
@login_required
def add_entry():
  #——————————————————————————————————————————————
  # 第一版登陆方式
  # if not session.get('logged_in'):
  #   abort(401)
  #——————————————————————————————————————————————

  title = request.form['title']
  content = request.form['text']
  category = Category(title,content)
  db.session.add(category)
  db.session.commit()
  flash('New entry was successfully posted')
  return redirect(url_for('show_entries'))

@app.route('/login',methods=['GET','POST'])
def login():
  error = None
  if request.method == 'POST':
    user = User.query.filter_by(username=request.form['username']).first()
    #——————————————————————————————————————————————————————————————————————————
    #第一版登陆方式
    # passwd = User.query.filter_by(password=request.form['password']).first()
    #
    # if user is None:
    #   error = 'Invalid username'
    # elif passwd is None:
    #   error = 'Invalid password'
    # else:
    #   session['logged_in'] = True
    #   flash('You were logged in')
    #   return redirect(url_for('show_entries'))
    #——————————————————————————————————————————————————————————————————————————

    login_user(user)
    flash('Logged in successfully.')
    return redirect(url_for('show_entries'))

  return render_template('login.html', error=error)

@app.route('/logout')
@login_required
def logout():
  #—————————————————————————————————————————————— 
  # 第一版登出方式
  # session.pop('logged_in', None)
  #—————————————————————————————————————————————— 
 logout_user()
  flash('You were logged out')
  return redirect(url_for('show_entries'))

通过flask-login管理登陆,代码非常简洁简单:

@login_required:该装饰器放到需要登陆才能访问的视图上,如果没有登陆访问有限制的视图就会跳到login页面,由__init__.py中login_manager.login_view = "login"控制
login_user(user):传入一个user对象进行登陆验证,正确返回true,否则返回false
logout_user():登出函数,清空session中用户信息

2.4) 模板中引用用户

{% if current_user.is_authenticated() %}
 Hi {{ current_user.name }}!
{% endif %}

将之前layout.html和show_entries.html模板中判断用户是否登陆方式改为flask-login中的方式:

{% if not current_user.is_authenticated() %}

current_user值:在用户没有登陆时,值为<flask_login.AnonymousUserMixin object at 0x0000000003DCF550>,也就是匿名用户
                用户登陆后,值为<User u'admin'>

当然,用户登陆还可以根据实际情况自己定制,具体不在一一详述。

【参考文档】

       Flask-Login中文版:http://www.pythondoc.com/flask-login/#id1
       Flask-Login英文版:http://flask-login.readthedocs.io/en/latest/

Python 相关文章推荐
Python类的基础入门知识
Nov 24 Python
如何运行Python程序的方法
Apr 21 Python
Python编写检测数据库SA用户的方法
Jul 11 Python
Python装饰器decorator用法实例
Nov 10 Python
Python实现栈的方法
May 26 Python
Python实现分割文件及合并文件的方法
Jul 10 Python
python搭建微信公众平台
Feb 09 Python
Python探索之SocketServer详解
Oct 28 Python
python爬取个性签名的方法
Jun 17 Python
Centos7下源码安装Python3 及shell 脚本自动安装Python3的教程
Mar 07 Python
Python logging模块原理解析及应用
Aug 13 Python
Python 通过爬虫实现GitHub网页的模拟登录的示例代码
Aug 17 Python
Flask框架的学习指南之制作简单blog系统
Nov 20 #Python
Flask框架的学习指南之开发环境搭建
Nov 20 #Python
Python 描述符(Descriptor)入门
Nov 20 #Python
独特的python循环语句
Nov 20 #Python
【Python】Python的urllib模块、urllib2模块批量进行网页下载文件
Nov 19 #Python
Python基础中所出现的异常报错总结
Nov 19 #Python
轻松掌握python设计模式之策略模式
Nov 18 #Python
You might like
php IP及IP段进行访问限制的代码
2008/12/17 PHP
使用PHPMyAdmin修复论坛数据库的图文方法
2012/01/09 PHP
laravel实现前后台路由分离的方法
2019/10/13 PHP
javascript div 弹出可拖动窗口
2009/02/26 Javascript
JQuery Ajax通过Handler访问外部XML数据的代码
2010/06/01 Javascript
jquery 清空file域示例(兼容个浏览器)
2013/10/11 Javascript
利用jq让你的div居中的好方法分享
2013/11/21 Javascript
jQuery使用post方法提交数据实例
2015/03/25 Javascript
浅谈JavaScript中的string拥有方法的原因
2015/08/28 Javascript
EasyUI 结合JS导出Excel文件的实现方法
2016/11/10 Javascript
Javascript使用SWFUpload进行多文件上传
2016/11/16 Javascript
微信小程序之仿微信漂流瓶实例
2016/12/09 Javascript
JavaScript输出所选择起始与结束日期的方法
2017/07/12 Javascript
vue-cli初始化项目中使用less的方法
2018/08/09 Javascript
微信小程序自定义toast的实现代码
2018/11/16 Javascript
jsonp格式前端发送和后台接受写法的代码详解
2019/11/07 Javascript
python类的继承实例详解
2017/03/30 Python
Python面向对象编程基础解析(二)
2017/10/26 Python
pandas object格式转float64格式的方法
2018/04/10 Python
详解Python中的type和object
2018/08/15 Python
python获取Pandas列名的几种方法
2019/08/07 Python
Python Celery异步任务队列使用方法解析
2020/08/10 Python
Django数据模型中on_delete使用详解
2020/11/30 Python
中国排名第一的外贸销售网站:LightInTheBox.com(兰亭集势)
2016/10/28 全球购物
新加坡网上花店:FlowerAdvisor新加坡
2018/10/05 全球购物
Electric官网:美国高级眼镜和配件品牌
2020/06/04 全球购物
Final类有什么特点
2012/04/25 面试题
学生打架检讨书1000字
2014/01/16 职场文书
简历的自我评价
2014/02/03 职场文书
事业单位竞聘上岗实施方案
2014/03/28 职场文书
体育活动总结范文
2014/05/04 职场文书
环保建议书500字
2014/05/14 职场文书
离婚协议书怎么写(范本参考)
2014/09/30 职场文书
中学生思想品德评语
2014/12/31 职场文书
回复函范文
2015/07/14 职场文书
2016年教师节贺卡寄语
2015/12/04 职场文书