4

我正在尝试将部分 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方法。诚然,它一开始并不是要被覆盖的。JSONEncoderjson

在 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]}
4

1 回答 1

5

当在幕后使用的 C 扩展扩展为涵盖更多编码过程时,您使用的未记录的私有接口似乎已经消失。

RawJavaScriptText一种替代方法是为您的值插入占位符字符串,并对输出进行后处理dumps以将这些占位符转换为您需要的形式。

例如:

>>> data = {'foo': '@@x@@'}
>>> print json.dumps(data)
{"foo": "@@x@@"}
>>> print json.dumps(data).replace('"@@x@@"', 'some_variable_name')
{"foo": some_variable_name}

如果您的 JSON 包含不受信任的数据,您将需要小心这种技术:您不希望处于一个局外人可以意外地将此类占位符添加到输出的位置。

于 2012-11-02T04:43:49.250 回答