创建具有dict的dict的名称空间

Question

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

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

但是，由于我当前正在将加载dict到，SimpleNamespace因此未在配置文件内加载子命令。例如，如果我这样做：

SETTINGS.server_info.port

我得到错误：

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

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

Answer 1

您必须递归地将该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'

Answer 2

不必将转换递归地应用于所返回的数据json.load。您可以通过为调用提供一种方法来简单地要求json.load返回SimpleNamespace实例而不是字典。该方法“将被解码的每个JSON对象的结果调用，并将使用其返回值代替给定。” （来自docs）。object_hookjson.loaddict

最简单的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']