Python生成指定大小的文件两种解决方案
作者：测试蔡坨坨
这篇文章主要介绍了Python生成指定大小的文件,这里提供两种解决方案帮助python完成我们生成任意大小的文件，需要的朋友可以参考下
在日常测试工作中，我们经常需要对上传的文件大小进行测试，例如：一个文件上传功能，限制文件大小最大为10MB，此时我们可能需要测试10MB以及其边界值9MB和11MB；再或者我们有时需要测试一个超大文件，进行大文件的测试。
针对以上情况，可能一时难以找到符合准确数据的测试文件，这时就可以使用Python来帮助我们生成任意大小的文件，这里提供两种解决方案。
方法1：
使用特定大小的文本重复生成，指定一个文本字符串text，然后将其重复复制直至达到所需的文件大小。
# author: 测试蔡坨坨
# datetime: 2023/6/8 1:31
# function: 使用特定大小的文本生成指定大小的文件
def generate_file(file_path, file_size_bytes):
text = "This is some sample text by caituotuo." # 要重复的文本
text_size_bytes = len(text.encode('utf-8')) # 每个重复的文本的大小（以字节为单位）
repetitions = file_size_bytes // text_size_bytes # 需要重复的次数
remainder = file_size_bytes % text_size_bytes # 剩余的字节数
with open(file_path, 'w') as file:
for _ in range(repetitions):
file.write(text)
if remainder > 0:
file.write(text[:remainder])
if __name__ == '__main__':
# 生成一个大小为10MB的PDF文件
generate_file('caituotuo.pdf', 1024 * 1024 * 10)
方法2：
使用特定大小的随机数生成，使用随机数生成器生成特定大小的字节，并将其写入文件中。
# author: 测试蔡坨坨
# datetime: 2023/6/8 2:31
# function: 使用特定大小的随机数生成文件
import os
def generate_file(file_path, file_size_bytes):
with open(file_path, 'wb') as file:
file.write(os.urandom(file_size_bytes))
if __name__ == '__main__':
# 生成一个大小为1MB的随机数据文件
generate_file('caituotuo.docx', 1024 * 1024)
到此这篇关于Python生成指定大小的文件的文章就介绍到这了,更多相关Python生成指定大小的文件内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！