基于python实现生成指定大小txt文档
作者:wen-cheng
这篇文章主要介绍了基于python实现生成指定大小txt文档,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
前言
在测试过程中经常遇到文件上传的功能,文件的大小边界值测试一直没有好的解决办法,这里我分享一个创建文件的脚本希望对大家有帮助。
demo
""" * Create by dell on 2020/7/10 * Author :wencheng * 微信公众 :自动化测试 To share """ # !/user/bin/env python # -*- coding:utf-8 -*- import os import time # 获取username, 如chinaren def getusername(): namelist = os.popen('echo %username%').readlines() username = namelist[0].replace("\n", "") # 获取当前的username return username # 获取时间和日期 def getnowdatatime(flag=0): ''' flag = 0为时间和日期 eg:2018-04-11 10:04:55 flag = 1仅获取日期 eg:2018-04-11 flag = 2仅获取时间 eg:10:04:55 flag = 3纯数字的日期和时间 eg:20180411100455 ''' now = time.localtime(time.time()) if flag == 0: return time.strftime('%Y-%m-%d %H:%M:%S', now) if flag == 1: return time.strftime('%Y-%m-%d', now) if flag == 2: return time.strftime('%H:%M:%S', now) if flag == 3: return time.strftime('%Y%m%d%H%M%S', now) # 生成指定大小的TXT档 def generateTXTFile(): fileSize = 0 # 判断输入是否有误 while True: size = input('请输入你想生成的TXT文件大小(MB):') if size.strip().isdigit() != True: print('只能输入整数,请重新输入!') continue else: fileSize = int(size) break if fileSize >= 200: print('正在生成TXT文件,请稍候... ...') # 生成指定大小的TXT档 filename = getnowdatatime(3) + '_' + size + 'MB.txt' print(f'文件名:{filename}') # 设置文件保存的路径 filepath = 'C:\\Users\\' + getusername() + '\\Desktop\\' f = open(filepath + filename, 'w') # 获取开始时间 starttime = getnowdatatime() startclock = time.clock() for i in range(fileSize): if i >= 100: if i % 100 == 0: print(f'已生成{i//100 * 100}MB数据.') for j in range(1024): try: f.write('01' * 512) except KeyboardInterrupt: print('\n异常中断:KeyboardInterrupt') f.close() exit(-1) f.close() print(f'文件已成生并保存在桌面, 文件大小:{fileSize}MB.\n') print(f'DetailInfo:') print(f'保存路径: {filepath + filename}') print(f'开始时间:{starttime}') print(f'结束时间:{getnowdatatime()}') print(f'总共耗时:{(time.clock() - startclock):<.3}sec.') if __name__ == '__main__': generateTXTFile()
创建1M文件
创建100M文件
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。