python把目录结构存入内存

原创

mob649e8161738c 2023-12-01 09:25:37 ©著作权

文章标签 目录结构 python os模块 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e8161738c的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python：如何将目录结构存入内存

简介

在Python中，我们可以使用os模块来处理文件和目录。其中，有一个特别常见的需求是将目录结构存入内存以便进一步处理。本文将向你介绍如何使用Python来实现这个功能，并给出详细的代码示例和解释。

流程图

graph TD;
    A[开始]-->B[输入目录路径];
    B-->C[获取目录结构];
    C-->D[将目录结构存入内存];
    D-->E[结束];

步骤解析

输入目录路径：首先，我们需要从用户处获得一个目录的路径。可以通过input()函数来实现。

path = input("请输入一个目录的路径：")

获取目录结构：接下来，我们将使用os模块中的函数来获取目录的结构。os.listdir(path)函数将返回指定目录下的所有文件和文件夹的名称列表。

import os

def get_directory_structure(path):
    """
    获取目录结构
    :param path: 目录路径
    :return: 目录结构
    """
    directory_structure = {}
    for item in os.listdir(path):
        item_path = os.path.join(path, item)
        if os.path.isdir(item_path):
            directory_structure[item] = get_directory_structure(item_path)
        else:
            directory_structure[item] = None
    return directory_structure

directory_structure = get_directory_structure(path)

将目录结构存入内存：我们使用一个字典来存储目录结构。字典的键是目录或文件的名称，字典的值是对应的子目录结构。如果一个键对应的值是None，则表示该键是一个文件而不是目录。

import pprint

pprint.pprint(directory_structure)

完整代码

import os
import pprint

def get_directory_structure(path):
    """
    获取目录结构
    :param path: 目录路径
    :return: 目录结构
    """
    directory_structure = {}
    for item in os.listdir(path):
        item_path = os.path.join(path, item)
        if os.path.isdir(item_path):
            directory_structure[item] = get_directory_structure(item_path)
        else:
            directory_structure[item] = None
    return directory_structure

path = input("请输入一个目录的路径：")
directory_structure = get_directory_structure(path)
pprint.pprint(directory_structure)

代码解释

import os：导入os模块以使用相关的函数；
import pprint：导入pprint模块，以便漂亮地打印出目录结构；
def get_directory_structure(path)：定义一个函数来获取目录结构，其参数为目录的路径；
directory_structure = {}：创建一个字典来存储目录结构；
for item in os.listdir(path):：遍历指定目录下的所有文件和文件夹；
item_path = os.path.join(path, item)：获取每个文件和文件夹的完整路径；
if os.path.isdir(item_path):：判断路径对应的是否为目录；
directory_structure[item] = get_directory_structure(item_path)：如果是目录，递归调用get_directory_structure()函数获取其子目录结构，并将其存储在字典中；
else:：如果是文件，则将其值设为None，表示该键是一个文件而不是目录；
return directory_structure：返回目录结构字典；
path = input("请输入一个目录的路径：")：使用input()函数获取用户输入的目录路径；
directory_structure = get_directory_structure(path)：调用get_directory_structure()函数获取目录结构，并将结果存储在directory_structure变量中；
pprint.pprint(directory_structure)：使用pprint.pprint()函数漂亮地打印出目录结构。