如何用python在Windows系统下,生成UNIX格式文件
平时测试工作中,少不了制造测试数据。最近一个项目,我就需要制造一批可在UNIX下正确读取的文件。为确保这批文件能从FTP下载成功,开发叮嘱我:“文件中凡是遇到换行,换行符必须是UNIX下的LF,而不是Dos\Windows下的CRLF。”
换行,在普通文档编辑中,就是按下“Enter”键。在编写代码时,就是在文件中写入字符串”\n”。但用python命令w在文件写入换行”\n”以后,由于是Windows系统的原故,会默认把”\n”实际保存为”\r\n”。
对于Windows下为何是”\r\n”,这里引用网上的解释:因为Windows采用了传统的英文打字机模式,它在换行时需分解成两步——回车”\r”和换行”\n”。回车(carriage return,CR)是将小车退回至起点,相当于把光标从行尾切换到行首,换行(line feed,LF)是将小车切换到下一行,这样的组合操作就完成了另起一行的目的。
我们首先确认Windows下的换行是否真的是CRLF:把文件用任意一款文本编辑器(以Notepad++为例)打开,在编辑器的菜单栏中,找到“视图”菜单,点击“显示符号”,勾选“显示所有字符”,就可以显示换行符了。我们在下图中可以看到,在Windows系统下,文件中的换行符确实是CRLF。
如何把CRLF变成LF呢?开发给出了一个解决方案。
Method A:
在Notepad++界面右下角,双击当前文件格式的描述:“Dos\Windows”,点击弹框中的“转换为UNIX格式”,最后保存文件。此时可以看到文件每行结尾都已经展示为“LF”了。
虽然整个转换过程极为简单,但要是有一大批这样的文件需要一个个手动转换,我不禁感慨光(xīn)阴(hǎo)虚(lèi)度,韶(bù)光(xiǎng)易(dòng)逝。
在表达抗议后,开发表示可以优化此方法,即使用文本编辑器的替换功能:
Method B:
使用Notepad++打开原文件,按下组合键Ctrl+F后,在弹框中选择“替换”栏,在“查找目标”输入框填写“\r\n”,“替换为”输入框填写“\n”,“查找模式”选择“扩展(\n\r\t\0\x...)”,点击“替换所有打开文件”后,一一保存文件即可。
然而把所有的文件保存一遍,并不能提升测试的幸福感。我追求的是尽量减少重复而繁重的手工劳动,解放双手,高效测试!类似这样的能力瓶颈,从我进入测试行业开始,就不断挤占着我的业余时间。为了更好的提高自己,系统地学习测试开发所必需的技能,我报名参加了吴老的测试开发培训班。
截至目前,吴老已经授课两个多月,我从无到有,学习了很多基本的python开发知识,也偶尔能接触到一些巧妙的编程思维。正好这次的数据准备,给我提供了一个实践的契机,让我运用最近学习的文件操作知识,体验了一次局部自动化的乐趣。
下面给大家分享一下我这次愉快的实践内容,由于真实的测试数据略有复杂,这里就用简单的两行数字来举个栗子。
首先按上课所讲的内容,依葫芦画瓢,使用Notepad++创建文件并写入文本内容:
1 #coding:utf-8(设定文件编码格式)
2 importos3 importtime4
5 #切换到创建文件目录
6 os.chdir(r"C:\Study\Chestnuts\01data")7
8 #新建创建文件函数
9 defcreate_file():10
11 #构造文件名:以“年月日”为文件名的.dat文件
12 t=time.localtime()13 file_name=time.strftime("%Y-%m-%d",t)+".dat"
14 #创建并打开文件
15 fp=open(file_name,'w+')16 #写入文件内容
17 fp.writelines("12345\n")18 fp.writelines("67890\n")19 #关闭文件
20 fp.close()21
22 #调用函数
23 create_file()
如上图所示,执行脚本就会生成一份原始文件,不过在两行内容的末尾,换行符均显示为CRLF,接下来就是转换的工作了。
Method C:对原文件进行格式转换
如果是对现有的文件进行处理,可以使用Notepad++再编写一个转换的脚本。参考网上的方法,使用rU方式读取文件内容,使用wb方式写入文件内容,如下图所示:
1 #coding:utf-8
2 importos3
4 #文件路径准备
5 route=r"C:\Study\Chestnuts\01data"
6
7 #遍历路径下目录,文件夹,文件
8 for root,dirs,files inos.walk(route):9 #遍历文件
10 for name infiles:11 #归纳文件名特征
12 if name[-3:]=='dat':13 #拼接文件名(目录+文件名称)
14 catalog=os.path.join(root,name)15 #把所有行分割符替换为换行符\n返回.
16 fp=open(catalog,"rU+")17 #读取文件并保存
18 strings=fp.read()19 fp.close()20 #使用二进制写文件
21 fp1=open(catalog,"wb")22 fp1.seek(0)23 fp1.write(strings)24 fp1.flush()25 fp1.close()
执行这个脚本,就能对现有文件格式进行转换,得到换行符是LF的文件了。
结果是对了,但执行两个脚本来生成一份最终文件,难免让我觉得还不够简单。我分析了以上文件转换的代码,发现关键步骤是使用二进制(wb)方式写入文件。
Method D:原文件使用二进制方式写入
既然这样,不如在一开始就使用二进制写文件,一步到位!并且只需把源代码中写入方式(w)变成二进制写入(wb)方式即可。你看,仅仅添加了一个字母,就省去了十几行代码,还达到了预期目标,可谓一举多得!
#coding:utf-8(设定文件编码格式)
importosimporttime#切换到创建文件目录
os.chdir(r"C:\Study\Chestnuts\01data")#新建创建文件函数
defcreate_file():#构造文件名:以“年月日”为文件名的.dat文件
t=time.localtime()
file_name=time.strftime("%Y-%m-%d",t)+".dat"
#创建并打开文件
fp=open(file_name,'wb+')#写入文件内容
fp.writelines("12345\n")
fp.writelines("67890\n")#关闭文件
fp.close()#调用函数
create_file()
窥一斑而见全豹,以上的代码优化,提醒我编程时需要多思考,理清逻辑,最终找到达成目标的最佳方案。在练习编写代码时,不仅要认真分析需求,分解步骤,也要注意积累好的方法进行沉淀,这也是我写下本文的目的所在。
如果大家在测试工作中,也遇到了类似的困难,觉得自己能力欠缺,期望通过自己编写代码解决问题,欢迎大家咨询我或者吴老。可以加入我们的“光荣之路python群(457561756)”进行提问,也可以来吴老的测试开发培训班一起学习。为了更全面的提升,为了更丰厚的待遇,也为了更美好的生活,让我们在光荣之路一起奋斗!
作者:Flyleaves
出处:http://www.cnblogs.com/Flyleaves/
本文版权归作者、微信公众号光荣之路和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。