Python自动化- 20210530 - 自动2 封装excel-openpyxl

最新推荐文章于 2024-02-28 18:26:15 发布

queen_zoe

最新推荐文章于 2024-02-28 18:26:15 发布

阅读量151

点赞数

分类专栏： python 自动化文章标签： python excel

本文链接：https://blog.youkuaiyun.com/queen_zoe/article/details/117404723

版权

python 自动化专栏收录该内容

6 篇文章

订阅专栏

本文档展示了如何使用openpyxl库读取Excel文件，并提供了测试用例来验证数据读取的正确性。`ExcelHandler`类实现了读取Excel工作簿的方法，将数据转换为字典列表。在测试类`TestLogin`中，针对登录数据进行了测试，确保用户名和密码的组合与预期结果一致。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# 读取excel的方法

import openpyxl
from openpyxl.worksheet.worksheet import Worksheet


class ExcelHandler:
    def __init__(self, path):
        self.path = path

    def read_excel(self,sheet_name):
        workbook=openpyxl.load_workbook(self.path)
        sheet : Worksheet =workbook[sheet_name] # 这里: Worksheet的写法是声明这个sheet是worksheet的对象，可以方便.出方法
        value=list(sheet.values)
        values = value[1:]
        return [dict(zip(value[0],row))for row in values]
        # 当直接 return sheet.values 时 是获取的一个generator的对象，这个是一个生成器，生成器可以方便转换成,因此写成 ：list(sheet.values)

# 测试方法：

import json
import unittest

from study import read_excel
from study.login import login


class TestLogin(unittest.TestCase):
    def test_data(self):
        values=read_excel.ExcelHandler("py40_case_login.xlsx").read_excel("login")
        # 不能将password全部转成str，因为当为空时会显示 'None'
        for i in values:
            if i['data_group'] =='success_data':
                password=str(i["password"])
            else:
                password = i["password"]
                # 要将返回值转换成string，因为返回值是对象，对象和字符串不相等
            res = json.dumps(login(i['username'], password), ensure_ascii=False)
            self.assertEqual(i["expect"], res)