一、前言
前几天在Python最强王者交流群【哎呦喂 是豆子~】问了一个Python
自动化办公的问题,一起来看看吧。
下图是他的原始数据和他想得到的目标数据,如下所示:
需要在标黄的两行里边进行相关操作。
二、实现过程
上一篇文章我们已经优化过一次了,这一篇文章我们小小的进行优化下,这一篇文章我们一起来看看吧!
import openpyxl
def fill_table(bug_name, sheet, row_number):
# 初始化IP地址列表,已提供误报证明ip
ip_list1 = []
# 初始化IP地址列表,已提供无法整改证明ip
ip_list2 = []
# 初始化IP地址列表,没有误报和无法整改证明的IP
ip_list3 = []
is_provided_misreport_list = []
is_provided_fixed_prove_list = []
# 遍历每一行
for row in range(2, sheet.max_row + 1):
system_name = sheet.cell(row=row, column=1).value
vulnerability_name = sheet.cell(row=row, column=2).value
ip = sheet.cell(row=row, column=3).value
is_provided_misreport = sheet.cell(row=row, column=4).value
is_provided_fixed_prove = sheet.cell(row=row, column=5).value
# 判断漏洞名称和是否提供误报证明,针对OpenSSH 'schnorr.c'远程内存破坏漏洞(CVE-2014-1692)漏洞
if vulnerability_name == bug_name and is_provided_misreport == '是':
is_provided_misreport_list.append(is_provided_misreport)
ip_list1.append(ip)
if vulnerability_name == bug_name and is_provided_fixed_prove == '是':
is_provided_fixed_prove_list.append(is_provided_fixed_prove)
ip_list2.append(ip)
if vulnerability_name == bug_name and is_provided_misreport == '否' and is_provided_fixed_prove == '否':
ip_list3.append(ip)
# 填写C15单元格
if '是' in is_provided_misreport_list:
sheet.cell(row=row_number, column=3).value = '是'
else:
sheet.cell(row=row_number, column=3).value = '否'
# 填写D15单元格
sheet.cell(row=row_number, column=4).value = ','.join(ip_list1)
# 填写E15单元格
if '是' in is_provided_fixed_prove_list:
sheet.cell(row=row_number, column=5).value = '是'
else:
sheet.cell(row=row_number, column=5).value = '否'
# 填写F15单元格
sheet.cell(row=row_number, column=6).value = ','.join(ip_list2)
# 填写G15单元格
if not ip_list3:
sheet.cell(row=row_number, column=7).value = "无"
else:
sheet.cell(row=row_number, column=7).value = ','.join(ip_list3)
if __name__ == '__main__':
# 打开Excel文件
workbook = openpyxl.load_workbook('测试.xlsx')
sheet = workbook.active
bug_name1 = "OpenSSH 'schnorr.c'远程内存破坏漏洞(CVE-2014-1692)"
bug_name2 = "OpenSSH 'x11_open_helper()'函数安全限制绕过漏洞(CVE-2015-5352)"
fill_table(bug_name1, sheet, row_number=15)
fill_table(bug_name2, sheet, row_number=16)
# 保存修改后的Excel文件
workbook.save('updated_excel_file.xlsx')
代码运行之后,可以得到下图的结果。
本文转自网络,如有侵权,请联系删除。
学习资源推荐
除了上述分享,如果你也喜欢编程,想通过学习Python获取更高薪资,这里给大家分享一份Python学习资料。
这里给大家展示一下我进的兼职群和最近接单的截图
😝朋友们如果有需要的话,可以V扫描下方二维码联系领取
学好 Python 不论是就业还是做副业赚钱都不错,但要学会 Python 还是要有一个学习规划。最后大家分享一份全套的 Python 学习资料,给那些想学习 Python 的小伙伴们一点帮助!