我正在尝试将部分 Python 2.6 代码升级到 Python 2.7。此代码使用该json
模块生成一些 JavaScript(不兼容 JSON),然后将其插入脚本的其余部分。
一般的想法是能够插入代码或引用在别处定义的变量:它不打算用作 JSON 数据,而是 JavaScript 代码。
这是适用于 Python 2.6 的自定义编码器:
import json
class RawJavaScriptText:
def __init__(self, jstext):
self._jstext = jstext
def get_jstext(self):
return self._jstext
class RawJsJSONEncoder(json.JSONEncoder):
def _iterencode_default(self, o, markers=None):
if isinstance(o, RawJavaScriptText):
yield self.default(o)
else:
json.JSONEncoder._iterencode_default(self, o, markers)
def default(self, o):
if isinstance(o, RawJavaScriptText):
return o.get_jstext()
else:
return json.JSONEncoder.default(self, o)
testvar = {
'a': 1,
'b': 'abc',
# RawJavaScriptText will be inserted as such, no serialisation.
'c': RawJavaScriptText('function() { return "Hello World"; }'),
'd': RawJavaScriptText('some_variable_name')
}
print json.dumps(testvar, cls=RawJsJSONEncoder)
使用 Python 2.6,我们得到了所需的结果:
{ "a": 1, "c": function() { return "Hello World"; },
"b": "abc", "d": some_variable_name }
使用 Python 2.7,一切都变成了字符串,从而失去了 JavaScript 代码的有效性:
{ "a": 1, "c": "function() { return \"Hello World\"; }",
"b": "abc", "d": "some_variable_name" }
(作为旁注,这仅与一组预定义的原始 JavaScript 值一起使用,以防止潜在的注入或滥用。)
当然,这样做的原因是Python 2.7 版本的模块中不存在_iterencode_default
方法。诚然,它一开始并不是要被覆盖的。JSONEncoder
json
在 Python 2.7 中是否有另一种方法可以实现这一目标?使用 JSON 库的基础能够以这种方式生成 JavaScript 代码是相当方便的。
编辑:这是完整的工作解决方案,按照 James Henstridge 的建议使用替换。我正在为替换令牌使用随机 UUID,这应该可以防止任何冲突。这样,这是一个直接替代 Python 2.6 和 2.7 的版本。
import json
import uuid
class RawJavaScriptText:
def __init__(self, jstext):
self._jstext = jstext
def get_jstext(self):
return self._jstext
class RawJsJSONEncoder(json.JSONEncoder):
def __init__(self, *args, **kwargs):
json.JSONEncoder.__init__(self, *args, **kwargs)
self._replacement_map = {}
def default(self, o):
if isinstance(o, RawJavaScriptText):
key = uuid.uuid4().hex
self._replacement_map[key] = o.get_jstext()
return key
else:
return json.JSONEncoder.default(self, o)
def encode(self, o):
result = json.JSONEncoder.encode(self, o)
for k, v in self._replacement_map.iteritems():
result = result.replace('"%s"' % (k,), v)
return result
testvar = {
'a': 1,
'b': 'abc',
'c': RawJavaScriptText('function() { return "Hello World"; }'),
'd': [ RawJavaScriptText('some_variable_name') ],
'e': {
'x': RawJavaScriptText('some_variable_name'),
'y': 'y'
}
}
print json.dumps(testvar, cls=RawJsJSONEncoder)
结果(2.6 和 2.7):
{"a": 1, "c": function() { return "Hello World"; },
"b": "abc",
"e": {"y": "y", "x": some_variable_name},
"d": [some_variable_name]}