From 6d6755907aaebc2c1793e59ee69413196333cf4d Mon Sep 17 00:00:00 2001
From: Peter Nardi <pete@nardi.com>
Date: Tue, 12 Mar 2024 13:10:46 -0400
Subject: [PATCH] Tuning.

---
 src/banip/country_analyzer.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/banip/country_analyzer.py b/src/banip/country_analyzer.py
index 3dc1bd5..2be6769 100755
--- a/src/banip/country_analyzer.py
+++ b/src/banip/country_analyzer.py
@@ -21,7 +21,12 @@
 from banip.contants import RENDERED_BLACKLIST
 from banip.utilities import clear
 
-target_country = sys.argv[1].upper()
+try:
+    target_country = sys.argv[1].upper()
+except IndexError:
+    print("Please provide a two-letter country code.")
+    sys.exit(1)
+
 regex = r"(?<=[\s=])(\w+)=([^=\s]+)(?=\s|$)"
 LOGS = HOME / "data/logs.txt"
 clear()