空间管理您的位置: 51Testing软件测试网 » 第四维空间testing » 日志

一个python3遍历目录文件程序

上一篇 / 下一篇 2014-03-27 18:58:38 / 个人分类：Python3

一个python3遍历目录文件程序，可以很方便遍历目录下的文件：把以下代码保存为filelist.py，使用方法在cmd下执行filelist.py 'c:\\'即可遍历整个c盘目录下的文件。

import os, sys
from stat import *

def walktree(top, callback):
'''recursively descend the directory tree rooted at top,
calling the callback function for each regular file'''

    for f in os.listdir(top):
        pathname = os.path.join(top, f)
        try:
            mode = os.stat(pathname).st_mode
        except PermissionError:
            continue
        else:
            if S_ISDIR(mode):
                # It's a directory, recurse into it
                try:
                    walktree(pathname, callback)
                except PermissionError:
                    continue
            elif S_ISREG(mode):
                    # It's a file, call the callback function
                    callback(pathname)
            else:
                # Unknown file type, print a message
                print('Skipping %s' % pathname)

def visitfile(file):
     # print('visiting', file)
      with open("filelist.txt",'a') as f:
          try:
              f.write(file+"\n")
          except Exception:
              pass

if __name__ == '__main__':
walktree(sys.argv[1], visitfile)

收藏举报

TAG:

查看全部评论