将所有pcap文件转换为包含所需列的csv python

import dpkt
import csv

def pcap_to_csv(pcap_file, csv_file):
    # 打开pcap文件
    with open(pcap_file, 'rb') as f:
        pcap = dpkt.pcap.Reader(f)
        
        # 创建csv文件并写入表头
        with open(csv_file, 'w', newline='') as csvfile:
            writer = csv.writer(csvfile)
            writer.writerow(['源IP', '目标IP', '源端口', '目标端口', '协议'])
            
            # 遍历pcap文件中的每个数据包
            for timestamp, buf in pcap:
                # 解析数据包
                eth = dpkt.ethernet.Ethernet(buf)
                ip = eth.data
                
                # 提取所需的列数据
                src_ip = '.'.join(map(str, ip.src))
                dst_ip = '.'.join(map(str, ip.dst))
                src_port = ''
                dst_port = ''
                protocol = ''
                
                if isinstance(ip.data, dpkt.tcp.TCP):
                    src_port = ip.data.sport
                    dst_port = ip.data.dport
                    protocol = 'TCP'
                elif isinstance(ip.data, dpkt.udp.UDP):
                    src_port = ip.data.sport
                    dst_port = ip.data.dport
                    protocol = 'UDP'
                
                # 将数据写入csv文件
                writer.writerow([src_ip, dst_ip, src_port, dst_port, protocol])

# 调用函数将pcap文件转换为csv文件
pcap_to_csv('input.pcap', 'output.csv')