From b23e2d93aa2fefd2affc1833e25bb6676061f2f8 Mon Sep 17 00:00:00 2001
From: iRaven4522 <horse@iraven.net>
Date: Sat, 26 Oct 2024 02:49:41 -0500
Subject: [PATCH] Error handling if Pi-hole password is wrong

---
 main.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 3ec5408..f20e577 100644
--- a/main.py
+++ b/main.py
@@ -58,7 +58,11 @@ def loginToPihole(apiurl, phpassword):
     loginrq = session.post(url,data={"pw": phpassword})
     phpsessid = (str(loginrq.cookies.get_dict()["PHPSESSID"]))
     regex = r'(<div id="token" hidden>)(\S+)(<\/div>)'
-    token = re.findall(regex, loginrq.text, re.MULTILINE)[0][1]
+    try:
+        token = re.findall(regex, loginrq.text, re.MULTILINE)[0][1]
+    except IndexError:
+        log.error(f"An authentication token wasn't returned from the Pi-hole server at {apiurl} - your password is likely incorrect!")
+        exit()
     log.debug("phpsessid: " + phpsessid + " Token: "+ token)
     return {'phpsessid': phpsessid, "csrftoken": token}