在 Python 中返回字典的递归函数

Question

我有上面的树。我需要以递归方式搜索树中的目录和文件，并将它们作为字典以以下形式返回-> 键：文件的目录/名称和值：文件的第一行

eg: key:1/2/5/test5    value:first line of test 5

到目前为止，我创建了下一个代码：

def search(root):
    items = os.listdir(root)
    
    for element in items:
        if os.path.isfile(element):
        
            with open (element) as file:
                one_line=file.readline()
                print(one_line)

        elif os.path.isdir(element):
            search(os.path.join(root,element))

问题是我的代码只搜索目录。请让我明白我错在哪里以及如何解决它。非常感谢任何帮助，谢谢！

Answer 1

您的代码几乎是正确的。不过，它必须稍微调整一下。 更具体地说，

1. element 是文件或目录名称（不是路径）。如果它是子目录或子目录中的文件，则if os.path.isfile(element) 和elif os.path.isdir(element) 的值将始终为False。因此，将它们分别替换为if os.path.isfile(os.path.join(root, element)) 和elif os.path.isdir(os.path.join(root, element))。

2. 同样，with open(element) 应替换为 with open(os.path.join(root,element))。

3. 读取文件的第一行时，您必须将路径和该行存储在字典中。

4. 调用elif os.path.isdir(element)中的递归函数时必须更新该字典。

完整的sn-p见下文：

import os

def search(root):

    my_dict = {}   # this is the final dictionary to be populated

    for element in os.listdir(root):
        
        if os.path.isfile(os.path.join(root, element)):
            try: 
                with open(os.path.join(root, element)) as file:
                    my_dict[os.path.join(root, element)] = file.readline() # populate the dictionary
            except UnicodeDecodeError: 
                # This exception handling has been put here to ignore decode errors (some files cannot be read)
                pass

        elif os.path.isdir(os.path.join(root, element)):
            my_dict.update(search(os.path.join(root,element)))  # update the current dictionary with the one resulting from the recursive call

    return my_dict

print(search('.'))

它会打印如下的字典：

{
 "path/file.csv": "name,surname,grade",
 "path/to/file1.txt": "this is the first line of file 1",
 "path/to/file2.py": "import os"
}

为了可读性，os.path.join(root, element)可以存储在一个变量中，那么：

import os

def search(root):

    my_dict = {}   # this is the final dictionary to be populated

    for element in os.listdir(root):
        path = os.path.join(root, element)

        if os.path.isfile(path):
            with open(path) as file:
                my_dict[path] = file.readline()

        elif os.path.isdir(path):
            my_dict.update(search(path))

    return my_dict

print(search('.'))

Answer 2

您可以使用 os.walk

以下函数将不包含空文件夹。

def get_tree(startpath):
    tree = {}
    for root, dirs, files in os.walk(startpath):
        for file in files:
            path = root+"/"+file
            with open(path,'r') as f:
                first_line =  f.readline()
            tree[path] = first_line
    return tree

输出将是这样的：

{
    file_path : first_line_of_the_file,
    file_path2 : first_line_of_the_file2,
    ...
}