ID2T-toolkit-BotnetTraffic/code/Iteration1/CSVParser.py

import xml.etree.ElementTree as ET
import csv

# def parseCSV:

def parse_csv(file: str):
	# build a tree structure
	root = ET.Element("mmcommunication")

	# parse the csvFile into reader
	with open(file) as csvFile:
	    reader = csv.reader(csvFile, delimiter=",")
		# loop through the parsed file, creating packet-elements with the structure of the csvFile as attributes
	    for line in reader:
	        packet = ET.SubElement(root, "packet")
	        for element in line:
	        	element = element.replace(" ", "")
	        	splitel = element.split(":")
	        	key, value = splitel[0], splitel[1]
		        packet.attrib[key] = value

	# writing the ElementTree into the .xml file
	tree = ET.ElementTree(root)
	tree.write("ExampleCSVBotnetCommunication.xml")


parse_csv("ExampleCSVBotnetCommunication.txt")