Flask - Cookie的使用
用户使用浏览器第一次访问服务端时,服务端生成cookie,放在响应response里返回,并保存在用户浏览器里.
from flask import Flask,make_response@app.route('/set/<name>')def set_cookie(name): response = make_response(redirect(url_for('hello'))) response.set_cookie('name',name) # set_cookie视图会在生成的响应报文首部中创建一个Set-Cookie字段,即"Set-Cookie: name=xxx;Path=/" return response
用户下次访问同一个服务端时,会在header里携带已保存的cookie.
from flask import Flask,request@app.route('/')@app.route('/hello')def hello(): name = request.args.get('name') #如果查询参数里没有name值则从cookie获取 if name is None: name = request.cookies.get('name','Human') #从Cookie中获取name值 return '<h1>Hello,%s</h1>' % name
因为用户可以直接修改浏览器的cookie值,冒用他人账户,所以直接使用cookie是不安全的.
Flask提供了session对象用来将Cookie数据加密存储.(默认情况下,它会把数据存储在浏览器上一个名为session的cookie里)
app.secret_key = 'secret string' #设置密钥
更安全的做法是把密钥写进系统环境变量(在命令行中使用export或set命令),或是保存在.env文件中
SECRET_KEY=secret string
然后在程序脚本中使用os模块提供的getenv()方法获取:
app.secret_key = os.getenv('SECRET_KEY','secret string') #第二个参数是作为没有获取到对应环境变量时使用的默认值`
#结合使用@app.route('/login')def login(): session['logged_in'] = True #写入session. 向session中添加一个名为logged_in的cookie,将它的值设为True. #使用session对象添加cookie时,数据会使用程序的密钥对其进行签名,加密后的数据存储在一块名为session的cookie里. 用户可以看到加密后的值,但无法修改它,一旦数据被修改,签名的值也会变化,这样在读取时会验证失败,对应的session值也会随之失效. return redirect(url_for('hello'))@app.route('/hello')def hello(): name = request.args.get('name') if name is None: name = request.cookies.get('name','Human') response = '<h1>Hello, %s!</h1>' % name #session中的数据可以像字典一样通过键读取,或是使用get()方法. if 'logged_in' in session: #根据用户认证状态返回不同的内容. response += '[Authenticated]' else: response += '[Not Authenticated]' return response
登出用户的账户,实际操作其实就是把代表用户认证的logged_in cookie删除,通过session对象的pop方法实现.
from flask import session@app.route('/logout')def logout(): if 'logged_in' in session: session.pop('logged_in') return redirect(url_for('hello'))
默认情况下,session cookie会在用户关闭浏览器时删除.
通过将session.permanent属性设置为True可以将session的有效期延长为Flask.permanent_session_lifetime属性值对应的datetime.timedelta对象.
也可以通过配置变量PERMANENT_SESSION_LIFETIME设置,默认为31天.
尽管session对象会对Cookie进行签名并加密,但这种方式仅能够确保session的内容不会被篡改,加密后的数据借助工具仍然可以轻易读取(即时不知道密钥),所以绝对不能在session中存储用户密码等敏感信息.
