Как преобразовать путь к файлу в древовидную структуру?

Прямо сейчас у меня есть несколько файлов в python, которые имеют следующий формат пути к файлу:

a/b/c/d/e/file1
a/b/c/d/e/file2
a/f/g/h/i/file3
a/f/g/h/i/file4

Теперь, когда я получаю эти пути в Python, я хочу преобразовать их в формат JSON, который может прочитать сторонний плагин jquery (например, fancytree, jqxTree), чтобы преобразовать его в древовидное представление. Может быть, есть другие более простые способы сделать это, пожалуйста, предложите. Спасибо!


person return 0    schedule 05.08.2015    source источник


Ответы (1)


Из Python вы можете сгенерировать нужный JSON следующим образом (обратите внимание, что отступ и сортировка не нужны, если вы просто отправляете JSON через строку, но они делают вывод читаемым для отладки):

import collections
import json

myfiles = '''
    a/b/c/d/e/file1
    a/b/c/d/e/file2
    a/f/g/h/i/file3
    a/f/g/h/i/file4
'''

# Convert to a list
myfiles = myfiles.split()


def recursive_dict():
    """
    This function returns a defaultdict() object
    that will always return new defaultdict() objects
    as the values for any non-existent dict keys.
    Those new defaultdict() objects will likewise
    return new defaultdict() objects as the values
    for non-existent keys, ad infinitum.
    """
    return collections.defaultdict(recursive_dict)


def insert_file(mydict, fname):
    for part in fname.split('/'):
        mydict = mydict[part]

    # If you want to record that this was a terminal,
    # you can do something like this, too:
    # mydict[None] = True

topdict = recursive_dict()
for fname in myfiles:
    insert_file(topdict, fname)

print(json.dumps(topdict, sort_keys=True,
                indent=4, separators=(',', ': ')))

Этот код генерирует следующий вывод:

{
    "a": {
        "b": {
            "c": {
                "d": {
                    "e": {
                        "file1": {},
                        "file2": {}
                    }
                }
            }
        },
        "f": {
            "g": {
                "h": {
                    "i": {
                        "file3": {},
                        "file4": {}
                    }
                }
            }
        }
    }
}
person Patrick Maupin    schedule 06.08.2015
comment
Что делает def recursive_dict(): return collections.defaultdict(recursive_dict)? - person return 0; 10.08.2015