python的N个小功能(文件内容的匹配替换)

# -*- coding: utf-8 -*-

"""

Created on Fri Feb 17 20:25:05 2017

@author: who

"""

import os

import os.path

import re

import string

rootdir=r'D:\test'

for parent, dirnames, filenames in os.walk(rootdir):   # 三个参数：分别返回1.父目录 2.所有文件夹名字（不含路径） 3.所有文件名字

try:

for filename in filenames:

filenamepre=os.path.splitext(filename.decode("gbk"))[0];#文件名前缀

filetype=os.path.splitext(filename.decode("gbk"))[1].lower();#文件扩展名

pswpath = os.path.join(parent, filename.decode("gbk"));

tmppath = os.path.join(r'D:\testxx',filename.decode("gbk")) #写到另一个文件夹#

if filetype=='.txt':

a=string.find(filenamepre,'9999')                      ####符合类型的文件####

if a==0:

tmp_file = open(tmppath, "w")

with open(pswpath) as f:

lines = f.readlines()

for line in lines:                  ####一行一行读取 ####

if line.find('aaa') > -1:         ####找到含aaa有的这行,匹配出对应整数数字####

m=re.compile('aaa([0-9]+)')

ms=m.search(line)

print ms.group(1)

line.replace(ms.group(1),filenamepre)   ####进行替换

tmp_file.write(line.replace(ms.group(1),filenamepre))  ###写出替换的该行

else:

tmp_file.write(line)

tmp_file.close()

else:

tmp_file = open(tmppath, "w")

with open(pswpath) as f:

lines = f.readlines()

for line in lines:                  ####一行一行读取 ####

tmp_file.write(line)

tmp_file.close()

except IOError:

pass