Flask web开发 处理Session

本文我们在上篇文章《Flask web开发  处理POST请求（登录案例）》的基础上，来讲述Flask对session的支持。

在上面案例上，我们需要修改和新增如下功能

1、登录成功后的 url不再是 http://192.168.142.138/home?username=admin

而是http://192.168.142.138/home

其中在页面显示的username信息，由模板代码从session中获取

2、当没有登录之前，在浏览器输入http://192.168.142.138/xxxx 地址，会跳转到http://192.168.142.138/login页面，

当登录成功后，会进入http://192.168.142.138/xxxx页面。 其中xxx是已经存在的url。

 

下面我们来一步步的实现（代码在上个案例基础上改进的，这里不再从头说起）：

1、修改run.py文件，修改后的文件内容如下

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

app = Flask(__name__)

@app.route('/login', methods=['POST','GET'])
def login():
    error = None
    if request.method == 'POST':
        if request.form['username']=='admin':
            session['username'] = request.form['username']
            return redirect('/home')
        else:
            error = 'Invalid username/password'
    return render_template('login.html', error=error)

@app.route('/home')
def home():
    return render_template('home.html',username=session['username'])

app.secret_key = 'A0Zr98j/3yX R~XHH!jmN]LWX/,?RT'

if __name__ == '__main__':
    app.debug = True
    app.run('0.0.0.0',80)

修改的内容有：
1）登录成功，将用户名保留到session中，代码如session['username'] = request.form['username']

2）登录成功，重定向到home上，不需要带username查询参数

3）home()方法传入模板的username参数直接从session中使用

4）为了让session有效，需要设置一个key

注意：上面的代码有个问题，如果在登录前，直接在浏览器输入home，是会报错的，因为这时session中的username还不存在。这个问题我们在下面会解决。

 

因为加了app.debug=true的设置，run.py修改后，服务自动重启。这时我们在浏览器中重新访问，发现 登录成功后的 url不再是 http://192.168.142.138/home?username=admin ，而是http://192.168.142.138/home。 这样文章开头说的第一个问题解决了。

下面我们来解决第二个问题。

2、在templates目录下，增加一个test.html文件，内容可以是随意一些文字。

我们下面要解决的是，当在登录之前在浏览器输入 test.html，会跳转到 login页面，登录成功后会自动跳转到test.html页面。

3、在run.py中增加一个对test.html的路由，代码如

@app.route('/test')
def test():
    return render_template('test.html');

4、修改run.py文件，修改后的内容如下

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

app = Flask(__name__)

@app.route('/login', methods=['POST','GET'])
def login():
    error = None
    if request.method == 'POST':
        if request.form['username']=='admin':
            session['username'] = request.form['username']
            if 'newurl' in session:
                newurl = session['newurl']
        session.pop('newurl', None)
                return redirect(newurl)
            else:
                return redirect('/home')
        else:
            error = 'Invalid username/password'
    return render_template('login.html', error=error)

@app.route('/home')
def home():
    return render_template('home.html',username=session['username'])

@app.route('/test')
def test():
    if 'username' in session:
        return render_template('test.html')
    else:
        session['newurl']='test'
        return redirect(url_for('login'))
 
app.secret_key = 'A0Zr98j/3yX R~XHH!jmN]LWX/,?RT'

if __name__ == '__main__':
    app.debug = True
    app.run('0.0.0.0',80)

上面的代码，与前面相比。在test请求的响应处理中，判断session中是否存在username，不存在就跳转到login页面，并把test地址在session中保存。
然后在login的post请求中，检查newurl是否存在，存在的话表示是由别的页面跳转到登录页面的，这样登录成功后跳转到相应页面，缺省是跳转到home页面。

 

这样我们文章开头提的第2个问题看上去也解决了。为什么说看上去呢？ 因为实际是有问题的。 我们这个是对test请求做了特殊处理。我们应该能对所有的请求（包括还存在问题的home请求）做拦截，判断是否登录，如果没登录，就跳转到登录页面。

这个问题我们在后续的文章中处理，处理的核心思路是在请求达到每个具体的路由代码之前，能进行拦截和判断是否登录。