python—测试工具开发流程（3）—案例生成指定长度的用户名：v5版本，数据保存功能

v5版本：数据写入本地文件：

改动点：
- 在main的循环中，把生成的数据，写入csv中
脚本设计流程图：

代码：

import csv
import random
import string



def get_randomstring(min_len, max_len,pre,num_name):
    # 在指定长度内生成一个随机数
    num = random.randint(min_len, max_len)
    # 设定初始为空
    str_data = ''
    # 循环累加2此,
    for i in range(1, 3):
        str_data = str_data + string.ascii_letters
    # 按随机数取获取随机字符串
    # ['i', 'k', 'p', 'y', 'E', 's', 'Y', 'o'],需要使用join连接成字符串
    str_name = ''.join(random.sample(str_data,num))
    # 把前缀和随机字符进行拼接
    pre_str_name = pre +num_name+ str_name
    # 输出对应用户名name
    return pre_str_name

if __name__ == '__main__':

    #打开一个文件(newline='', 在新生成的行里直接写，不设置的话每行数据间就会有空行)
    file = open('username.csv', 'w', newline='')
    # 需要先创建csv编辑器
    cw = csv.writer(file)

    min_num = int(input('请输入最小值：'))
    max_num = int(input('请输入最大值：'))
    pre_name = input('请输入前缀：')
    count = int(input('请输入需要的个数：'))

    for i in range(1,count+1):
        str_name = get_randomstring(min_num,max_num,pre_name,str(i))
        print(str_name)

        # 写入数据(需要把写入的数据放在中括号中，否则数据会被拆开)
        cw.writerow([i,str_name])
        # 关闭文件

    file.close()

数据写入效果：
- 截图：

技术试验：

csv文件写入：

# 导入csv库
import csv

# 打开一个文件(newline='',在新生成的行里直接写，不设置的话每行数据间就会有空行)
file = open('username.csv','w',newline='')
# 需要先创建csv编辑器
cw = csv.writer(file)
# 写入数据(需要把写入的数据放在中括号中，否则数据会被拆开)
cw.writerow(['ssssssdd'])
# 在一行里写入多列数据，用逗号隔开
cw.writerow(['ssssssdd','66666','dddddd'])
# 关闭文件
file.close()