From aa6a4728ba6cd598b237c1d58b99ca36a717f852 Mon Sep 17 00:00:00 2001
From: Philipp <philipp@boehm.sh>
Date: Sat, 1 Oct 2022 09:50:28 +0200
Subject: [PATCH] added logging

---
 requirements.txt |  1 +
 src/main.py      | 19 +++++++++++++------
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index afe1a7b..5c4ccee 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,3 +2,4 @@ discord-py-interactions==4.3.1
 steamid==1.0.2
 python-dotenv==0.21.0
 mysql-connector-python==8.0.30
+loguru==0.6.0
diff --git a/src/main.py b/src/main.py
index 20c17cc..514b760 100644
--- a/src/main.py
+++ b/src/main.py
@@ -6,12 +6,12 @@ import database
 
 from dotenv import load_dotenv
 from steamid import SteamID
+from loguru import logger
 
 class GracefulDeath:
     """Catch signals to allow graceful shutdown."""
-
-    def __init__(self):
-        self.receivedSignal=self.receivedSignalTermSignal=False
+    def __init__(self): 
+        self.receivedSignal=self.receivedSignalTermSignal=False 
         catchSignals = [
                 1,
                 2,
@@ -19,7 +19,7 @@ class GracefulDeath:
                 10,
                 12,
                 15,
-                ]
+        ]
         for signum in catchSignals:
             signal.signal(signum, self.handler)
 
@@ -44,8 +44,9 @@ def main(args):
     try:
         if args[0] == "init_database":
             database.init_test_database(mysql)
+            logger.info(f"First init of database... Please disable this afterwards with the flag init_dabase")
     except:
-        print("Skipping database init...")
+        logger.info(f"Skipping database init...")
         pass
 
     @bot.command(
@@ -68,31 +69,37 @@ def main(args):
 
         if channel != "433738529898627073":
             await ctx.send(f"This command only works in the Channel <#433738529898627073>!", ephemeral=True)
+            logger.debug(f"Command was used in the wrong channel. ({channel})")
             return
 
         try:
             steam_id = SteamID(steamid)
         except:
             await ctx.send(f"You failed to give a valid SteamID as it seems, please head to <https://steamid.xyz/> and check.", ephemeral=True)
+            logger.debug(f"User ({user.id}) has give a invalid SteamID. Could not convert!")
             return
 
         if steam_id.isValid() == False:
             await ctx.send(f"You failed to give a valid SteamID as it seems, please head to <https://steamid.xyz/> and check.", ephemeral=True)
+            logger.debug(f"User ({user.id}) has given a invalid SteamID!")
             return
 
         try:
             database.create_reserved_slot(mysql, user, steam_id.steam2(1))
             await ctx.send(f"You just claimed a reserved slot for: `{steam_id.steam2(1)}`!")
+            logger.debug(f"{steam_id.steam2(1)} has just claimed a reserved slot.")
         except Exception as err:
             if str(err) != "Duplicate entrys":
                 await ctx.send(f"Something failed with the database, please try again later.", ephemeral=True)
+                logger.warning(f"Error while trying to access database.", err)
                 return
             await ctx.send(f"You have already claimed a reserved slot!", ephemeral=True)
+            logger.debug(f"{steam_id.steam2(1)} or {user.id} is already in the database.")
 
     bot.start()
 
     while True:
-        print("Graceful shutdown...")
+        logger.info("Trying graceful shutdown...")
         if sighandler.receivedSignal:
             bot.remove("reserve_slot", remove_commands=True)
             exit()