十年网站开发经验 + 多家企业客户 + 靠谱的建站团队
量身定制 + 运营维护+专业推广+无忧售后,网站问题一站解决
创新互联www.cdcxhl.cn八线动态BGP香港云服务器提供商,新人活动买多久送多久,划算不套路!
成都创新互联公司是一家专业提供海州企业网站建设,专注与成都网站设计、网站制作、H5技术、小程序制作等业务。10年已为海州众多企业、政府机构等服务。创新互联专业的建站公司优惠进行中。小编给大家分享一下Flask框架实现用户登录功能的方法,希望大家阅读完这篇文章后大所收获,下面让我们一起去探讨吧!
Flask框架如何通过Flask_login实现用户登录功能?下面给大家具体介绍一下步骤:
运行环境:
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
目录结构:
直接看代码,具体功能有注释
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 '' % self.username
templates/login.html
Title
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语句查到的字段并添加给实例化对象的属性。
看完了这篇文章,相信你对Flask框架实现用户登录功能的方法有了一定的了解,想了解更多相关知识,欢迎关注创新互联-成都网站建设公司行业资讯频道,感谢各位的阅读!