Spaenny/bruder-py
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

added logging

Browse source
This commit is contained in:
Philipp 2022-10-01 09:50:28 +02:00
parent 9d30aea5a8
commit aa6a4728ba
Signed by: Philipp
GPG key ID: 9EBD8439AFBAB750
2 changed files with 14 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
requirements.txt
View file

@ -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

15
src/main.py
View file

@ -6,10 +6,10 @@ 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
catchSignals = [
@ -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()