用python生成限定大小的输入数据

在做一些健壮性测试的时候，需要知道上传文件的大小的边界值，以及输入数据的多少，因此写了一个小的python脚本生成固定大小的数据

#!/usr/bin/python2
#encoding=utf8

#  Copyright Nothing
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

import sys
import random
import os.path

KILOBYTES = 1024

if len(sys.argv) != 3:
    print 'invalie argus---usage eg: %s /temp/outputname 1024'%sys.argv[0]
    sys.exit()
    
filename,size=sys.argv[1],sys.argv[2]

try:
    with open(filename, 'w') as file:
        for i in range(KILOBYTES):
            file.write(chr(random.randint(0,95)+32)*int(size)+'\n')
except ValueError as v:
    print 'the argu %s not the invalid size'