python实现的用于搜索文件并进行内容替换的类实例

这篇文章主要介绍了python实现的用于搜索文件并进行内容替换的类,涉及Python针对文件及字符串的相关操作技巧,需要的朋友可以参考下

本文实例讲述了python实现的用于搜索文件并进行内容替换的类。分享给大家供大家参考。具体实现方法如下：

1. #!/usr/bin/python -O 
2. # coding: UTF-8 
3. """ 
4. -replace string in files (recursive) 
5. -display the difference. 
6. v0.2 
7. - search_string can be a re.compile() object -> use re.sub for replacing 
8. v0.1 
9. - initial version 
10. Useable by a small "client" script, e.g.: 
11. ------------------------------------------------------------------------------- 
12. #!/usr/bin/python -O 
13. # coding: UTF-8 
14. import sys, re 
15. #sys.path.insert(0,"/path/to/git/repro/") # Please change path 
16. from replace_in_files import SearchAndReplace 
17. SearchAndReplace( 
18. search_path = "/to/the/files/", 
19. # e.g.: simple string replace: 
20. search_string = 'the old string', 
21. replace_string = 'the new string', 
22. # e.g.: Regular expression replacing (used re.sub) 
23. #search_string = re.compile('{% url (.*?) %}'), 
24. #replace_string = "{% url '/g<1>' %}", 
25. search_only = True, # Display only the difference 
26. #search_only = False, # write the new content 
27. file_filter=("*.py",), # fnmatch-Filter 
28. ) 
29. ------------------------------------------------------------------------------- 
30. :copyleft: 2009-2011 by Jens Diemer 
31. """ 
32. __author__ = "Jens Diemer" 
33. __license__ = """GNU General Public License v3 or above - 
34. http://www.opensource.org/licenses/gpl-license.php""" 
35. __url__ = "http://www.jensdiemer.de" 
36. __version__ = "0.2" 
37. import os, re, time, fnmatch, difflib 
38. # FIXME: see http://stackoverflow.com/questions/4730121/cant-get-an-objects-class-name-in-python 
39. RE_TYPE = type(re.compile("")) 
40. class SearchAndReplace(object): 
41. def __init__(self, search_path, search_string, replace_string, 
42. search_only=True, file_filter=("*.*",)): 
43. self.search_path = search_path 
44. self.search_string = search_string 
45. self.replace_string = replace_string 
46. self.search_only = search_only 
47. self.file_filter = file_filter 
48. assert isinstance(self.file_filter, (list, tuple)) 
49. # FIXME: see http://stackoverflow.com/questions/4730121/cant-get-an-objects-class-name-in-python 
50. self.is_re = isinstance(self.search_string, RE_TYPE) 
51. print "Search '%s' in [%s]..." % ( 
52. self.search_string, self.search_path 
53. ) 
54. print "_" * 80 
55. time_begin = time.time() 
56. file_count = self.walk() 
57. print "_" * 80 
58. print "%s files searched in %0.2fsec." % ( 
59. file_count, (time.time() - time_begin) 
60. ) 
61. def walk(self): 
62. file_count = 0 
63. for root, dirlist, filelist in os.walk(self.search_path): 
64. if ".svn" in root: 
65. continue 
66. for filename in filelist: 
67. for file_filter in self.file_filter: 
68. if fnmatch.fnmatch(filename, file_filter): 
69. self.search_file(os.path.join(root, filename)) 
70. file_count += 1 
71. return file_count 
72. def search_file(self, filepath): 
73. f = file(filepath, "r") 
74. old_content = f.read() 
75. f.close() 
76. if self.is_re or self.search_string in old_content: 
77. new_content = self.replace_content(old_content, filepath) 
78. if self.is_re and new_content == old_content: 
79. return 
80. print filepath 
81. self.display_plaintext_diff(old_content, new_content) 
82. def replace_content(self, old_content, filepath): 
83. if self.is_re: 
84. new_content = self.search_string.sub(self.replace_string, old_content) 
85. if new_content == old_content: 
86. return old_content 
87. else: 
88. new_content = old_content.replace( 
89. self.search_string, self.replace_string 
90. ) 
91. if self.search_only != False: 
92. return new_content 
93. print "Write new content into %s..." % filepath, 
94. try: 
95. f = file(filepath, "w") 
96. f.write(new_content) 
97. f.close() 
98. except IOError, msg: 
99. print "Error:", msg 
100. else: 
101. print "OK" 
102. print 
103. return new_content 
104. def display_plaintext_diff(self, content1, content2): 
105. """ 
106. Display a diff. 
107. """ 
108. content1 = content1.splitlines() 
109. content2 = content2.splitlines() 
110. diff = difflib.Differ().compare(content1, content2) 
111. def is_diff_line(line): 
112. for char in ("-", "+", "?"): 
113. if line.startswith(char): 
114. return True 
115. return False 
116. print "line | text/n-------------------------------------------" 
117. old_line = "" 
118. in_block = False 
119. old_lineno = lineno = 0 
120. for line in diff: 
121. if line.startswith(" ") or line.startswith("+"): 
122. lineno += 1 
123. if old_lineno == lineno: 
124. display_line = "%4s | %s" % ("", line.rstrip()) 
125. else: 
126. display_line = "%4s | %s" % (lineno, line.rstrip()) 
127. if is_diff_line(line): 
128. if not in_block: 
129. print "..." 
130. # Display previous line 
131. print old_line 
132. in_block = True 
133. print display_line 
134. else: 
135. if in_block: 
136. # Display the next line aber a diff-block 
137. print display_line 
138. in_block = False 
139. old_line = display_line 
140. old_lineno = lineno 
141. print "..." 
142. if __name__ == "__main__": 
143. SearchAndReplace( 
144. search_path=".", 
145. # e.g.: simple string replace: 
146. search_string='the old string', 
147. replace_string='the new string', 
148. # e.g.: Regular expression replacing (used re.sub) 
149. #search_string = re.compile('{% url (.*?) %}'), 
150. #replace_string = "{% url '/g<1>' %}", 
151. search_only=True, # Display only the difference 
152. # search_only = False, # write the new content 
153. file_filter=("*.py",), # fnmatch-Filter 
154. ) 

希望本文所述对大家的Python程序设计有所帮助。