python用于数据构造 - 生成指定长度的测试数据

今天在工作的时候，需要测试数据最大长度，所以临时写了一个简单的python脚本，用于生成指定长度的字符串

import random

while True:
    num = int(input('请输入要生成字符串的长度:'))

    if num<5:
        print('太懒了，5个以内的字符串请自己写吧~')
    else:
        start='A'
        end='Z'
        list1=[]
        list1.append(start)
        for i in range(0,num-2,1):
            a=random.randint(0,9)
            list1.append(str(a))
        list1.append(end)
        str_final="".join(list1)
        print(str_final)
        print("------------")

脚本说明：
1、长度在5个以上的，可以生成，长度在5位以内的，不给生成
2、每个生成的字符串，都是以A开头，以Z结束，这样方便用户查看提交的参数、或保存到数据库的数据是否完整。
3、脚本有一个while True循环，也就是，会持续不断的执行，方便使用

脚本虽然很简单，却挺实用的，我下午准备数据时效率变得比较高，在这里备份一下，后续再用的时候，直接找出来~