替换指定目录下的文本内容

# coding=utf-8
# @author:Damon Qu
# @contact:qufengw@126.com
# @date:2018/5/23
import os
import time
'''
替换指定目录下的文本内容
'''


def replace_file(dir_path, old_str, new_str):
    count = 0
    for file in os.listdir(dir_path):
        file_path = "%s/%s" % (dir_path, file)
        if os.path.isfile(file_path):
            s1 = time.time()
            print('%s is run...' % file)
            count += 1
            f1 = open(file_path, 'r+')
            infos = f1.readlines()
            f1.seek(0,0)
            for line in infos:
                line_new = line.replace(old_str, new_str)
                f1.write(line_new)
            f1.close()
            s2 = time.time()
            print('%s has replaced, %ds' % (file, s2-s1))
    print("run over, the count of files is: %d" % count)


print("替换指定目录下的文本内容")
dir_path = input("dir_path: ")
old_str = input("enter old string: ")
new_str = input("enter new string: ")
replace_file(dir_path, old_str, new_str)