From 0e49e636ce7439502f4242da5aa413fecce66621 Mon Sep 17 00:00:00 2001 From: Yaro Kasear Date: Wed, 30 Apr 2025 14:27:39 -0500 Subject: [PATCH] Refactor environment variable loading to improve error handling and clarity --- listener.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/listener.py b/listener.py index ea1e284..718b6c5 100755 --- a/listener.py +++ b/listener.py @@ -6,9 +6,15 @@ from collections import defaultdict from scapy.all import sniff, Dot11, RadioTap from dotenv import dotenv_values -# === Load ENV === -config = dotenv_values(os.path.expanduser("~/wifi_test/settings.env")) +settings_path = os.path.expanduser("~/wifi_test/settings.env") +config = dotenv_values(settings_path) + +if not config: + print(f"[!] Failed to load settings from {settings_path}") + exit(1) + LISTEN_INTERFACE = config.get("LISTEN_INTERFACE", "wlan0") +print(f"[+] Using LISTEN_INTERFACE = {LISTEN_INTERFACE}") # === Globals === clients_per_channel = defaultdict(set)