import codecs
import csv
from openpyxl import load_workbook


def read_csv(file_name):
    data_list = list()
    with codecs.open(file_name, encoding="utf-8") as f:
        for row in csv.DictReader(f, skipinitialspace=True):
            data_list.append(row)

    return data_list


def read_excel(file_name, sheet_name):
    # 打开Excel文件，加载workbook
    workbook = load_workbook(filename=file_name)
    # 根据名称获取worksheet
    worksheet = workbook.get_sheet_by_name(sheet_name)
    # 获取所有数据
    rows = worksheet.rows

    # 将第一行数据作为字典的键
    header = []
    for cell in next(rows):
        header.append(cell.value)

    # 从第二行起遍历每一行数据，并存储到字典中
    data_list = []
    for row in rows:
        data = {}
        for i, cell in enumerate(row):
            data[header[i]] = cell.value
        data_list.append(data)

    return data_list