Flask Session 与 Cookie

from flask import Flask, request, make_response

app = Flask(__name__)

# 1. 设置 Cookie
@app.route('/set-cookie')
def set_cookie():
    # 必须使用 make_response 将返回值包装成响应对象
    resp = make_response('Cookie 设置成功！')

    # 设置 Cookie: key, value, max_age(有效期，单位：秒)
    # 如果不设置 max_age，则默认为会话级 Cookie（浏览器关闭即失效）
    resp.set_cookie('username', 'john_doe', max_age=3600) 
    resp.set_cookie('theme', 'dark') # 浏览器关闭即失效
    return resp

# 2. 读取 Cookie
@app.route('/get-cookie')
def get_cookie():
    # 使用 request.cookies 字典获取
    # 推荐使用 .get() 方法，避免 key 不存在时报 KeyError
    username = request.cookies.get('username', '匿名用户')
    theme = request.cookies.get('theme', 'light')

    return f'你好, {username}! 你的主题设置是: {theme}'

# 3. 删除 Cookie
@app.route('/delete-cookie')
def delete_cookie():
    resp = make_response('Cookie 已删除！')
    # 删除 Cookie 实际上是将它的过期时间设置为过去
    resp.delete_cookie('username')
    resp.delete_cookie('theme')
    return resp

app = Flask(__name__)
# 必须设置一个复杂且保密的密钥，用于对 Session 数据进行加密签名
app.config['SECRET_KEY'] = 'your-super-secret-key-here-change-in-production'

from flask import Flask, session, redirect, url_for, request

app = Flask(__name__)
app.config['SECRET_KEY'] = 'super-secret-key-123'

# 1. 登录：写入 Session
@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        username = request.form.get('username')
        # 将数据存入 session 字典
        session['username'] = username
        session['is_logged_in'] = True
        return redirect(url_for('index'))

    return '''
        <form method="post">
            <input type="text" name="username" placeholder="Username">
            <input type="submit" value="Login">
        </form>
    '''

# 2. 首页：读取 Session
@app.route('/')
def index():
    # 使用 .get() 安全读取
    if session.get('is_logged_in'):
        return f'欢迎回来, {session.get("username")}! <a href="/logout">登出</a>'
    return '你还未登录，请 <a href="/login">登录</a>。'

# 3. 登出：删除 Session
@app.route('/logout')
def logout():
    # 移除特定 key
    session.pop('username', None)
    session.pop('is_logged_in', None)

    # 或者清空整个 session
    # session.clear() 

    return redirect(url_for('index'))

   # 仅通过 HTTPS 传输 Cookie
   app.config['SESSION_COOKIE_SECURE'] = True 
   # 禁止 JavaScript 读取 Cookie (防 XSS 攻击)
   app.config['SESSION_COOKIE_HTTPONLY'] = True 
   # 限制 Cookie 只能同源发送 (防 CSRF 攻击)
   app.config['SESSION_COOKIE_SAMESITE'] = 'Lax' 

pip install Flask-Session redis

from flask import Flask, session
from flask_session import Session
import redis

app = Flask(__name__)

# 配置 Flask-Session 使用 Redis 作为后端
app.config['SESSION_TYPE'] = 'redis'
app.config['SESSION_REDIS'] = redis.from_url('redis://127.0.0.1:6379')
app.config['SECRET_KEY'] = 'super-secret-key'

# 初始化扩展
Session(app)

@app.route('/')
def index():
    # 用法与默认 Session 完全一样！
    session['huge_data'] = 'A' * 10000  # 现在可以存大数据了
    return 'Data stored in Redis!'