快速生成指定大小的zip文件

# -*- coding:utf-8 -*-# 生成指定大小文件import timeimport osimport zipfileimport shutildef creatfilesize(n, dirname):    local_time = time.strftime("%Y%m%d%H%M%S", time.localtime())    file_name = dirname + str(local_time)+".txt"    bigFile= open(file_name, 'w')    bigFile.seek(1024*1024*n)    bigFile.write('test')    #bigFile.write("test")    bigFile.close()    print('creat %s MB data' % n)    return file_namedef creatzfile(filename, writefile):    zfile = zipfile.ZipFile(filename, 'a', compression=zipfile.ZIP_STORED, allowZip64=True)   #写大文件,开启zip64    zfile.write(writefile)    zfile.close()    size = float(os.path.getsize(filename))    size_MB = size / (1024*1024)    print('zipSize(MB):', size_MB)    return size_MBdef delete_file(file_path):    """    删除文件下的文件    """    listdir = os.listdir(file_path)    for i in listdir:        os.remove(os.path.join(file_path, i))def zip_dir(dirname, zipfilename):    filelist = []    if os.path.isfile(dirname):        filelist.append(dirname)    else:        for root, dirs, files in os.walk(dirname):            for name in files:                filelist.append(os.path.join(root, name))    zf = zipfile.ZipFile(zipfilename, "w", zipfile.ZIP_STORED, allowZip64=True)    for tar in filelist:        arcname = tar[len(dirname):]        print('arcname', arcname)        zf.write(tar, arcname)    zf.close()    size = float(os.path.getsize(zipfilename))    size_MB = size / (1024 * 1024)    print('zipSize(MB):', size_MB)    return size_MBif __name__ == '__main__':    # n = input("输入你要生成的文件大小（单位为M）:")    # creatfilesize(n)    zipfilename = 'out.zip'    dirname = './data/'    n = 23.4    writefile = creatfilesize(n, dirname)    # 生成指定大小的txt文件    num = 0    while True:        zipsize = zip_dir(dirname, zipfilename)        num += 1        if zipsize >= n:            break        shutil.copyfile(writefile, os.path.join(dirname, '_new%s.txt' % num))    # 清理生成的文件    delete_file(dirname)    z = zipfile.ZipFile(zipfilename, "r")    for filename in z.namelist():        print(filename)