`bot.lavalink = None` if no connection is made

1 files changed, 15 insertions, 3 deletions

diff --git a/code/cogs/owner/stats.py b/code/cogs/owner/stats.py
index 5409bfb..04da7ca 100644
--- a/code/cogs/owner/stats.py
+++ b/code/cogs/owner/stats.py
@@ -2,6 +2,7 @@ from discord.ext import commands, tasks
 import sqlite3
 import discord
 import os
+import lavalink
 
 from utils.config import BOT_COLOR, LOG
 
@@ -80,13 +81,24 @@ class Stats(commands.Cog):
             description=(
                 f"Total Guilds: `{len(self.bot.guilds):,}`\n"
                 f"Total Commands: `{total_commands:,}`\n\n"
-                f"Total Players: `{len(self.bot.lavalink.get_players())}`\n"
-                "Load:"
-                f" `{round(self.bot.lavalink.nodes[0].stats.lavalink_load * 100, 2)}%`"
             ),
             color=BOT_COLOR,
         )
 
+        # Determine the content of the Lavalink description
+        if self.bot.lavalink:
+            embed.description += (
+                "Total Players:"
+                f" `{len(self.bot.lavalink.get_players())}`\n"
+                "Load:"
+                f" `{round(self.bot.lavalink.nodes[0].stats.lavalink_load * 100, 2)}%`"
+            )
+        else:
+            embed.description += (
+                "Total Players: `NO LAVALINK CONNECTION`\n"
+                "Load: `NO LAVALINK CONNECTION`"
+            )
+
         for entry in data:
             embed.add_field(
                 name=entry[0], value=f"` {entry[1]:,} `", inline=True