简单实用Python程序

# 描述：此为遍历一个文件夹里面的所有*.py文件并将文件名输出到txt文件

import os

import os.path
import re
rootdir = "D:\dpkt-master\dpkt"
fileTest = open(r'D:\filenames.txt', 'w')

pattern = re.compile("[a-z]+\.")
allData = set()
for parent,dirnames,filenames in os.walk(rootdir):
    for dirname in  dirnames:
        print "parent is:" + parent
        print  "dirname is" + dirname


    for filename in filenames:
        match = pattern.match(filename)
        if match:
            allData.add(match.group()[:-1])


for name in allData:
    fileTest.write(name + ',')

fileTest.close()

后续增加