使用 dict 的 dict 创建命名空间

Question

目前我正在使用json将 dict 保存到配置文件中。 我加载它以将其转换为dict然后将其转换为SimpleNamespace因为我更喜欢使用点表示法来访问设置。 为此，我按照本示例加载它：

import json
from types import SimpleNamespace
SETTINGS = json.load(open("config.json", 'r'))
SETTINGS = SimpleNamespace(**SETTINGS)

但是，由于我目前正在将dict加载到SimpleNamespace因此它没有在配置文件中加载子dict 。 例如，如果我这样做：

SETTINGS.server_info.port

我收到错误：

AttributeError: 'dict' object has no attribute 'port'

我想知道如何将所有字典作为名称空间加载到命名空间中，以便我能够在字典中一直使用点表示法。

Answer 1

没有必要将转换递归地应用于json.load返回的数据。 您可以通过向json.load调用提供object_hook方法来简单地要求json.load返回SimpleNamespace实例而不是字典。 此方法“将使用解码的每个 JSON 对象的结果调用，其返回值将代替给定的dict 。” （来自文档）。

最简单的object_hook可能如下所示：

def dict_to_sns(d):
    return SimpleNamespace(**d)

例如，给定以下输入：

{
  "settings": {
    "foo": {
      "number": 4,
      "size": "large"
    },
    "bar": {
      "color": "orange",
      "widgets": [
        "gizmo",
        "gadget",
        "thing"
      ]
    }
  }
}

我们可以执行以下操作：

>>> import json
>>> from types import SimpleNamespace
>>> def dict_to_sns(d):
...     return SimpleNamespace(**d)
... 
>>> with open('settings.json') as fd:
...     data = json.load(fd, object_hook=dict_to_sns)
... 
>>> data
namespace(settings=namespace(bar=namespace(color='orange', widgets=['gizmo', 'gadget', 'thing']), foo=namespace(number=4, size='large')))
>>> data.settings.foo
namespace(number=4, size='large')
>>> data.settings.foo.number
4
>>> data.settings.bar.widgets
['gizmo', 'gadget', 'thing']

Answer 2

您必须递归地将SimpleNamespace类应用于嵌套字典； 对于这种情况，我更喜欢使用@functools.singledispatch() ：

from functools import singledispatch
from types import SimpleNamespace

@singledispatch
def wrap_namespace(ob):
    return ob

@wrap_namespace.register(dict)
def _wrap_dict(ob):
    return SimpleNamespace(**{k: wrap_namespace(v) for k, v in ob.items()})

@wrap_namespace.register(list)
def _wrap_list(ob):
    return [wrap_namespace(v) for v in ob]

然后将其用作：

with open('config.json') as settings_file:
    SETTINGS = wrap_namespace(json.load(settings_file))

演示：

>>> SETTINGS = wrap_namespace({'foo': 'bar', 'ham': {'spam': 'eggs', 'monty': [{'name': 'Eric Idle'}]}})
>>> SETTINGS.foo
'bar'
>>> SETTINGS.ham.monty[0].name
'Eric Idle'