利用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()函数用于获取目录列表。使用该函数可以方便地获取指定目录下的所有文件和文件夹,并进行进一步的操作。
