always restart bot in case of crash

bot.py

@@ -21,14 +21,14 @@ class Bot(nerimity.Client):
 
         # initialize last.fm client
         self.lastfm = pylast.LastFMNetwork(
-            api_key = lastfm_api_key,
+            api_key    = lastfm_api_key,
             api_secret = lastfm_api_secret
         )
 
         # initialize collage generator
         self.collage_generator = CollageGenerator(
-            lastfm_api_key=lastfm_api_key, 
-            lastfm_api_secret=lastfm_api_secret
+            lastfm_api_key    = lastfm_api_key, 
+            lastfm_api_secret = lastfm_api_secret
         )
 
         # initialize catbox client

commands/collage.py

@@ -46,11 +46,12 @@ def setup(bot: bot.Bot):
                     return
 
         match timeframe:
-            case "a" | "all" | "alltime" | "o" | "overall": timeframe = "overall"
-            case "y" | "year" | "yearly": timeframe = "12month"
-            case "m" | "month" | "monthly": timeframe = "1month"
-            case "w" | "week" | "weekly": timeframe = "7day"
-            case "d" | "day" | "daily": timeframe = "1day"
+            case "a" | "all" | "alltime" | "o": timeframe = "overall"
+            case "y" | "year" | "yearly":       timeframe = "12month"
+            case "m" | "month" | "monthly":     timeframe = "1month"
+            case "w" | "week" | "weekly":       timeframe = "7day"
+            case "d" | "day" | "daily":         timeframe = "1day"
+
             case _: "7day"
 
         if timeframe == "1day":
@@ -61,11 +62,11 @@ def setup(bot: bot.Bot):
             temp_msg = await ctx.send(f"Generating album chart for **{username}**...")
 
             image = bot.collage_generator.generate(
-                entity = "album", 
+                entity   = "album", 
                 username = username,
-                rows = int(size.split(sep="x")[0]),
-                cols = int(size.split(sep="x")[1]),
-                period = timeframe
+                rows     = int(size.split(sep="x")[0]),
+                cols     = int(size.split(sep="x")[1]),
+                period   = timeframe
             )
 
             img_bytes = io.BytesIO()
@@ -86,9 +87,10 @@ def setup(bot: bot.Bot):
             link = await bot.zxz.upload_from_bytes(img_bytes.getvalue(), "chart.png")
 
             await ctx.send(u.good_msg(f"**{size}** {timeframe} chart for [{username}](https://last.fm/user/{username}) successfully generated:\n{link}"))
-            temp_msg.delete()
 
         except Exception as e:
-            if temp_msg: temp_msg.delete()
             print(e)
             await ctx.send(u.error_msg(f"I crashed and burned while generating the chart:\n`{e}`"))
+
+        finally: 
+            if temp_msg: temp_msg.delete()

commands/config.py

@@ -8,8 +8,7 @@ def setup(bot: bot.Bot):
     @bot.slash_command(name="setfm", description="Sets your Last.fm username.")
     async def setfm(ctx: nerimity.Context, username: str = None):
         if not username:
-            await ctx.send(u.error_msg("Please provide your Last.fm username.\n\
-                                        usage: `/fm <your last.fm username>`"))
+            await ctx.send(u.error_msg("Please provide your Last.fm username.\nusage: `/fm <your last.fm username>`"))
             return
 
         try:

commands/now_playing.py

@@ -11,13 +11,11 @@ def setup(bot: bot.Bot):
         if lookup: username = lookup
         else:
             try: username = await bot.get_lastfm(ctx.author.id)
-            except Exception as e:
-                print(e)
+            except Exception as e: 
                 await ctx.send(u.error_msg(f"Unknown database error:\n{e}"))
 
         if not username:
             await ctx.send(u.error_msg("Please provide a Last.fm username (or set yours with `/setfm`)"))
-            print("returned")
             return
 
         try: 
@@ -35,10 +33,9 @@ def setup(bot: bot.Bot):
             track_artist = track.get_artist().get_name()
             track_artist_url = track.get_artist().get_url()
 
-            if now_playing == True:
-                await ctx.send(f"Now playing for **{username}**: [{track_name}]({track_url}) by [{track_artist}]({track_artist_url})")
-            else:
-                await ctx.send(f"Last played for **{username}**: [{track_name}]({track_url}) by [{track_artist}]({track_artist_url})")
+            np = "Now playing for" if now_playing else "Last played by"
+
+            await ctx.send(f"{np} **{username}**: [{track_name}]({track_url}) by [{track_artist}]({track_artist_url})")
 
         except IndexError:
             await ctx.send(u.error_msg(f"User **{username}** has no recent tracks or they're set to private."))

commands/ping.py

@@ -1,6 +1,8 @@
 import nerimity
 
-def setup(bot):
+import bot
+
+def setup(bot: bot.Bot):
     @bot.command(name="ping")
     @bot.slash_command(name="ping", description="Ping...")
     async def ping(ctx: nerimity.Context):

main.py

@@ -5,12 +5,11 @@ import bot
 with open("config.toml", "rb") as f:
     config = tomllib.load(f)
 
-# nerimity client
 bot = bot.Bot(
-    prefix = '!',
-    token = config['token'],
-    owner_id = config['owner_id'],
-    lastfm_api_key = config['api_key'],
+    prefix            = '!',
+    token             = config['token'],
+    owner_id          = config['owner_id'],
+    lastfm_api_key    = config['api_key'],
     lastfm_api_secret = config['api_secret']
 )
 
@@ -21,4 +20,4 @@ async def on_ready(params):
     print(f"Logged in as {bot.account.username}")
     await bot.init_db()
 
-bot.run()
+bot.run(restart_always = True)