最近项目测试,经常需要生成一些固定大小的,内容随意的文件。
于是便萌生了写一个脚本来实现生成固定大小文件。
代码如下:
#*-*coding:utf-8 *-*
'''
Created on 2012-6-14
@author: bitter_tea
'''
def gen_file(path,size):
#首先以路径path新建一个文件,并设置模式为写
file = open(path,'w')
#根据文件大小,偏移文件读取位置
file.seek(1024*1024*1024*size)#姑且以GB为单位吧
#然后在当前位置写入任何内容,必须要写入,不然文件不会那么大哦
file.write('\x00')
file.close()
gen_file('./test.dat',1)
主要原理如下(摘自chinaunix):
根据unix 里面lseek的描述:
lseek仅将当前的文件位移量记录在内核内,它并不引起任何I/O操作。然后,该位移量用于下一个读或写操作。
文件位移量可以大于文件的当前长度,在这种情况下,对该文件的下一次写将延长该文件,并在文件中构成一个空调,这一点是允许的。位于文件中但没有写过的字节都被读为0
注意红色这一行的描述,如果你seek到很长的位置去但是没有写入内容,那么这个文件是不会被延长(增大), 中间没有写数据的地方相当于只是占位,没有IO写,所以会很快。
使用这个方法几秒钟就可以生成上百G的文件,非常好用。