【python接口自动化】- openpyxl读取excel数据
<p><strong><font size="5" color="#DC143C">如果想从头学起接口自动化测试，可以去看看这个系列的文章哦！</font></strong></p><p><font size="3">https://www.cnblogs.com/miki-peng/category/1671261.html</font><br>
<br></p>
<h1 id="openpyxl简单介绍">openpyxl简单介绍</h1>
<p> 前言：目前我们进行测试时用于存储测试数据的软件几乎都是excel，excel方便存储和管理数据，读取数据时也比较清晰，测试时我们需要从excel从读取测试数据，结束后还需把测试结果写入到excel中，读取这一动作以往都是通过人工来操作，学会使用openpyxl后，就可以实现自动化地读取数据，解放双手。</p>
<p> openpyxl是一个开源项目，它是一个用于读取/写入Excel 2010文档（如<code>xlsx</code> 、<code>xlsm</code> 、<code>xltx</code> 、<code>xltm</code>文件 ）的Python库，如果要处理更早格式的Excel文档（<code>xls</code>），需要用到其它库（如：xlrd、xlwt等），这是openpyxl比较其他模块的不足之处。openpyxl是一款比较综合的工具，不仅能够同时读取和修改Excel文档，而且可以对Excel文件内单元格进行详细设置，包括单元格样式等内容，甚至还支持图表插入、打印设置等内容。</p>
<p> python中与excel操作相关的模块：</p>
<ul>
<li>📔 xlrd库：从excel中读取数据，支持xls、xlsx格式</li>
<li>📔 xlwt库：对excel进行修改操作，但不支持对xlsx格式的修改</li>
<li>📔 xlutils库：在xlwt和xlrd中，对一个已存在文件进行修改</li>
<li>📘 <strong>openpyxl库：主要针对xlsx格式的excel进行读取和编辑</strong></li>
</ul>
<h1 id="openpyxl的安装">openpyxl的安装</h1>
<p><code>pip install openpyxl</code></p>
<h1 id="openpyxl的三个基本概念">openpyxl的三个基本概念</h1>
<ul>
<li>WorkBook：工作簿对象</li>
<li>Sheet：表单对象</li>
<li>Cell：表格对象</li>
</ul>
<p> 也就是我们使用excel时可以看到的三个对象，我们操作excel的步骤是打开工作簿，选中表单，对单元格进行操作，而openpyxl的工作原理也是如此。</p>
<img src="https://img2020.cnblogs.com/blog/1804660/202005/1804660-20200517112136855-1460937618.png" alt="image-20200406160120665" style="width: 80%; padding-left: 3em">
<h2 id="openpyxl对excel进行操作">openpyxl对excel进行操作</h2>
<h2 id="新建excel">新建excel</h2>
<p> 运行以下代码后，就会发现项目中多了一个excel文件。</p>
<pre><code class="language-python">import openpyxl
# 创建一个工作簿
workbook = openpyxl.Workbook()
# 创建一个表单
sheet = workbook.create_sheet('表单1')
# 写入一个数据
sheet.cell(row=1, column=1, value="python")
# 保存
workbook.save('test.xlsx')
</code></pre>
<p> 🎈 <strong>注意点</strong>：如果保存的文件名已存在，将会覆盖原来的文件</p>
<h2 id="打开已存在的文件">打开已存在的文件</h2>
<ul>
<li>打开已存在的文件：<code>load_workbook()</code></li>
<li>获取表单：<code>workbook['表单名']</code></li>
<li>读取指定表格：<code>cell(row, column).value</code></li>
</ul>
<pre><code class="language-python">import openpyxl
# 打开工作簿
workbook = openpyxl.load_workbook('test.xlsx')
# 获取表单
sheet = workbook['表单1']
# 读取指定的单元格数据
cell = sheet.cell(row=1, column=1).value
print(cell)
</code></pre>
<p> 运行结果：</p>
<pre><code class="language-shell">C:\software\python\python.exe D:/learn/python24/python_base/test.py
python
Process finished with exit code 0
</code></pre>
<h2 id="读取单个单元格">读取单个单元格</h2>
<pre><code class="language-python"># 方式一：读取A6单元格的值
cell1 = sheet['A6'].value
# 方式二：读取第3行,第4列单元格的值
cell2 = sheet.cell(row=3, column=4).value
</code></pre>
<h2 id="读取多个单元格">读取多个单元格</h2>
<pre><code class="language-python"># 读取A1-B4的单元格，共8个单元格
cell3 = sheet['A1':'B4']
# 读取A1-B4的单元格，共8个单元格
cell4 = sheet['A1:B4']
# 读取第2行的单元格
cell5 = sheet
# 读取第1-2行的单元格
cell5 = sheet
</code></pre>
<p> 读取多个单元格时，都是以元组的形式返回</p>
<h2 id="其他相关操作">其他相关操作</h2>
<table>
<thead>
<tr>
<th>关键代码</th>
<th>相关描述</th>
</tr>
</thead>
<tbody>
<tr>
<td><code>cell(row, column, value) </code></td>
<td>写入数据，三个参数分别是行，列，值</td>
</tr>
<tr>
<td><code>cell(row, column).value</code></td>
<td>获取指定单元格的值，两个参数分别是行，列</td>
</tr>
<tr>
<td><code>workbook.remove('表单名')</code></td>
<td>删除表单</td>
</tr>
<tr>
<td><code>del workbook['表单名']</code></td>
<td>删除表单</td>
</tr>
<tr>
<td><code>workbook.save('文件名')</code></td>
<td>保存工作簿</td>
</tr>
<tr>
<td><code>sheet.max_row</code></td>
<td>获取表单数据的总行数</td>
</tr>
<tr>
<td><code>sheet.max_column</code></td>
<td>获取表单数据的总列数</td>
</tr>
<tr>
<td><code>sheet.rows</code></td>
<td>获取按行所有的数据</td>
</tr>
<tr>
<td><code>sheet.columns</code></td>
<td>获取按列所有的数据</td>
</tr>
</tbody>
</table>
<p> 🎈 <strong>注意点</strong>：</p>
<ol>
<li>openpyxl中列和行的起始标识都是1，不是从0开始</li>
<li>从excel中读取出来的数据只有两种类型，即<strong>数值类型</strong>和<strong>字符串类型</strong></li>
<li>不要随便在表格中敲空格，会影响判断最大行数和最大列数</li>
<li>运行操作excel的代码时，要先关闭在操作系统中打开的相关excel表，否则可能会无法读取/写入数据</li>
<li>写入数据时要执行保存</li>
</ol>
<pre><code class="language-python">import openpyxl
# 打开工作簿
wb = openpyxl.load_workbook('test.xlsx')
# 获取表单
sh = wb['表单1']
# 读取指定的单元格数据
res1 = sh.cell(row=1, column=1).value
print(res1)
# 获取最大行数
print(sh.max_row)
# 获取最大列数
print(sh.max_column)
# 按列读取所有数据，每一列的单元格放入一个元组中
print(sh.columns) # 直接打印，打印结果是一个可迭代对象，我们可以转换成列表来查看
# 按行读取所有数据，每一行的单元格放入一个元组中
rows = sh.rows
# print(list(rows)) # 转换成列表之后打印结果为具体的单元格，如下
# [(<Cell '表单1'.A1>, <Cell '表单1'.B1>, <Cell '表单1'.C1>, <Cell '表单1'.D1>, <Cell '表单1'.E1>, <Cell '表单1'.F1>, <Cell '表单1'.G1>),
#(<Cell '表单1'.A2>, <Cell '表单1'.B2>, <Cell '表单1'.C2>, <Cell '表单1'.D2>, <Cell '表单1'.E2>, <Cell '表单1'.F2>, <Cell '表单1'.G2>),
#(<Cell '表单1'.A3>, <Cell '表单1'.B3>, <Cell '表单1'.C3>, <Cell '表单1'.D3>, <Cell '表单1'.E3>, <Cell '表单1'.F3>, <Cell '表单1'.G3>)]
# 我们可以通过for循环以及value来查看单元格的值
for row in list(rows):# 遍历每行数据
case = [] # 用于存放一行数据
for c in row:# 把每行的每个单元格的值取出来，存放到case里
case.append(c.value)
print(case)
</code></pre>
<p> 运行结果：</p>
<pre><code class="language-shell">C:\software\python\python.exe D:/learn/python24/python_base/test.py
case_id
3
7
<generator object Worksheet._cells_by_col at 0x0000012FEF90D8B8>
['case_id', 'interface', 'title', 'method', 'url', 'data', 'expected']
</code></pre>
<p> excel表中的原数据如下图：</p>
<img src="https://img2020.cnblogs.com/blog/1804660/202005/1804660-20200518222603187-1275077669.png">
<h1 id="openpyxl封装">openpyxl封装</h1>
<p> 在测试过程中，我们会经常需要访问excel读取测试数据，如果毎访问一次我们就调用一次openpyxl，进行打开excel，选取表单，读取数据等操作，将会增加我们的工作量和代码量。凡是需要重复使用的，我们都可以对它进行二次封装，写成我们自己的版本，还可以在封装过程中加入我们想要的内容，使用的时候直接调用即可。</p>
<pre><code class="language-python"># read_excel.py
import openpyxl
class CasesData:
"""用于保存测试用例数据"""
pass
class ReadExcel:
def __init__(self, file_name, sheet_name):
self.file_name = file_name
self.sheet_name = sheet_name
def open(self):
self.wb = openpyxl.load_workbook(self.file_name)
self.sh = self.wb
def close(self):
self.wb.close()
def read_data(self):
"""按行读取数据，最后返回一个存储字典的列表"""
self.open()
rows = list(self.sh.rows)
titles = []
for t in rows:
title = t.value
titles.append(title)
cases = []
for row in rows:
case = []
for r in row:
case.append(r.value)
cases.append(dict(zip(titles, case)))
self.close()
return cases
def write_data(self, row, column, msg):
self.open()
self.sh.cell(row=row, column=column, value=msg)
self.wb.save(self.file_name)
self.close()
if __name__ == '__main__':
# 直接运行本文件时执行，下面是一个应用实例
# 需要读取excel时直接调用ReadExcel类
test2 = ReadExcel('cases.xlsx', 'login')
res2 = test2.read_data() # 最后返回一个存储字典的列表
print(res2['case_id']) # 打印第1个用例的case_id
print(res2['title']) # 打印第4个用例的title
</code></pre>
<p> 上面封装的openpyxl中，应用实例的excel表结构如下：</p>
<img src="https://img2020.cnblogs.com/blog/1804660/202005/1804660-20200518222905574-1033824249.png"><br><br>
来源：https://www.cnblogs.com/miki-peng/p/12904383.html
頁:
[1]