【转】使用python 修改文件内容

原文地址：http://blog.csdn.net/xsckernel/article/details/9007517

由于在一个把ads下的arm 工程项目移植到gnu项目时候需要做大量重复的修改，比如把

[plain] view plaincopyprint?

1. ABC     EQU 1

修改为:

[plain] view plaincopyprint?

1. #define     ABC 1

如果用手工一个个修改很浪费时间，所以就用python脚本来做这些工作，发现很容易就搞定了（以前遇到类似问题总是用c代码来写，代码量很多而且容易出错！！）

源代码如下：

[python] view plaincopyprint?

1. def func():
2. ffrom=open("2440init.s","r")
3. fto=open("2440init1.s","w")
4. while True:
5. l = ffrom.readline()
6. if not l:
7. break
8. if ‘EQU‘ in l:
9. temp = l.split("EQU")
10. temp1 = ‘#define  ‘ + temp[0] + temp[1]
11. #print temp1
12. fto.write(temp1)
13. else:
14. temp1 = l
15. fto.write(temp1)
16. if __name__ == "__main__":
17. func()

用一个文件 2440init.s 来测试下：

[plain] view plaincopyprint?

1. abc             EQU             1
2. pds             EQU             9

最终生成的文件2440init1.s 内容如下所示：

[plain] view plaincopyprint?

1. #define  abc                            1
2. #define  pds                            9

前面既然说了是替换文件的内容 ffrom 跟 fto 打开的应该是同一个文件，但是发现 写文件输出流打开后，会自动清空文件(append模式除外) 貌似和java表现一样的。

可以用如下代码完成

[python] view plaincopyprint?

1. def func():
2. input   = open("2440init.s")
3. lines   = input.readlines()
4. input.close()
5. output  = open("2440init.s",‘w‘);
6. for line in lines:
7. #print line
8. if not line:
9. break
10. if ‘EQU‘ in line:
11. temp    = line.split("EQU")
12. temp1   = ‘#define  ‘ + temp[0] + temp[1]
13. output.write(temp1)
14. else:
15. output.write(line)
16. output.close()
17. if __name__ == "__main__":
18. func()

如果一个比较大的工程文件，需要遍历工程中的每一个文件。如果文件中包含指定的字符串比如说 #include "appdef.h"  则将之替换为 #include "datatype.h" :

[python] view plaincopyprint?

1. import os
2. def direc():
3. for d,fd,fl in os.walk(‘/home/shichao/gun-ucos‘):
4. for f in fl:
5. sufix = os.path.splitext(f)[1][1:]
6. if ( (sufix == ‘h‘) or (sufix == ‘c‘) ):
7. #print sufix
8. func(d + ‘/‘ + f)
9. def func(filename):
10. input   = open(filename)
11. lines   = input.readlines()
12. input.close()
13. output  = open(filename,‘w‘)
14. for line in lines:
15. if not line:
16. break
17. if ((‘appdef.h‘ in line) and (‘include‘ in line) ):
18. temp    = line.split("appdef")
19. temp1   = temp[0] + ‘datatype‘ + temp[1]
20. output.write(temp1)
21. else:
22. output.write(line)
23. output.close()
24. if __name__ == "__main__":
25. direc()