Python 目录遍历

 

import os,sys
import re

def walkdir(dirname,dep=1):
    if(dep<=0):
        return
    try:
        ls=os.listdir(dirname)
    except:
        print ('access deny '+dirname)
    else:
        for l in ls:
            temp=os.path.join(dirname,l)        
            if(os.path.isdir(temp)):
                walkdir(temp,dep-1)
            else:
                if(re.search(r".*\.txt",temp)):
                    print(temp)

if(__name__=="__main__"):
    r=input("type a directory name:")
    walkdir(r,2)
    for root,dirs,files in os.walk(r):
        for f in files:
            if(re.search(r".*\.txt",f)):
                print(root,f)