文件中替换字符串

233倔强不秃

1. """replace strings in text"""
   
2. 
   
3. import os
   
4. 
   
5. def Replace(file_name, rep_word, new_word):
   
6.     with open(file_name) as f:
   
7.         content = []
   
8.         count = 0
   
9. 
   
10.         for eachline in f:
    
11.             if rep_word in eachline:
    
12.                 count += eachline.count(rep_word)
    
13.                 eachline = eachline.replace(rep_word, new_word)
    
14.             content.append(eachline)
    
15. 
    
16.         decide = input('文件 {0} 中共有{1}个【{2}】\n您确定要把所有的【{3}】替换为【{4}】吗？\n【YES/NO】：'.format\
    
17.                 (file_name, count, rep_word, rep_word, new_word))
    
18. 
    
19.         if decide in ['YES', 'Yes', 'yes']:
    
20.             with open(file_name, 'w') as f:
    
21.                 f.writelines(content)
    
22.             print('Succeed!')
    
23.         else:
    
24.             print('Exit!')
    
25. 
    
26. if __name__ == '__main__':
    
27.     while True:
    
28.         file_name = input('请输入文件名：')
    
29. 
    
30.         if file_name in os.listdir():
    
31.             rep_word = input('请输入需要替换的单词或字符：')
    
32.             new_word = input('请输入新的单词或字符：')
    
33.             Replace(file_name, rep_word, new_word)
    
34.             break
    
35.         else:
    
36.             print('Do not find such a file {}'.format(file_name))

复制代码