Python实现文件夹深度优先遍历

来源:互联网 发布:土豆网视频下载 mac 编辑:程序博客网 时间:2024/06/16 22:23
import osdef dfs_myDir(path, printDir = None, printFile = None):    stack = []    ret = []    stack.append(path)    while len(stack) > 0:        tmp = stack.pop(len(stack) - 1)        if(os.path.isdir(tmp)):            ret.append(tmp)            for item in os.listdir(tmp):                stack.append(os.path.join(tmp, item))            if printDir:                printDir(tmp)        elif(os.path.isfile(tmp)):            ret.append(tmp)            if printFile:                printFile(tmp)    return retdef printDir(path):    print "dir: " + pathdef printFile(path):    print "file: " + pathd = dfs_myDir(r'E:\AndroidAppDevelop', printDir, printFile)

0 0
原创粉丝点击