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。