连接预定义数据结构的序列

Question

我一直在努力构建这段代码，我想知道是否还有其他更简单/有效的方法来执行此操作：

fsSchema = {'published': {'renders': {'SIM': ('fold1', 'fold2'), 'REN': ('fold1', 'fold2')}}}


def __buildPathFromSchema(self, schema, root=''):        
    metaDirs = []
    for dir_ in schema.keys():
        root = os.path.join(root, dir_)
        if isinstance(schema[dir_], dict):
            return self.__buildPathFromSchema(schema[dir_], root)

        if isinstance(schema[dir_], tuple):
            for i in schema[dir_]:
                bottom = os.path.join(root, i)
                metaDirs.append(bottom)
            root = os.sep.join(os.path.split(root)[:-1])
    return metaDirs

基本上我想要做的是从 fsSchema 等预定义结构生成路径。请注意，最新的迭代始终是一个元组。

输出看起来像：

['已发布\渲染\REN\fold1', '已发布\渲染\REN\fold2', '已发布\渲染\SIM\fold1', '已发布\渲染\SIM\fold2']

谢谢！

Answer 1

您可以使用递归函数来生成所有路径：

def flatten(data):
   if isinstance(data, tuple):
      for v in data:
         yield v
   else:
      for k in data:
         for v in flatten(data[k]):
            yield k + '\\' + v

这应该能够处理任何类型的嵌套字典：

>>> fsSchema = {'published': {'renders': {'SIM': ('fold1', 'fold2'), 'REN': ('fold1', 'fold2')}}}
>>> list(flatten(fsSchema))
['published\\renders\\REN\\fold1', 'published\\renders\\REN\\fold2', 'published\\renders\\SIM\\fold1', 'published\\renders\\SIM\\fold2']

请注意，路径是按“随机”顺序生成的，因为字典没有任何内部排序。

Answer 2

而不是：

for dir_ in schema.keys():
    ...
    if isinstance(schema[dir_], dict):

你可以这样做：

for dir_name, dir_content in schema.iteritems():
    ...
    if isinstance(dir_content, tuple):

它更快，更具可读性。

Answer 3

我会像您一样继续递归地执行此操作，但将步行器与路径生成器分开：

def walk(data):
    if hasattr(data, 'items'):
        for outer_piece, subdata in data.items():
            for inner_piece in walk(subdata):
                yield (outer_piece, ) + inner_piece
     else:
         for piece in data:
             yield (piece, )

def paths(data):
    for path in walk(data):
        yield os.sep.join(path)

原因是它实际上是两个独立的功能部分，并且将它们实现为单独的函数，因此更容易调试、维护、实现，并且只是一般都会想到。