“懒人”小工具之txt自动化写入excel脚本

alsly安全：

很多时候，我们都需要将文本里面的内容提取粘贴到我们的表格中，但是这个工作既费时又费力，有没有能够快速地完成这个工作，同时又能够有很高的成功率呢？这个是我在本篇文章中想分享给大家的内容。

本篇文章的内容需要有一定的Python基础，不然会看得迷迷糊糊。没有Python基础的小伙伴可以选择先了解一下，然后去查看一下本人的另一篇文章《「Python“偷懒”入门专栏」”偷懒”入门思路》，先去看看里面内容。当然，后续也会出教程，教大家怎么一步一步地完成这样的一个小工具。

本工具实现txt文件内容填入到excel文件中的自动化脚本，适用与txt文件存在固定的格式，需要将txt格式的内容填入表格的环境。

使用方法：将通报数据整理到txt文件中，执行py脚本。需要根据当下情况进行py脚本微调，目前是将当前日期的txt文件写入excel中，如4月10日对应的txt文件名为0410.txt，也可以根据需要进行脚本调整。

如下场景为，hw防守需要发送事件通报，同时需要将通报事件汇总到excel表格的情况：

通报数据如下：

将通报数据整理到txt文件中，执行py脚本，得到如下excel文件：

通过该方法能够很大程度上减轻防守值班人员的重复性工作量。

最后附上代码：

import xlwt
import time

# 将读取的数据写入到字典
def read_txt(file):
    result = []
    dictt = {}
    f = open(file, "r", encoding="utf-8")
    lines = f.readlines()
    a = 0
    for line in lines: #以下写入字典情况，需要根据具体情况进行调整
        a += 1
        if a % 7 == 2:
            dictt["time"] = line.split("：")[1][:-1]
        elif a % 7 == 3:
            dictt["src_ip"] = line.split("：")[1][:-1]
        elif a % 7 == 4:
            dictt["dst_ip"] = line.split("：")[1][:-1]
        elif a % 7 == 5:
            dictt["attack_type"] = line.split("：")[1][:-1]
        elif a % 7 == 6:
            dictt["chuzhi"] = line.split("：")[1][:-1]
            result.append(dictt)
            dictt = {}  # 因为每次添加的都是同一个内存到list中去了.dictt每次写入的时候改变了内存中的value,但是地址不变.即是,创建了一次内存空间,只会不断的改变value了,添加到list中的时候value已经改了。所以需要在for循环里面去每次循环都创建一个空的dict，以保证之前添加过的不会被改变。
        else:
            continue
    return result

# 将列表写入到excel表
def write_exec(list):
    workbook = xlwt.Workbook(encoding="utf-8", style_compression=0)
    # Cell_overwirte_ok 是能够覆盖单元表格的意思。
    sheet = workbook.add_sheet("2", cell_overwrite_ok=True)
    for i in range(len(list)):  #以下写入表格的具体位置需要根据具体情况进行调整
        if i+1 < 10:
            num = '00' + str(i+1)
        elif i+1 < 100:
            num = '0' + str(i+1)
        else: num = str(i+1)
        sheet.write(i, 0, num)
        sheet.write(i, 1, list[i]['time'])
        sheet.write(i, 2, '/')
        sheet.write(i, 3, list[i]['dst_ip'])
        sheet.write(i, 4, list[i]['src_ip'])
        sheet.write(i, 5, list[i]['attack_type'])
        sheet.write(i, 6, list[i]['chuzhi'])
        sheet.write(i, 7, '是') #默认填入'是'
    workbook.save(r"./01-攻击事件汇总清单_{}.xls".format(time.strftime('%Y%m%d',time.localtime(time.time())))) #导出的excel文件名



filename = time.strftime('%m%d',time.localtime(time.time())) #写入的txt文件名
file = filename + '.txt' 
dictt = read_txt(file)
# print(dictt)
write_exec(dictt)
print('写入完成')

当然，对本工具的细节进行调整，也能够达到意想不到的效果。发挥你们的崇明材质吧。想进一步了解更多的#网络安全#Python#入门知识，可以持续关注本头条号，也可以移步关注同步的微信公众号“alsly安全”。

END