Whole bot rewrite

.gitignore

@@ -1,2 +1,3 @@
-dbconfig.json
-__pycache__/
+__pycache__/
+/utils/dbconstants.py
+utils/dbconstants.py

bot/bot.py

@@ -0,0 +1,32 @@
+import nextcord
+import os
+from nextcord.ext import commands
+from utils.utils import *
+from utils.constants import TOKEN
+
+class Bot(commands.AutoShardedBot):
+    def __init__(self):
+        super().__init__(shard_count=1, command_prefix="nom!")
+        self.unloaded_cogs = []
+
+    def initialize(self):
+        self.load_extensions()
+        self.run(TOKEN)
+
+    def load_extensions(self):
+        for cog in os.listdir("./cogs"):
+            if cog.endswith(".py"):
+                try:
+                    self.load_extension(name=f"cogs.{cog[:-3]}")
+                    print(f"Loaded {cog[:-3]} cog")
+
+                except Exception as e:
+                    print(e)
+                    print(f"Failed to load {cog[:-3]} cog")
+                    self.unloaded_cogs.append(cog.capitalize()[-3])
+
+    async def on_ready(self):
+        print(f"Logged in as {self.user}!")
+        print(self.shards)
+        for shard in self.shards:
+            await self.change_presence(status=nextcord.Status.online, activity=nextcord.Activity(type=nextcord.ActivityType.playing, name=f"Verifying | Shard: {shard+1}/{len(self.shards)}"), shard_id=shard)

cogs/dashboard.py

@@ -0,0 +1,35 @@
+import nextcord
+from nextcord.ext import commands
+from nextcord import Interaction
+from utils import check_premium, DBENDPOINT, DBNAME, DBPASS, DBUSER, generate_dashboard, create_warning_embed
+from views import DashboardButtons
+import pymysql
+
+class Dashboard(commands.Cog):
+    def __init__(self, client: commands.Bot):
+        self.client = client
+
+    @nextcord.slash_command(name="dashboard", description="Configure the bot on the in-discord dashboard")
+    async def dashboard(self, interaction: Interaction):
+        await interaction.response.defer(with_message=True, ephemeral=True)
+        if not interaction.user.guild_permissions.administrator:
+            await interaction.send(embed=create_warning_embed(title="Insufficient permissions", description="You need the `administrator` permission to use this."))
+            return
+
+        conn = pymysql.connect(host=DBENDPOINT, port=3306, user=DBUSER, password=DBPASS, db=DBNAME)
+        cur = conn.cursor()
+        cur.execute(f"SELECT * FROM guild_configs WHERE id='{interaction.guild.id}'")
+        data = cur.fetchall()
+        if not data:
+            cur.execute(f"INSERT INTO guild_configs (id) VALUES ('{interaction.guild.id}')")
+            conn.commit()
+        msg = await interaction.send("Generating Dashboard...", ephemeral=True)
+        embed = generate_dashboard(self, data=data)
+        view = DashboardButtons(msg, premium=check_premium(self, guild = True, user = False, type_id=interaction.guild.id))
+        await msg.edit(content="", embed=embed, view=view)
+
+
+
+
+def setup(client: commands.Bot):
+    client.add_cog(Dashboard(client))

cogs/error_handler.py

@@ -0,0 +1,24 @@
+import nextcord
+from cooldowns import CallableOnCooldown
+from nextcord.ext import commands
+from bot.bot import Bot
+from nextcord.errors import Forbidden
+from utils import *
+import time
+
+class Error_handler(commands.Cog):
+    def init(self, client: Bot):
+        self.client = client
+
+    @commands.Cog.listener()
+    async def on_application_command_error(self, interaction:nextcord.Interaction, error):
+        error = getattr(error, "original", error)
+        if isinstance(error, CallableOnCooldown):
+            await interaction.send(f"That command is on cooldown. You can use it again <t:{int(error.retry_after) + round(int(time.time()))}:R>")
+        elif isinstance(error, Forbidden):
+            await interaction.send("I don't have permission to do that")
+        else:
+            raise error
+
+def setup(client: Bot):
+    client.add_cog(Error_handler(client))

cogs/help.py

@@ -1,44 +1,29 @@
 import nextcord
-from nextcord.ext import commands, tasks
-from nextcord import Interaction, SlashOption
-import json
-
-
-class HelpButtons(nextcord.ui.View):
-    def __init__(self):
-        super().__init__()
-        self.add_item(nextcord.ui.Button(label="Support Server", url="https://discord.gg/aGzvXvTkP8"))
-        self.add_item(nextcord.ui.Button(label="Invite", url="https://discord.com/api/oauth2/authorize?client_id=981835181243658260&permissions=8&scope=bot%20applications.commands"))
-        self.add_item(nextcord.ui.Button(label="Vote", url="https://top.gg/bot/828584622156939274/vote"))
-
+from nextcord.ext import commands
+from nextcord import Interaction
+from utils.constants import COLOUR_MAIN, VOTELINK, INVITELINK, DISCORDLINK, PRIVACYLINK
+from views import BotInfoLinkButton
 
 class Help(commands.Cog):
-
-    def __init__(self, client):
+    def __init__(self, client: commands.Bot):
         self.client = client
-    
+
     @nextcord.slash_command(name="help", description="Help command")
-    async def help(self,
-        ctx: Interaction):
-        await ctx.response.defer()
-        embed = nextcord.Embed(title=(f"Help"), description=(f"""Below is a list of all commands you will need:"""), colour=0xadd8e6)
-        embed.add_field(name=f"/config enable", value=f"""Explanation: Enable any settings you want to
+    async def help(self, interaction: Interaction):
+        await interaction.response.defer()
+        embed = nextcord.Embed(title=("Help"), description=("""Below is a list of all commands you will need:"""), colour=COLOUR_MAIN)
+        embed.add_field(name=f"/dashboard", value="""Explanation: Manage the bot's settings
 Requires: Administrator
-Usage: ``/config enable``""")
-        embed.add_field(name=f"/config disable", value=f"""Explanation: Disable any settings you want to
-Requires: Administrator
-Usage: ``/config disable``""")
+Usage: ``/dashboard``""")
         embed.add_field(name=f"/verifymessage", value=f"""Explanation: Send a verification message to a channel
 Requires: Administrator
 Usage: ``/verifymessage <#channel> [Custom (True/False)]``""")
         embed.add_field(name=f"/botinfo", value=f"""Explanation: Shows general bot info
 Requires: None
 Usage: ``/botinfo``""")
-        embed.add_field(name="\u200B", value=f"[Support Server](https://discord.gg/aGzvXvTkP8) | [Invite Me](https://discord.com/api/oauth2/authorize?client_id=981835181243658260&permissions=8&scope=bot%20applications.commands) | [Vote](https://top.gg/bot/828584622156939274/vote)", inline=False)
-
-
-        await ctx.send(embed=embed, view=HelpButtons())
+        embed.add_field(name="\u200B", value=f"[Support Server]({DISCORDLINK}) | [Invite Me]({INVITELINK}) | [Vote]({VOTELINK}) | [Privacy Policy]({PRIVACYLINK})", inline=False)
 
-
-def setup(client):
+        await interaction.send(embed=embed, view=BotInfoLinkButton())
+
+def setup(client: commands.Bot):
     client.add_cog(Help(client))

cogs/misc.py

@@ -0,0 +1,83 @@
+import nextcord
+from nextcord.ext import commands
+import time
+from nextcord import Interaction
+from views import BotInfoLinkButton, PrivacyPolicyButton
+import cooldowns
+from utils.constants import COLOUR_MAIN, VOTELINK, INVITELINK, DISCORDLINK, PRIVACYLINK
+
+class Misc(commands.Cog):
+    def __init__(self, client: commands.Bot):
+        self.client = client
+
+    @nextcord.slash_command(name="botinfo", description="Information about the bot")
+    async def botinfo(self, interaction: Interaction):
+        await interaction.response.defer()
+        before = time.monotonic()
+        msg = await interaction.send("Loading bot information")
+        ping = (time.monotonic() - before) * 1000
+        users = 0
+        for guild in self.client.guilds:
+            users+=guild.member_count
+        embed = nextcord.Embed(title="Bot Infomation", description=f"""Ping: {round(ping)}ms
+Server count: {str(len(self.client.guilds))}
+User count: {users:,}
+Support Server: [Need some support?]({DISCORDLINK})
+Invite: [Invite Me]({INVITELINK})
+Vote: [Vote for me]({VOTELINK})
+Privacy: [Privacy Policy]({PRIVACYLINK})""", colour=COLOUR_MAIN)
+        await msg.edit(content = " ", embed=embed, view=BotInfoLinkButton())
+
+    @nextcord.slash_command(name="privacy", description="Get the link to our privacy policy")
+    async def privacy(self, interaction:Interaction):
+        await interaction.send(content="", view=PrivacyPolicyButton())
+
+
+    @nextcord.slash_command(name="shardinfo", description="Get information on all shards")
+    @cooldowns.cooldown(1, 30, bucket=cooldowns.SlashBucket.author)
+    async def shardinfo(self, interaction:Interaction):
+        await interaction.response.defer()
+        description = ""
+        for shard in self.client.shards:
+            specshard = self.client.get_shard(shard)
+            shard_servers = len([guild for guild in self.client.guilds if guild.shard_id == shard])
+            description+=f"\n**Shard {shard+1}** has `{round(specshard.latency*100)}ms` latency with `{shard_servers} servers`"
+
+        if not interaction.guild:
+            await interaction.send(embed=nextcord.Embed(title="Simple Verification Shard Information", description=f"All current shards:\n {description}", colour=COLOUR_MAIN))
+        else:
+            guild_shard = interaction.guild.shard_id
+            guildspecshard = self.client.get_shard(guild_shard)
+            guild_shard_servers = len([guild for guild in self.client.guilds if guild.shard_id == guild_shard])
+            await interaction.send(f"This server's shard is shard **{guild_shard+1}** with `{round(guildspecshard.latency*100)}ms` and `{guild_shard_servers} servers`", embed=nextcord.Embed(title="Simple Verification Shard Information", description=f"All current shards:\n {description}", colour=COLOUR_MAIN))
+
+
+
+    @nextcord.slash_command(name="shardinfo", description="Get information on all shards")
+    @cooldowns.cooldown(1, 30, bucket=cooldowns.SlashBucket.author)
+    async def shardinfo(self, interaction:Interaction):
+        await interaction.response.defer()
+        description = ""
+        for shard in self.client.shards:
+            specshard = self.client.get_shard(shard)
+            shard_servers = len([guild for guild in self.client.guilds if guild.shard_id == shard])
+            description+=f"\n**Shard {shard+1}** has `{round(specshard.latency*100)}ms` latency with `{shard_servers} servers`"
+
+        if not interaction.guild:
+            await interaction.send(embed=nextcord.Embed(title="Simple Verification Shard Information", description=f"All current shards:\n {description}", colour=COLOUR_MAIN))
+        else:
+            guild_shard = interaction.guild.shard_id
+            guildspecshard = self.client.get_shard(guild_shard)
+            guild_shard_servers = len([guild for guild in self.client.guilds if guild.shard_id == guild_shard])
+            await interaction.send(f"This server's shard is shard **{guild_shard+1}** with `{round(guildspecshard.latency*100)}ms` and `{guild_shard_servers} servers`", embed=nextcord.Embed(title="Simple Verification Shard Information", description=f"All current shards:\n {description}", colour=COLOUR_MAIN))
+
+
+    @nextcord.slash_command(name="debug", description="Get information useful for debugging")
+    @cooldowns.cooldown(1,10, bucket=cooldowns.SlashBucket.channel)
+    async def _debug(self, interaction:Interaction):
+        await interaction.response.defer()
+        await interaction.send(f"**Guild ID:** {interaction.guild.id}\n**Channel ID:** {interaction.channel.id}")
+
+
+def setup(client: commands.Bot):
+    client.add_cog(Misc(client))