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()