From 1db29f56d2ed8a6470b46f53f6506585051bd2f1 Mon Sep 17 00:00:00 2001
From: Johannes Findeisen <you@hanez.org>
Date: Mon, 5 Oct 2020 01:24:01 +0200
Subject: [PATCH] added error handling when database connection fails

---
 uplink | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/uplink b/uplink
index 4038b70..152fe5d 100755
--- a/uplink
+++ b/uplink
@@ -107,10 +107,15 @@ def get_data(config, inc):
     #conn = sqlite3.connect(config["database"])
     #c = conn.cursor()
 
-    con = pymysql.connect(config["database_host"],
-                          config["database_user"],
-                          config["database_password"],
-                          config["database"])
+    try:
+        con = pymysql.connect(config["database_host"],
+                              config["database_user"],
+                              config["database_password"],
+                              config["database"])
+    except Exception as err:
+        logger.error("ERROR: Database connection failed!" + str(err))
+        #thread.interrupt_main()
+        os._exit(1)
 
     #for i in range(len(config["uplinks"])):
     timestamp = calendar.timegm(time.gmtime())
@@ -132,7 +137,7 @@ def get_data(config, inc):
         with con.cursor() as cur:
             cur.execute(sql)
         con.commit()
-        exit()
+        exit(1)
 
     if fc.is_connected:
         status = "UP"