在《Python中文件名和路径的操作》中提到,os模块中的函数可以对文件进行操作。通过递归以及os模块中提供的函数,可以获取指定目录下所有的文件名。
1 基本流程
通过递归获取指定目录下所有文件名的基本流程,如图1所示。
图1 基本流程
2 函数实现
2.1 定义函数
定义一个名为walk的函数,该函数有一个参数dirname,表示指定的当前目录,代码如下所示:
def walk(dirname):
2.2 获取并访问当前目录下的所有子文件夹名及文件名
在《Python中文件名和路径的操作》中提到,可以通过os.listdir()函数获取指定目录下的所有子文件夹名及文件名,因此,在walk()函数内容,有如下代码:
for name in os.listdir(dirname):
以上代码通过for循环访问了os.listdir()函数获取到的所有子文件夹名和文件名,并将其保存到变量name中。
2.3 获取文件的绝对路径
在for循环内部,通过os.path.join()函数,获取文件的绝对路径,代码如下所示:
path = os.path.join(dirname, name)
其中,os.path.join()函数作用是将文件夹名和文件名进行连接,参数dirname是walk()函数的参数,即是文件夹名;参数name是os.listdir()函数获取的子文件夹名或文件名。最后,获取到的绝对路径保存在变量path中。
2.4 判断是否是文件
在for循环内部,获取到了绝对路径path之后,接下来判断path是否是文件,代码如下所示:
if os.path.isfile(path):
print(path)
else:
walk(path)
以上代码中,如果path是文件,则打印出该文件名;如果path不是文件,则说明是子文件夹,递归调用walk()函数,显示path子文件夹中的所有文件名。
3 函数调用
要想获取当前Python源文件所在目录的所有文件名,可以通过如下代码实现:
cwd = os.getcwd()
walk(cwd)
其中,os.getcwd()函数获取到当前Python源文件所在目录,将该目录作为参数传递给walk()函数,获取到该目录下的所有文件名。
4 完整代码
该程序的完整代码如下所示:
import os
def walk(dirname):
for name in os.listdir(dirname):
path = os.path.join(dirname, name)
if os.path.isfile(path):
print(path)
else:
walk(path)
cwd = os.getcwd()
walk(cwd)