python_测试数据生成

  

''' 
需要你写一个函数，批量生成一些注册使用的账号。
产生的账号是以@163.com结尾，长度由用户输入，产生多少条也由用户输入，用户名不能重复，用户名必须由大写字母、小写字母、数字组成

'''
import  string
import  random

num=int(input("请输入要产生的数据条数："))

#生成用户名函数
def username(num):
    user = []
    lenght=int(input("请输入用户名长度：")) #输入用户名长度
    while len(user)!=num:
        str1=random.sample((string.ascii_letters+string.digits),lenght) #随机产生用户名
        str2=''.join(str1) #用户名由list拼接成字符串
        if str2 not in user: #字符串去重
            user1=str2+"@163.com"+"\n" #字符串拼接邮箱
            user.append(user1) #字符串放入list
    return user

#用户名写入外部文件
def openfile(filename,user):
    f=open(filename,'w',encoding="utf-8") #打开文件
    f.writelines(user) #把list逐行写入文件
    f.close()


openfile('a.txt',username(num) ) #调取函数