python3+Appium自动化11-data数据封装之python读取csv文件

使用背景

实际项目中，我们的测试数据可能存储在一个数据文件中，如txt、excel、csv文件类型。我们可以封装一些方法来读取文件中的数据来实现数据驱动

enumerate()简介

enumerate()是python的内置函数

• enumerate在字典上是枚举、列举的意思
• 对于一个可迭代的（iterable）/可遍历的对象（如列表、字符串），enumerate将其组成一个索引序列，利用它可以同时获得索引和值
• enumerate多用于在for循环中得到计数

实例

对一个列表，既要遍历索引又要遍历元素

     list = ["这", "是", "一个", "测试", "数据"]
     for i in range(len(list)):
         print(i,list[i])
     print("==========")
     for index,item in enumerate(list):
         print(index,item)

输出结果

数据读取方法封装

 #coding=utf-8
 import csv

 def get_csv_data(csv_file, line):
     '''
     获取csv文件指定行的数据
     :param csv_file: csv文件路径
     :param line: 数据行数
     :return:
     '''
     with open(csv_file, 'r', encoding='utf-8-sig') as file:
         reader = csv.reader(file)
         for index, row in enumerate(reader, 1):
             if index == line:
                 return row
 if __name__=='__main__':
     csv_file = '../data/account.csv'
     data = get_csv_data(csv_file, 2)
     print(data)

注：创建csv文件可以使用excel进行创建，保存时另存为.csv文件即可，如果直接将后缀改为.csv有可能读取不到

utf-8与utf-8-sig两种编码格式的区别

UTF-8以字节为编码单元，它的字节顺序在所有系统中都是一样的，没有字节序的问题，也因此它实际上并不需要BOM(“ByteOrder Mark”)。但是UTF-8 with BOM即utf-8-sig需要提供BOM。