在linux系统下遍历文件不算是新鲜功能,如果给出一个目录,要根据这个目录的结构去做某些操作,这个就要到遍历了,谈到遍历就不能不说递归,因为遍历是递归的典型应用场景,那什么是递归?其实递归就是定义的函数又调用了自己,比如列出指定目录下的所有文件,如果这个目录下除了文件还有目录,就就需要再进入这个子目录里,依次类推直到没有目录了结束,递归比较抽象,我们直接上例子说明:
#!/usr/mport os
import os
def Rec(path):
all_files = os.listdir(path)
for file in all_files:
pf = os.path.join(path,file)
if os.path.isdir(pf):
Rec(pf)
else:
print os.path.join(path,pf)
Rec('/root/script')
这就是一个简单的列子,例子中如果变量pf是目录,就再调用Rec函数,如果不是就直接打印出目录和文件内容。
如果以上的例子比较难理解,那大家也不用担心,谁让咱用的是python这门大神的语言呢, 因为在python里已经给我们提供了一个os.walk()函数帮我们解决了这个较为烧脑的问题,这个函数接受一个目录,然后返回一个三元组tupple(dirpath, dirnames, filenames),,分别表示:
根路径,根路径下的子目录列表,根路径下的所有文件列表。
所以要遍历一个目录下文件,最后用os.path.join(dirpath, name).就可以得到文件的全路径了,我们来个实际例子,比如我要备份1台机器上的所有conf文件,我们采用os.walk()函数可以这样遍历所有系统的conf文件,然后拷贝到我们指定的目录中去,脚本如下:
import os, shutil, sys
if not os.geteuid()==0:
sys.exit(0)
if not os.path.isdir("/backup"):
os.mkdir("/backup", 384)
for root, dirs, files in os.walk('/'):
for filename in files:
if ".conf" in filename:
abspath = os.path.join(root, filename)
shutil.copy2(abspath, "/backup")
有了这个os.walk()函数可以帮我们很快遍历指定的文件夹内容,所以在实际中非常有用,建议多写几个例子来深入理解它的含义,这篇就到这里,欢迎留言交流。