python 기반 지정 한 크기 txt 문서 생 성

"""
 * Create by dell on 2020/7/10
 * Author ：wencheng
 *      ：      To share
"""
# !/user/bin/env python
# -*- coding:utf-8 -*-

import os
import time


#   username,  chinaren
def getusername():
  namelist = os.popen('echo %username%').readlines()
  username = namelist[0].replace("
", "")
  #      username
  return username


#        
def getnowdatatime(flag=0):
  '''
  flag = 0           eg:2018-04-11 10:04:55
  flag = 1           eg:2018-04-11
  flag = 2           eg:10:04:55
  flag = 3           eg:20180411100455
  '''
  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.txt'
  print(f'   ：{filename}')
  #          
  filepath = 'C:\\Users\\' + getusername() + '\\Desktop\\'
  f = open(filepath + filename, 'w')
  #       
  starttime = getnowdatatime()
  startclock = time.clock()
  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('
    :KeyboardInterrupt')
        f.close()
        exit(-1)
  f.close()
  print(f'           ,     :{fileSize}MB.
')
  print(f'DetailInfo:')
  print(f'    : {filepath + filename}')
  print(f'    :{starttime}')
  print(f'    :{getnowdatatime()}')
  print(f'    :{(time.clock() - startclock):<.3}sec.')


if __name__ == '__main__':
  generateTXTFile()