Cab*_*abu 5 python session json decoding flask
我正在尝试向Flask JSON编码器/解码器添加一些重载以添加日期时间编码/解码，但只能通过“ hack”成功。
from flask import Flask, flash, url_for, redirect, render_template_string
from flask.json import JSONEncoder, JSONDecoder
template = """
<!DOCTYPE html>
<html><head><title>Test JSON encoder/decoder</title></head><body>
{% with messages = get_flashed_messages(with_categories=true) %}{% if messages %}{% for message in messages %}
<p>Flash: {{ message }}</p>
{% endfor %}{% endif %}{% endwith %}
<p>Flash should be: ['Flash message', 'success']</p>
<p><a href="{{ url_for('index') }}">Try again</a></p>
</body></html>
"""
class CustomJSONEncoder(JSONEncoder):
""" Do nothing custom json encoder """
def default(self, obj):
# My custom logic here
# ...
# or
return super(CustomJSONEncoder, self).defaults(obj)
class CustomJSONDecoder(JSONDecoder):
""" Do nothing custom json decoder """
def __init__(self, *args, **kargs):
_ = kargs.pop('object_hook', None)
super(CustomJSONDecoder, self).__init__(object_hook=self.decoder, *args, **kargs)
def decoder(self, d):
# My custom logic here
# ...
# or
return d
app = Flask(__name__, static_url_path='')
app.config['SECRET_KEY'] = 'secret-key'
app.json_encoder = CustomJSONEncoder
app.json_decoder = CustomJSONDecoder
@app.route('/')
def index():
flash('Flash message', 'success')
return redirect(url_for('display'))
@app.route('/b')
def display():
return render_template_string(template)
if __name__ == '__main__':
app.run(debug=True, port=5200)
hack是我应该像这样从Flask.sessions.TaggedJSONSerializer复制一些代码：
import uuid
from base64 import b64decode
from werkzeug.http import parse_date
from markupsafe import Markup
from flask._compat import iteritems
class CustomJSONDecoder(JSONDecoder):
""" Do nothing custom json decoder """
def __init__(self, *args, **kargs):
_ = kargs.pop('object_hook', None)
super(CustomJSONDecoder, self).__init__(object_hook=self.decoder, *args, **kargs)
def decode(self, d):
# My custom logic here
# ...
# Copy of the code from Flask.sessions.TaggedJSONSerializer(object).loads(self, value).object_hook(obj)
if len(d) == 1:
the_key, the_value = next(iteritems(d))
if the_key == ' t':
return tuple(the_value)
elif the_key == ' u':
return uuid.UUID(the_value)
elif the_key == ' b':
return b64decode(the_value)
elif the_key == ' m':
return Markup(the_value)
elif the_key == ' d':
return parse_date(the_value)
return d
我是否“正确”地进行操作，或者我错过了某些事情？
您可以通过显式调用基类的 default() 方法来使用基类的功能。我已经在自定义 JSONEncoder 中成功完成了此操作：
class CustomJSONEncoder(JSONEncoder):
def default(self, obj):
# Calling custom encode function:
jsonString = HelperFunctions.jsonEncodeHandler(obj)
if (jsonString != obj): # Encode function has done something
return jsonString # Return that
return JSONEncoder.default(self, obj) # else let the base class do the work
但是，在解码器中，您应该记住传递给函数的对象钩子__init__()并从您自己的钩子中调用它：
class CustomJSONDecoder(JSONDecoder):
def __init__(self, *args, **kwargs):
self.orig_obj_hook = kwargs.pop("object_hook", None)
super(CustomJSONDecoder, self).__init__(*args,
object_hook=self.custom_obj_hook, **kwargs)
def custom_obj_hook(self, dct):
# Calling custom decode function:
dct = HelperFunctions.jsonDecodeHandler(dct)
if (self.orig_obj_hook): # Do we have another hook to call?
return self.orig_obj_hook(dct) # Yes: then do it
return dct # No: just return the decoded dict
顺便说一句：您的解码器中有一个拼写错误：您在基类中注册的对象挂钩名为 name self.decoder，但成员定义为def decode(...)（末尾没有r ）。在您的示例中，您注册了一个空钩子，并且decode()永远不应该调用它。