4

我有一本像这样的字典:

{
   "checksum": "b884cbfb1a6697fa9b9eea9cb2054183",
   "roots": {
      "bookmark_bar": {
         "children": [ {
            "date_added": "12989159740428363",
            "id": "4",
            "name": "test2",
            "type": "url",
            "url": "chrome://bookmarks/#1"
         } ],
         "date_added": "12989159700896551",
         "date_modified": "12989159740428363",
         "id": "1",
         "name": "bookmark_bar",
         "type": "folder"
      },
      "other": {
         "children": [ {
            "date_added": "12989159740428363",
            "id": "4",
            "name": "test",
            "type": "url",
            "url": "chrome://bookmarks/#1"
         } ],
         "date_added": "12989159700896557",
         "date_modified": "0",
         "id": "2",
         "name": "aaa",
         "type": "folder"
      },
      "synced": {
         "children": [  ],
         "date_added": "12989159700896558",
         "date_modified": "0",
         "id": "3",
         "name": "bbb",
         "type": "folder"
      }
   },
   "version": 1
}

一切都从“根”开始,它们有两种类型的数据:URL 和文件夹,它们是字典。如果是文件夹,则必须有key 'children',key 的值是一个列表,我们可以在里面放更多的URL 和文件夹。

现在我想遍历这个嵌套字典,获取所有子文件夹中的URL,所以我写了一个函数:

def traverse(dic):
    for i in dic:
        if i['type'] == 'folder':
            for j in traverse(i['children']):
                yield j
        elif i['type'] == 'url':
            yield i

我可以这样使用它:

traverse(dictionary['roots']['bookmark_bar']['children'])

它完美地工作。但它只是生成一个 URL 的字典,我不知道它在哪里。我也想得到路径。我该怎么做?

4

2 回答 2

8

我的用例与您略有不同:我需要将表示客户端设置的可变深度 JSON 结构展平为键值对,以便存储在数据库中。我无法得到 jsbueno 的答案,而且由于我还需要一些可以处理没有明确列出或包含儿童的案例的东西,所以我对其进行了修改以满足我的需要:

def traverse(dic, path=None):
    if not path:
        path=[]
    if isinstance(dic,dict):
        for x in dic.keys():
            local_path = path[:]
            local_path.append(x)
            for b in traverse(dic[x], local_path):
                 yield b
    else: 
        yield path,dic

最终结果是我可以将这样的 JSON 字符串传递给我的脚本(具有可变深度),它将其转换为嵌套的字典:

{
  "servers": {
    "uat": {
      "pkey": true,
      "user": "testval",
      "pass": true
    },
    "dev": {
      "pkey": true,
      "user": "testval",
      "pass": true
    }
  }
}

对它运行上面的生成器会创建一个漂亮的列表,如下所示:

([u'servers', u'uat', u'pkey'], True)
([u'servers', u'uat', u'user'], u'testval')
([u'servers', u'uat', u'pass'], True)
([u'servers', u'dev', u'pkey'], True)
([u'servers', u'dev', u'user'], u'testval')
([u'servers', u'dev', u'pass'], True)

其中,使用类似的东西:

for x in traverse(outobj):
    pprint(('.'.join(x[0]),x[1]))

然后可以转换为我想要的键值对格式,如下所示:

(u'servers.uat.pkey', True)
(u'servers.uat.user', u'testval')
(u'servers.uat.pass', True)
(u'servers.dev.pkey', True)
(u'servers.dev.user', u'testval')
(u'servers.dev.pass', True)

我知道我在接受接受的答案后以这种方式发布,但由于接受的答案对我不起作用,也许这个稍微与结构无关的版本会帮助其他人!

于 2015-08-13T21:44:00.837 回答
1

如果我得到你想要的东西,我不确定,但你可能想要这样做:

def traverse(dic, path=None):
    if not path:
        path = []
    for i in dic:
        local_path = path[:].append(i)
        if i['type'] == 'folder':
            for j in traverse(i['children'], local_path):
                yield j, local_path
        elif i['type'] == 'url':
            yield i, local_path

现在,您的函数会生成项目和一系列键,以在特定位置获取项目。

于 2012-08-13T12:47:55.030 回答