欢迎访问宙启技术站
智能推送

利用Python的list_directory()函数实现目录列表的获取

发布时间:2023-12-11 05:00:57

Python提供了os模块,其中包含了listdir()函数用于获取指定目录下的所有文件和文件夹列表。下面是使用Python的listdir()函数实现目录列表获取的示例:

import os

def list_directory(path):
    try:
        # 获取指定目录下的所有文件和文件夹
        listing = os.listdir(path)
        for item in listing:
            # 判断是否为文件
            if os.path.isfile(os.path.join(path, item)):
                print('File:', item)
            # 判断是否为文件夹
            elif os.path.isdir(os.path.join(path, item)):
                print('Directory:', item)
            else:
                print('Unknown:', item)
    except FileNotFoundError:
        print('Path not found!')

# 测试示例
list_directory('C:/Users/Username/Documents')

在上面的示例中,定义了一个list_directory函数,接受一个参数path,表示要获取列表的目录路径。

函数中,首先调用os.listdir(path)来获取指定目录下的所有文件和文件夹列表。然后遍历列表中的每个元素,使用os.path.join(path, item)来获取文件或文件夹的完整路径。

接着,使用os.path.isfile()函数判断当前元素是否为文件,如果是,则打印"File:"加上文件名,否则使用os.path.isdir()函数判断当前元素是否为文件夹,如果是,则打印"Directory:"加上文件夹名,最后,如果既不是文件也不是文件夹,则打印"Unknown:"加上元素名。

最后,在例子中调用了list_directory函数,并传入了一个示例目录路径。

执行示例代码后,控制台会打印出指定目录下的所有文件和文件夹,每个文件和文件夹前面会有对应的标识。

需要注意的是,如果传入的目录路径不存在,会捕获到FileNotFoundError异常,并打印"Path not found!"。

总结:Python的os模块提供了listdir()函数用于获取目录列表。使用该函数可以方便地获取指定目录下的所有文件和文件夹,并进行进一步的操作。