python 查询整个磁盘或目录系统，并将文件和文件夹信息收集到对应的字典格式中

查询整个磁盘或目录系统，并将文件和文件夹信息收集到对应的字典格式中：

python
import os

def collect_directory_info(path):
    result = {'files': [], 'directories': []}

    # 遍历目录
    for root, dirs, files in os.walk(path):
        for file in files:
            file_path = os.path.join(root, file)
            file_info = {
                'name': file,
                'path': file_path,
                'size': os.path.getsize(file_path),
                'extension': os.path.splitext(file)[1]
            }
            result['files'].append(file_info)

        for dir in dirs:
            dir_path = os.path.join(root, dir)
            dir_info = {
                'name': dir,
                'path': dir_path,
                'size': 0,
                'extension': ''
            }
            result['directories'].append(dir_info)

    return result
使用上述函数，你可以提供一个路径作为参数，例如path = '/home/user/Documents'，然后调用collect_directory_info(path)函数，它将返回一个包含文件和文件夹信息的字典。其中，字典的files键对应一个列表，列表中的每个元素都是一个文件的字典，包含文件名、路径、大小和扩展名等信息。字典的directories键也对应一个列表，列表中的每个元素都是一个文件夹的字典，包含文件夹名、路径、大小和扩展名（为空）等信息。

请注意，这个函数会递归地遍历目录，因此可能需要一些时间来完成，尤其是当目录结构较大时。在执行之前，请确保有足够的权限来访问指定的路径