使用Python递归在根目录下面写入文件

         有这么个需求，那就是在给定目录（例如：/dirA）下的所有子目录的根目录下创建相同的一个文件，有多层目录嵌套的情况，那么需要解决这个问题就需要通过递归的方式来解决。

为了复现这个问题，首先需要递归的创建空目录，那么我们就创建宽度为三，深度为三的目录。

mkdir_recursive函数：

def mkdir_recursive(path,width,depth):
    if depth == 0:
        return
    for inc in range(width):
        dirname = 'D'+str(depth)+'W'+str(inc)
        os.system('mkdir -p ' + path + os.sep + dirname)
        path_temp = os.sep.join([path,dirname])
        mkdir_recursive(path_temp,width,depth-1)

函数的原型就是通过for循环遍历每一个目录（在这里是创建每一个目录），在每一个目录下再递归创建，递归函数的出口在于深度的依次递减。

递归添加文件函数traverse_root_dir_add：

def traverse_root_dir_add(path):
    dirs = os.listdir(path)
    if dirs == [] or len(dirs) == 0:
        cmd = 'touch ' + path + os.sep + 'abcd'
        print cmd
        os.system(cmd)
        return
    for file in dirs:
        path_temp = os.sep.join([path,file])
        if os.path.isdir(path_temp):
            traverse_root_dir_add(path_temp)

函数的原型为列出当前目录path下的所有目录，如果是根目录（即没有子目录），那么创建文件，且返回；如果不是那么递归调用所有目录。

递归删除文件函数traverse_root_dir_remove：

def traverse_root_dir_remove(path):
    dirs = os.listdir(path)
    if dirs==[] or len(dirs)==0:
        return

    for file in dirs:
        path_temp = os.sep.join([path,file])
        if os.path.isdir(path_temp):
            traverse_root_dir_remove(path_temp)
        else:
            cmd = 'rm -rf ' + path_temp
            print cmd
            os.system(cmd)

对于删除文件，要做的判断就是判断根目录下是文件的情况下将其删除。

Python获取当前文件绝对路径的方式是：

dirname = os.path.dirname(os.path.realpath(__file__))  

dirname1 = os.path.realpath(os.path.dirname(__file__))

那么整体代码是：

#!/usr/bin/env python                                                                                                                                                                    

import sys
import os

default_path = os.path.realpath( os.path.dirname(__file__) )

def traverse_root_dir_remove(path):
    dirs = os.listdir(path)
    if dirs==[] or len(dirs)==0:
        return

    for file in dirs:
        path_temp = os.sep.join([path,file])
        if os.path.isdir(path_temp):
            traverse_root_dir_remove(path_temp)
        else:
            cmd = 'rm -rf ' + path_temp
            print cmd
            os.system(cmd)


def traverse_root_dir_add(path):
    dirs = os.listdir(path)
    if dirs == [] or len(dirs) == 0:
        cmd = 'touch ' + path + os.sep + 'abcd'
        print cmd
        os.system(cmd)
        return
    for file in dirs:
        path_temp = os.sep.join([path,file])
        if os.path.isdir(path_temp):
            traverse_root_dir_add(path_temp)

def mkdir_recursive(path,width,depth):
    if depth == 0:
        return
    for inc in range(width):
        dirname = 'D'+str(depth)+'W'+str(inc)
        os.system('mkdir -p ' + path + os.sep + dirname)
        path_temp = os.sep.join([path,dirname])
        mkdir_recursive(path_temp,width,depth-1)

if __name__ == "__main__":
	if len(sys.argv) == 1:
		path = default_path
	elif len(sys.argv) == 2:
		path = sys.argv[1]
	else:
		exit(-1)

	mkdir_recursive(path,3,3)
	traverse_root_dir_add(path)
	traverse_root_dir_remove(path)

Author：忆之独秀

Email：leaguenew@qq.com

注明出处：http://blog.csdn.net/lavorange/article/details/50409469