From da15e54a6410a35c3701cae9550cb2cabce9a042 Mon Sep 17 00:00:00 2001
From: Michael Scalzetti <maestromike5@gmail.com>
Date: Wed, 26 Oct 2022 20:40:01 -0400
Subject: [PATCH] fixed rewrite and stuff

---
 src/sniffer.py | 41 ++++++++++-------------------------------
 1 file changed, 10 insertions(+), 31 deletions(-)

diff --git a/src/sniffer.py b/src/sniffer.py
index 95a2f25..1945435 100644
--- a/src/sniffer.py
+++ b/src/sniffer.py
@@ -40,7 +40,7 @@ class Sniffer:
         timeout = int(timeout+0.5)
         dump_file = f"{self.base_dir}/cap-{ int(time.time()) }"
 
-        stdout, stderr, code = run_cmd(f"airodump-ng --write {dump_file} --write-interval 1 --output-format csv walfa0", timeout=timeout)
+        stdout, stderr, code = run_cmd( f"airodump-ng --write {dump_file} --write-interval 1 --output-format csv walfa0", timeout=timeout )
         if code != 0 and code != -1:
             return code
         
@@ -55,28 +55,6 @@ class Sniffer:
                 file_data.append(item)
         return file_data
 
-    def parse_into_dicts_safe(self, csv_data):
-        template = []
-        sniff_data = []
-        
-
-        for line_index in range(len(csv_data)):
-            line = csv_data[line_index]
-            if line_index == 0:
-                continue
-            if line_index == 1:
-                for item in line:
-                    template.append(item)
-                continue
-            temp = {}
-            if line == []:
-                break
-            for item_index in range(len(line)):
-                temp[template[item_index]] = line[item_index]
-            sniff_data.append(temp)
-
-        return sniff_data
-
     def parse_into_dicts(self, csv_data):
         template = []
         sniff_data = []
@@ -89,26 +67,27 @@ class Sniffer:
                 if not sniff_data == []:
                     sniff_data_parts.append(sniff_data)
                     sniff_data = []
+                continue
             if make_template:
                 template = []
                 for item in line:
-                    template.append(item)
+                    template.append(item.strip())
+                make_template = False
                 continue
             temp = {}
             for item_index in range(len(line)):
-                temp[template[item_index]] = line[item_index]
+                try:
+                    temp[template[item_index]] = line[item_index]
+                except:
+                    pass
             sniff_data.append(temp)
 
         return sniff_data_parts
 
 
     def parse_sniff(self, full_filepath):
-        file_data = self.parse_csv()
-        sniff_data = self.pars_into_dicts(file_data)
-
-
-        for packet in sniff_data[1]:
-            print(packet)
+        file_data = self.parse_csv(full_filepath)
+        sniff_data = self.parse_into_dicts(file_data)
         return sniff_data