python小练习一

1、你是一个高级测试工程师，现在要做性能测试，需要你写一个函数，批量生成一些注册使用的账号。

产生的账号是以@163.com结尾，长度由用户输入，产生多少条也由用户输入，用户名不能重复，用户名必须由大写字母、小写字母、数字组成

'''创建测试数据，并且写入到txt文件中'''
import random,string
data = input("需要产生多少条数据？").strip()
len = input("数据长度是多少？").strip()
f =open('test.txt','w',encoding='utf-8')
for i in range(int(data)):
    rdata = ''.join(random.sample(string.ascii_lowercase+string.ascii_uppercase+string.digits,int(len)))
    t = rdata + "@163.com"
    f.write(t+'\n')
    print(t)


'''创建测试数据，并且写入到excel文件中'''
import random,string,xlwt
data = input("需要产生多少条数据？").strip()
len = input("数据长度是多少？").strip()
workbook = xlwt.Workbook()    
sheet = workbook.add_sheet('sheet1')
for i in range(int(data)):
    rdata = ''.join(random.sample(string.ascii_lowercase+string.ascii_uppercase+string.digits,int(len)))
    t=rdata+"@163.com"
    sheet.write(i,0,rdata)
    print(t)
workbook.save('test.xls')