用Python生成指定大小的文本文件
平时经常会遇到各种大小的文件，当要使用指定大小的文件来作测试时，又很难的找到。很多时候都会有这样的情况，当你不需要它的时候，它总在你身边；但当你要找它的时候，它偏偏不在。唉，又扯远了。
现在，停止去各个文件夹中查看那些文件的大小，自己用py来解决这个问题，因为工作中要用各种大小的文件来作ftp测试，比如10K,1M,10M,50M,100M,...还是自己动手，丰衣足食。
#filename: mkfile.py # coding=utf-8 #此程序用于生成指定大小的文本文件，可以修改文本中每一行的字符数 #文本内容为随机生成 import os import random def getLine(diction,L,n): '随机生成一行' str = [] for i in range(n): str.append( diction[ random.randint(0,L-1) ] ) str.append( '\n' ) return ''.join( str ) def getDiction(): '用于生成随机字母的字典' result = [] for i in range( ord('A'), ord('z')+1 ): result.append( chr(i) ) result.append(' ') #增加多一些空格 return ''.join(result) def main(): fileSize = 1024 * 1024 #这里指定生成文件的大小 lineLen = 50 #这里指定每一行的字符个数 N = fileSize / (lineLen + 1) #计算出行数 left = fileSize % (lineLen + 1)
di = getDiction() L = len(di) f = open('result.txt','w') #这里修改生成文件的文件名 for i in range(N): f.write( getLine(di,L,lineLen) ) if left > 0 : f.write( getLine(di,L, left) ) f.close() if __name__ == '__main__': main() |
最新评论