fixing buggy gpt code :D

CyberRoute · Sep 23, 2024 · 9ecebff · 9ecebff
1 parent 7b6b2b9
commit 9ecebff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/core/arp_scanner.py b/core/arp_scanner.py
@@ -275,7 +275,7 @@ class ARPScannerThread(QThread):
     progress_updated = pyqtSignal(int)
     verbose_output = pyqtSignal(str)  # Signal to emit verbose output
 
-    def __init__(self, interface, mac_vendor_lookup, timeout=5):
+    def __init__(self, interface, mac_vendor_lookup, timeout=1):
         super().__init__()
         self.interface = interface
         self.mac_vendor_lookup = mac_vendor_lookup

diff --git a/core/vendor.py b/core/vendor.py
@@ -75,7 +75,7 @@ def _load_from_url(cls, url):
             dict: A dictionary mapping OUI (first 6 characters of MAC address) to vendor names.
             If the request fails, returns an empty dictionary.
         """
-        response = requests.get(url, timeout=5)
+        response = requests.get(url, timeout=1)
         if response.status_code == 200:
             csv_data = StringIO(response.text)
             csvreader = csv.reader(csv_data)
@@ -102,7 +102,6 @@ def _save_to_file(cls, filename, data):
         """
         with open(filename, 'w', encoding='utf-8') as file:
             writer = csv.writer(file)
-            writer.writerow(["Organizationally Unique Identifier", "Organization Name"])
             for oui, vendor in data.items():
                 writer.writerow([oui, vendor])