瀏覽代碼

CSV parser commented

Joshua 7 年之前
父節點
當前提交
4668eeacd1
共有 2 個文件被更改,包括 27 次插入0 次删除
  1. 27 0
      code/Iteration1/CSVParser.py
  2. 0 0
      code/Iteration1/test.txt

+ 27 - 0
code/Iteration1/CSVParser.py

@@ -0,0 +1,27 @@
+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")

+ 0 - 0
code/Iteration1/test.txt