使用dicts的dict创建命名空间

时间:2018-05-23 14:22:15

标签: python python-3.x dictionary

目前我正在使用json将dict保存到配置文件中。我加载此项将其转换为dict然后将其转换为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'

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

2 个答案:

答案 0 :(得分:4)

不一定以递归方式将转换应用于json.load返回的数据。您可以通过向json.load电话提供SimpleNamespace方法,让object_hook返回json.load个实例而非字典。将使用解码的每个JSON对象的结果调用此方法"并且将使用其返回值代替给定的dict。" (来自docs)。

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

答案 1 :(得分: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'