目录
一 背景
压测需要准备测试数据,即1000以上个文件,其中只有一个值不一样,因此想通过脚本你批量生成文件
二 思路
1、先创建一个demo文件
2、将demo文件复制到新文件
3、修改新文件
三 代码示例(demo)
import os
import uuid
import resave
data_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'test_data') # 数据存放目录
file_path = os.path.join(data_path, 'data_demo') # demo文件所在目录
for i in range(5):
file_path1 = os.path.join(data_path, f'test_data{i}') # 新文件所在目录
resave.File_SaveAs(file_path, file_path1) # 复制demo文件,并另存为新文件
# 修改f2文件内容(f1文件保持不变)
with open(file_path1, 'w') as fs:
guid = str(uuid.uuid1()) # 要修改的内容(根据实际情况更改)
fs.write(guid)
resave.py
引载:Python实现 文件另存为功能_qq_278339675的博客-CSDN博客_python另存为文件
# 复制f1文件,存为f2文件
def File_SaveAs(f1, f2):
with open(f1, 'rb') as fp1:
b1 = fp1.read()
with open(f2, 'wb') as fp2:
fp2.write(b1)