py-》生成自定义文件大小

'''

生成自定义文件大小

'''

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):

   

    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.jpeg'

    print(f'文件名：{filename}')

    # 设置文件保存的路径

    filepath = 'C:\\Users\\' + getusername() + '\\Desktop\\'

    f = open(filepath + filename, 'w')

    # 获取开始时间

    starttime = getnowdatatime()

    # startclock = time.clock()

    startclock = time.perf_counter()

    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.perf_counter() - startclock):<.3}sec.')

if __name__ == '__main__':

    generateTXTFile()

 

——————————

 

效果：