Flask 实现登录状态持久化:让用户 1 天内无需重新登录

news/2024/10/18 6:58:24/

在开发 Web 应用时,用户登录状态的管理至关重要。默认情况下,Session 是临时的,一旦用户关闭浏览器,Session 通常会被清除,这意味着用户需要重新登录。然而,在许多应用场景中,我们希望实现持久化登录,即用户即使关闭了浏览器,在一段时间内仍然不需要重新登录。本文将介绍如何在 Flask 中实现这一功能,包括使用 Session 持久化、JWT Token、以及Redis 存储 Session 的方法。


目录

  1. 使用 Flask-Session 持久化登录状态
  2. 使用 Cookies 存储 JWT Token 实现持久化登录
  3. Redis 持久化 Session,支持分布式架构
  4. 总结

1. 使用 Flask-Session 持久化登录状态

Flask 使用 Session 管理用户的登录状态。默认情况下,Session 是临时的,通常在用户关闭浏览器后失效。为了让用户的登录状态可以持续一段时间(如 1 天),我们可以将 Session 配置为“永久”,并设置一个有效期。

代码实现
python">from flask import Flask, session, redirect, url_for, request
import datetimeapp = Flask(__name__)
app.secret_key = 'your_secret_key'# 设置 Session 的过期时间为1天
app.config['PERMANENT_SESSION_LIFETIME'] = datetime.timedelta(days=1)@app.route('/login', methods=['GET', 'POST'])
def login():if request.method == 'POST':username = request.form['username']password = request.form['password']# 假设用户名和密码正确if username == 'test_user' and password == 'test_password':session.permanent = True  # 使 session 永久化session['username'] = username  # 存储用户信息在 session 中return redirect(url_for('dashboard'))return 'Invalid credentials'return '''<form method="post"><p><input type=text name=username><p><input type=password name=password><p><input type=submit value=Login></form>'''@app.route('/dashboard')
def dashboard():if 'username' in session:return f'Hello, {session["username"]}! You are logged in.'return redirect(url_for('login'))@app.route('/logout')
def logout():session.pop('username', None)  # 删除 session 中的用户信息return redirect(url_for('login'))if __name__ == '__main__':app.run(debug=True)
解释
  • Session 持久化:通过 session.permanent = True 将 Session 设置为永久,即使关闭浏览器,Session 仍然会保持。
  • PERMANENT_SESSION_LIFETIME:设置为 1 天,使得用户在登录后的 1 天内保持登录状态。

2. 使用 Cookies 存储 JWT Token 实现持久化登录

另一种实现持久化登录的方式是通过 JWT (JSON Web Token),将生成的 Token 存储在 Cookies 中,并设置过期时间。在用户访问受保护的路由时,服务器可以通过解码 Token 验证用户身份,从而实现 1 天内无需重新登录。

代码实现
python">from flask import Flask, request, redirect, url_for, render_template, make_response
import datetime
import jwtapp = Flask(__name__)
app.secret_key = 'your_secret_key'# JWT 秘钥
JWT_SECRET = 'your_jwt_secret'def create_jwt_token(username):payload = {'username': username,'exp': datetime.datetime.utcnow() + datetime.timedelta(days=1)  # 1天后过期}token = jwt.encode(payload, JWT_SECRET, algorithm='HS256')return tokendef verify_jwt_token(token):try:decoded = jwt.decode(token, JWT_SECRET, algorithms=['HS256'])return decoded['username']except (jwt.ExpiredSignatureError, jwt.InvalidTokenError):return None@app.route('/login', methods=['GET', 'POST'])
def login():if request.method == 'POST':username = request.form['username']password = request.form['password']if username == 'test_user' and password == 'test_password':token = create_jwt_token(username)resp = make_response(redirect(url_for('dashboard')))# 设置 Cookie,保存1天resp.set_cookie('auth_token', token, max_age=60*60*24)  # 1天有效期return respreturn 'Invalid username or password', 401return render_template('login.html')@app.route('/dashboard')
def dashboard():token = request.cookies.get('auth_token')if token:username = verify_jwt_token(token)if username:return f'Hello, {username}! You are logged in.'return redirect(url_for('login'))@app.route('/logout')
def logout():resp = make_response(redirect(url_for('login')))resp.delete_cookie('auth_token')  # 删除 Cookiereturn respif __name__ == '__main__':app.run(debug=True)
解释
  • JWT:登录成功后生成 JWT Token,将其存储在 Cookie 中,并设置有效期为 1 天。
  • Token 验证:在用户访问受保护的路由时,通过解码 Token 来验证用户身份。

3. Redis 持久化 Session,支持分布式架构

在更复杂的应用场景中,可能需要跨多个服务器或进程保持用户的登录状态。这时,可以使用 Redis 或其他数据库来持久化 Session。Redis 允许将 Session 存储在服务器端,并支持集群化和分布式应用场景。

代码实现
python">from flask import Flask, session
from flask_session import Session
import redisapp = Flask(__name__)# 配置 Redis 作为 Session 后端
app.config['SESSION_TYPE'] = 'redis'
app.config['SESSION_PERMANENT'] = True
app.config['PERMANENT_SESSION_LIFETIME'] = datetime.timedelta(days=1)
app.config['SESSION_USE_SIGNER'] = True
app.config['SESSION_KEY_PREFIX'] = 'myapp_'
app.config['SESSION_REDIS'] = redis.StrictRedis(host='localhost', port=6379, db=0)# 初始化 Session
Session(app)@app.route('/login')
def login():session['username'] = 'test_user'  # 存储用户名到 Sessionreturn 'Logged in successfully!'@app.route('/dashboard')
def dashboard():username = session.get('username')if username:return f'Hello, {username}!'return 'You need to log in first.', 401if __name__ == '__main__':app.run(debug=True)
解释
  • Redis 后端:Session 数据保存在 Redis 中,支持多服务器之间的 Session 同步和持久化。

4. 总结

通过本文,我们介绍了如何在 Flask 应用中实现持久化登录。无论是使用 Session 还是 JWT Token,都可以让用户在 1 天内无需重新登录。如果您的应用需要分布式架构支持,Redis 是一个很好的选择,用于持久化存储 Session 数据。根据应用的需求,您可以选择适合的方案来优化用户体验。



http://www.ppmy.cn/news/1532833.html

相关文章

leetcode每日一题day19(24.9.29)——买票需要的时间

思路&#xff1a;在最开始的情况下每人需要买的票数减一是能保持相对位置不变的&#xff0c; 如果再想减一就有可能 有某些人只买一张票&#xff0c;而离开了队伍&#xff0c; 所有容易想到对于某个人如果比当前的人买的多就按当前的人数量算 因为在一次次减一的情况下&#xf…

【学习笔记】手写 Tomcat 六

目录 一、线程池 1. 构建线程池的类 2. 创建任务 3. 执行任务 测试 二、URL编码 解决方案 测试 三、如何接收客户端发送的全部信息 解决方案 测试 四、作业 1. 了解工厂模式 2. 了解反射技术 一、线程池 昨天使用了数据库连接池&#xff0c;我们了解了连接池的优…

学MybatisPlus

1.设置MySql的数据库 spring:datasource:url: jdbc:mysql://127.0.0.1:3306/mp?useUnicodetrue&characterEncodingUTF-8&autoReconnecttrue&serverTimezoneAsia/Shanghaidriver-class-name: com.mysql.cj.jdbc.Driverusername: rootpassword: MySQL123 logging:l…

如何使用Optuna在PyTorch中进行超参数优化

所有神经网络在训练过程中都需要选择超参数,而这些超参数对收敛速度和最终性能有着非常显著的影响。 这些超参数需要特别调整,以充分发挥模型的潜力。超参数调优过程是神经网络训练中不可或缺的一部分,某种程度上,它是一个主要基于梯度优化问题中的“无梯度”部分。 在这…

H.264编解码介绍

一、简介 H.264,又称为AVC(Advanced Video Coding),是一种广泛使用的视频压缩标准。它由国际电信联盟(ITU)和国际标准化组织(ISO)联合开发,并于2003年发布。 H.264的发展历史可以追溯到上个世纪90年代。当时,视频压缩技术的主要标准是MPEG-2,但它在压缩率和视频质…

Spring Boot房屋租赁平台:现代化解决方案

1 绪论 1.1 研究背景 中国的科技的不断进步&#xff0c;计算机发展也慢慢的越来越成熟&#xff0c;人们对计算机也是越来越更加的依赖&#xff0c;科研、教育慢慢用于计算机进行管理。从第一台计算机的产生&#xff0c;到现在计算机已经发展到我们无法想象。给我们的生活改变很…

学Java还是c++好?

Java在互联网行业中的应用非常广泛&#xff0c;就业方面&#xff0c;Java岗位较多&#xff0c;就业市场相对稳定&#xff0c;尤其是在Android应用开发、企业级应用、大数据技术等领域有大量的岗位。 Java的语法相对简单&#xff0c;对于初学者来说更容易上手。有很多成熟的框架…

git直接推送项目到仓库

要将您的本地项目推送到新创建的 GitHub 仓库&#xff0c;请按照以下步骤操作&#xff1a; 定位到项目目录&#xff1a; 确保你的命令行或终端当前目录为你的项目根目录。 初始化 Git 仓库&#xff08;如果尚未初始化&#xff09;&#xff1a; git init添加文件到暂存区&#…