Flask框架通过Flask_login实现用户登录功能示例

作者:铠甲巨人 时间:2021-07-19 07:15:38 

本文实例讲述了Flask框架通过Flask_login实现用户登录功能。分享给大家供大家参考,具体如下:

通过Flask_Login实现用户验证登录,并通过login_required装饰器来判断用户登录状态来判断是否允许访问视图函数。

运行环境:

python3.5
Flask 0.12.2
Flask_Login 0.4.1
Flask-WTF 0.14.2
PyMySQL 0.8.0
WTForms 2.1
DBUtils 1.2

目录结构:

Flask框架通过Flask_login实现用户登录功能示例

直接看代码,具体功能有注释

Model/User_model.py


#创建一个类,用来通过sql语句查询结果实例化对象用
class User_mod():
def __init__(self):
 self.id=None
 self.username=None
 self.task_count=None
 self.sample_count=None
def todict(self):
 return self.__dict__
#下面这4个方法是flask_login需要的4个验证方式
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
# def __repr__(self):
#  return '<User %r>' % self.username

templates/login.html


<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<body>
<div class="login-content">
 <form class="margin-bottom-0" action="{{ action }}" method="{{ method }}" id="{{ formid }}">
  {{ form.hidden_tag() }}
  <div class="form-group m-b-20">
   {{ form.username(class='form-control input-lg',placeholder = "用户名") }}
  </div>
  <div class="form-group m-b-20">
   {{ form.password(class='form-control input-lg',placeholder = "密码") }}
  </div>
  <div class="checkbox m-b-20">
   <label>
    {{ form.remember_me() }} 记住我
   </label>
  </div>
  <div class="login-buttons">
   <button type="submit" class="btn btn-success btn-block btn-lg">登 录</button>
  </div>
 </form>
</div>
</body>
</html>

User_dal/dal.py


import pymysql
from DBUtils.PooledDB import PooledDB
POOL = PooledDB(
creator=pymysql, # 使用链接数据库的模块
maxconnections=6, # 连接池允许的最大连接数,0和None表示不限制连接数
mincached=2, # 初始化时,链接池中至少创建的空闲的链接,0表示不创建
maxcached=5, # 链接池中最多闲置的链接,0和None不限制
maxshared=3,
# 链接池中最多共享的链接数量,0和None表示全部共享。PS: 无用,因为pymysql和MySQLdb等模块的 threadsafety都为1,所有值无论设置为多少,_maxcached永远为0,所以永远是所有链接都共享。
blocking=True, # 连接池中如果没有可用连接后,是否阻塞等待。True,等待;False,不等待然后报错
maxusage=None, # 一个链接最多被重复使用的次数,None表示无限制
setsession=[], # 开始会话前执行的命令列表。如:["set datestyle to ...", "set time zone ..."]
ping=0,
# ping MySQL服务端,检查是否服务可用。# 如:0 = None = never, 1 = default = whenever it is requested, 2 = when a cursor is created, 4 = when a query is executed, 7 = always
host='192.168.20.195',
port=3306,
user='root',
password='youpassword',
database='mytest',
charset='utf8'
)
class SQLHelper(object):
@staticmethod
def fetch_one(sql,args):
 conn = POOL.connection() #通过连接池链接数据库
 cursor = conn.cursor() #创建游标
 cursor.execute(sql, args) #执行sql语句
 result = cursor.fetchone() #取的sql查询结果
 conn.close() #关闭链接
 return result
@staticmethod
def fetch_all(self,sql,args):
 conn = POOL.connection()
 cursor = conn.cursor()
 cursor.execute(sql, args)
 result = cursor.fetchone()
 conn.close()
 return result

User_dal/user_dal.py


from Model import User_model
from User_dal import dal
from User_dal import user_dal
class User_Dal:
persist = None
#通过用户名及密码查询用户对象
@classmethod
def login_auth(cls,username,password):
 print('login_auth')
 result={'isAuth':False}
 model= User_model.User_mod() #实例化一个对象,将查询结果逐一添加给对象的属性
 sql ="SELECT id,username,sample_count,task_count FROM User WHERE username ='%s' AND password = '%s'" % (username,password)
 rows = user_dal.User_Dal.query(sql)
 print('查询结果>>>',rows)
 if rows:
  result['isAuth'] = True
  model.id = rows[0]
  model.username = rows[1]
  model.sample_count = rows[2]
  model.task_count = rows[3]
 return result,model
#flask_login回调函数执行的,需要通过用户唯一的id找到用户对象
@classmethod
def load_user_byid(cls,id):
 print('load_user_byid')
 sql="SELECT id,username,sample_count,task_count FROM User WHERE id='%s'" %id
 model= User_model.User_mod() #实例化一个对象,将查询结果逐一添加给对象的属性
 rows = user_dal.User_Dal.query(sql)
 if rows:
  result = {'isAuth': False}
  result['isAuth'] = True
  model.id = rows[0]
  model.username = rows[1]
  model.sample_count = rows[2]
  model.task_count = rows[3]
 return model
#具体执行sql语句的函数
@classmethod
def query(cls,sql,params = None):
 result =dal.SQLHelper.fetch_one(sql,params)
 return result

denglu.py  flask主运行文件


from flask import Flask,render_template,redirect
from flask_login import LoginManager,login_user,login_required,current_user
from flask_wtf.form import FlaskForm
from wtforms import StringField, PasswordField, BooleanField
from wtforms.validators import Length,DataRequired,Optional
from User_dal import user_dal
app = Flask(__name__)
#项目中设置flask_login
login_manager = LoginManager()
login_manager.init_app(app)
app.config['SECRET_KEY'] = '234rsdf34523rwsf'
#flask_wtf表单
class LoginForm(FlaskForm):
username = StringField('账户名:', validators=[DataRequired(), Length(1, 30)])
password = PasswordField('密码:', validators=[DataRequired(), Length(1, 64)])
remember_me = BooleanField('记住密码', validators=[Optional()])
@app.route('/login',methods=['GET','POST'])
def login():
form = LoginForm()
if form.validate_on_submit():
 username = form.username.data
 password = form.password.data
 result = user_dal.User_Dal.login_auth(username,password)
 model=result[1]
 if result[0]['isAuth']:
  login_user(model)
  print('登陆成功')
  print(current_user.username) #登录成功之后可以用current_user来取该用户的其他属性,这些属性都是sql语句查来并赋值给对象的。
  return redirect('/t')
 else:
  print('登陆失败')
  return render_template('login.html',formid='loginForm',action='/login',method='post',form=form)
return render_template('login.html',formid='loginForm',action='/login',method='post',form=form)
'''登录函数,首先实例化form对象
然后通过form对象验证post接收到的数据格式是否正确
然后通过login_auth函数,用username与password向数据库查询这个用户,并将状态码以及对象返回
判断状态码,如果正确则将对象传入login_user中,然后就可以跳转到正确页面了'''
@login_manager.user_loader
def load_user(id):
return user_dal.User_Dal.load_user_byid(id)
'''
load_user是一个flask_login的回调函数,在登陆之后,每访问一个带Login_required装饰的视图函数就要执行一次,
该函数返回一个用户对象,通过id来用sql语句查到的用户数据,然后实例化一个对象,并返回。
'''
#登陆成功跳转的视图函数
@app.route('/t')
@login_required
def hello_world():
print('登录跳转')
return 'Hello World!'
#随便写的另一个视图函数
@app.route('/b')
@login_required
def hello():
print('视图函数b')
return 'Hello b!'
if __name__ == '__main__':
app.run()

简单总结一下:

通过flask的form表单验证数据格式
然后通过用户名密码从数据库取用户对象,将sql执行结果赋值给一个实例化的对象
将这个对象传给login_user,
然后成功跳转。
注意要写一个load_user回调函数吗,返回的是通过id取到的数据库并实例化的对象的用户对象。
这个回调函数每次访问带login_required装饰器的视图函数都会被执行。
还有一个就是current_user相当于就是实例化的用户对象,可以取用户的其他属性,注意,其他属性仅限于sql语句查到的字段并添加给实例化对象的属性。

代码比较简单,只是为了实现功能,敬请谅解,如有错误欢迎指出。

希望本文所述对大家Python程序设计有所帮助。

来源:https://www.cnblogs.com/ArmoredTitan/p/8985656.html

标签:Flask框架,Flask,login,用户登录
0
投稿

猜你喜欢

  • 彻底卸载MySQL的方法分享

    2024-01-23 11:49:08
  • python traceback捕获并打印异常的方法

    2022-11-06 10:35:30
  • JavaScript lastIndexOf方法入门实例(计算指定字符在字符串中最后一次出现的位置)

    2024-04-18 10:54:17
  • Sqlserver 2000/2005/2008 的收缩日志方法和清理日志方法

    2012-07-21 14:55:18
  • python sleep和wait对比总结

    2023-04-30 18:26:04
  • 详解SQL游标的用法

    2024-01-18 02:41:30
  • 优化mysql数据库的经验总结

    2024-01-17 23:56:24
  • 使用Python来编写HTTP服务器的超级指南

    2022-12-28 20:40:24
  • 使用pt-kill根据一定的规则来kill连接的方法

    2024-01-26 01:18:37
  • php 模拟get_headers函数的代码示例

    2023-09-09 06:16:36
  • Go高级特性探究之对象比较详解

    2024-04-26 17:26:59
  • 使用PyQt的QLabel组件实现选定目标框功能的方法示例

    2022-01-08 22:43:21
  • Python带动态参数功能的sqlite工具类

    2022-10-02 14:05:15
  • Python实现npy/mat文件的保存与读取

    2023-07-22 08:49:48
  • 分发服务器 系统抛出18483错误,未能连接服务器,因为'distributor_admin'未定义远程登陆

    2024-01-24 00:39:48
  • JQuery中对Select的option项的添加、删除、取值

    2024-04-22 12:59:14
  • 使用DataFrame删除行和列的实例讲解

    2021-03-05 21:06:04
  • Python实现石头剪刀布游戏

    2022-09-03 10:37:30
  • windows环境下mysql数据库的主从同步备份步骤(单向同步)

    2024-01-19 16:53:34
  • 趣用文化元素

    2009-09-03 11:53:00
  • asp之家 网络编程 m.aspxhome.com