如果cookie值是字符串,则值错误,但如果是int,则不是

时间:2015-11-03 01:51:34

标签: python json python-3.x cookies flask

在我的make_session函数中,如果表单中的密码与某些凭据不匹配,则返回一个字符串,最终存储为名为session的cookie。在我的索引文件中,我将它设置在index.html上,如果cookie值为{},则会询问登录信息。如果我通过替换"无效"来改变从make_session函数返回的内容。使用任何整数,代码按预期工作。

我的问题的一个不好的解决方案是添加

except:
    data = {}

在返回我的get_saved_data函数设置数据等于{}之前,但结果与我在浏览器中根本没有cookie的结果相同,但它摆脱了我的错误:{{3 }}

我怀疑get_saved_data函数可能是错误。

我的目录地图:

                         |---- layout.html
                         |---- index.html
         |--- templates -|
Project -|
         |--- test.py

test.py:

from flask import Flask, render_template, redirect, url_for, request, make_response
import json

def get_saved_data(key):
    try:
        data = json.loads(request.cookies.get(key))
    except TypeError:
        data = {}
    return data


def make_session(form_data):
    if form_data.get('username') == "username" and form_data.get('password') == "password":
        return "12345"
    else:
        return "invalid"


app = Flask(__name__)


@app.route('/')
def index():
    data = get_saved_data("session")
    return render_template('index.html', saves=data)


@app.route('/login', methods=['POST'])
def login():
    response = make_response(redirect(url_for('index')))
    response.set_cookie("session", make_session(dict(request.form.items())))
    return response

app.run(debug=True, host='0.0.0.0', port=8000)

的index.html:

{% extends "layout.html" %}

{% block content %}
    {% if saves == {}: %}
        <p>Please log in.</p>
    {% else: %}
        <p>Your Session value is: {{ saves }}</p>
    {% endif %}
    {% if saves == {}: %}
        <form action="{{ url_for('login') }}" method="POST">
            <p>We take your private information very seriously. All data is encrypted not once but twice! in ROT13 to provide the best security.</p><br />
            <label for="username">Please enter your username:</label>
            <input type="text" name="username" /><br />
            <label for="password">Please enter your password:</label>
            <input type="text" name="password" /><br />
            <button class="btn">Log In</button>
        </form>
    {% endif %}
{% endblock %}

的layout.html:

<!DOCTYPE html>
<html>
    <head>
        <title>Character Generator</title>
    </head>
    <body>
        {% block content %}{% endblock %}
    </body>
</html>

2 个答案:

答案 0 :(得分:1)

将您的get_saved_data更改为以下内容会有什么不同吗?

def get_saved_data(key):
    data = json.loads(request.cookies.get(key, '{}'))
    return data

如果没有这样的密钥则返回'{}',确保json不会抱怨任何内容。

答案 1 :(得分:0)

虽然我还没有找到直接解决问题的方法,并且使用lord63.j的答案为我节省了几行代码,间接解决问题的方法是保存cookie值通过将登录功能修改为字典作为字典:

@app.route('/login', methods=['POST'])
def login():
    response = make_response(redirect(url_for('index')))
    response.set_cookie("data", json.dumps({"session": make_session(dict(request.form.items()))}))
    return response

从他们开始,为了访问会话值,我需要使用

saves.get('session')

因为保存相当于

json.loads(request.cookies.get('data', '{}'))

在索引函数

@app.route('/')
def index():
    data = get_saved_data("data")
    return render_template('index.html', saves=data)