1. 程式人生 > 其它 >用python生成你想要的任意大小檔案

用python生成你想要的任意大小檔案

在測試的日常工作中,我們經常會需要測試上傳檔案的邊界值。今天分享一段30行的簡單程式碼,可以生成任意大小的檔案,方便測試。

file_size=input("請輸入想要生成檔案的大小:(單位MB)")
class MakeDir:
    def __init__(self):
        self.file_path="D:/dir_ytj/"
        self.file_name="檔案大小"+file_size+"MB"
        new_file_size = file_size.strip()
        file_size_list = new_file_size.split("
.") self.file_size_list = file_size_list def fileSize_making(self): #輸入的檔案大小去除首尾空格後切割,如果列表中只有一個數字說明是整數,否則就是小數 if len(self.file_size_list) == 1: self.int_size_mb() print("檔案大小{}MB,已存入地址{}".format(file_size,self.file_path)) else: self.int_size_mb() self.float_size_mb()
print("檔案大小{}MB,已存入地址{}".format(file_size,self.file_path)) def int_size_mb(self): #整數部分用寫入檔案w方式 with open(self.file_path+self.file_name,"w") as file: #b-kb-mb檔案大小轉化 for i in range(int(self.file_size_list[0])): for j in range(1024): file.write(
"01"*512) def float_size_mb(self): #小數部分用追加寫入a方法 with open(self.file_path+self.file_name,"a") as file: #獲取小數(單位mb) float_size_mb=float(file_size)-int(self.file_size_list[0]) for i in range(1024): file.write("1"*int(1024*float_size_mb)) #呼叫生成檔案 MakeDir().fileSize_making()