Added Bump

cogs/bump.py

@@ -0,0 +1,99 @@
+import discord, io, traceback, json, os
+
+from colorama import Fore, Style, init
+
+init(autoreset=True)
+
+from core.database import Servers
+from core.files import Data
+from core.embeds import Embeds
+
+commands = discord.ext.commands
+
+settings = Data("settings").json_read()
+
+class Bumps(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+        self.config = Data("config").yaml_read()
+
+    @commands.guild_only()
+    @commands.cooldown(1, settings['cooldown'], commands.BucketType.guild)
+    @commands.command()
+    async def bump(self, ctx):
+        server = Servers(ctx.guild.id)
+        
+        if not server.get():
+            ctx.command.reset_cooldown(ctx)
+            return await ctx.send(embed=Embeds(f"You must setup this server first! Use `{self.config['prefix']}setup` to do so!"))
+
+        servers = Servers().get_all()
+
+        success, fail = 0, 0
+
+        for entry in servers:
+            try:
+                server = self.bot.get_guild(entry['_id'])
+                webhook = await self.bot.fetch_webhook(entry['webhook'])
+                invite_channel = self.bot.get_channel(entry['invite'])
+                
+                invite = await invite_channel.create_invite(max_uses=0, max_age=0, unique=False)
+
+                embed = discord.Embed(
+                    title=server.name,
+                    description=entry['description'],
+                    color=discord.Color(value=entry['color']),
+                    url=invite.url
+                )
+
+                embed.add_field(name="🌍 Members", value=len(server.members))
+                embed.add_field(name="🤣 Emojis", value=f"{len(server.emojis)}/{server.emoji_limit}")
+                embed.add_field(name="💎 Boost Tier", value=f"Tier {server.premium_tier} ({server.premium_subscription_count} Boosts)")
+                embed.add_field(name="👑 Owner", value=str(server.owner))
+                embed.add_field(name="🔗 Invite", value=f"[Click to join!]({invite.url})")
+                embed.set_thumbnail(url=server.icon_url_as(static_format="png"))
+                embed.set_footer(text=f"Powered by • {self.config['bot_name']}")
+
+
+                await webhook.send(
+                    username=self.config['bot_name'],
+                    avatar_url=self.bot.user.avatar_url,
+                    embed=embed
+                )
+
+                success += 1
+            except Exception as e:
+                error = f"{e.__class__.__name__}"
+                value = io.StringIO().getvalue()
+                print(f"{Fore.RED}[ERROR] {Style.RESET_ALL}{error}\n{Fore.YELLOW}Error was recorded in {Fore.RED}error.log")
+                #with open("error.log", "w+") as f:
+                    #f.write(f"{value}{traceback.format_exc()}")
+
+                """
+                with open("cache_data.json", "w+") as f:
+                    json.dump(entry, f, indent=4)
+                """
+                Servers(entry['_id']).delete()
+
+                try:
+                    await self.bot.get_guild(entry['_id']).owner.send(embed=discord.Embed(
+                        title="⚠️ Server Removed ⚠️",
+                        description="Your server was removed from the database because it caused an error! Make sure I have permission to `Manage Webhooks` and `Create Instant Invites`! I've attached your server info below.",
+                        color=discord.Color.red()
+                    ))
+
+                except: pass
+
+                fail += 1
+
+                #os.remove("cache_data.json")
+        
+        return await ctx.send(embed=discord.Embed(
+            title="⏫ Server Bumped",
+            description=f"Your server was bumped to `{success+fail}` servers!\n✅ There were `{success}` successful bumps!\n❎ There were `{fail}` failed ones, they got booted from the Database!",
+            color=discord.Color.green()
+        )
+        .set_footer(text=f"Powered by • {self.config['bot_name']}"))
+
+def setup(bot):
+    bot.add_cog(Bumps(bot))

cogs/handler.py

@@ -1,4 +1,6 @@
 import discord
+
+from humanfriendly import format_timespan as ftime
 from core import embeds
 
 commands = discord.ext.commands
@@ -17,7 +19,7 @@ class ErrorHandler(commands.Cog):
             return await ctx.send(embed=embeds.Embeds("You are not allowed to do this.").error())
         elif isinstance(error, commands.CommandOnCooldown):
             seconds = error.retry_after
-            return await ctx.send(embed=embeds.Embeds(f"**You are on cooldown!** You can use this command again in **{seconds} seconds**.").error())
+            return await ctx.send(embed=embeds.Embeds(f"**You are on cooldown!** You can use this command again in **{ftime(seconds)}**.").error())
         else:
             await ctx.send(embed=embeds.Embeds("There was an error executing this command.").error(Error=error))
             raise error

cogs/setup.py

@@ -94,7 +94,7 @@ class BumpSetup(commands.Cog):
 
         webhook = await listing.create_webhook(name=self.config['bot_name'])
 
-        Servers(ctx.guild.id).add(webhook=webhook.id, listing=listing.id, color=color, description=description)
+        Servers(ctx.guild.id).add(webhook=webhook.id, invite=invite.id, color=color, description=description)
 
         await ctx.send("Setup complete! Server added to DB and the webhook was created.")
 
@@ -152,9 +152,19 @@ class BumpSetup(commands.Cog):
                 check=lambda r,u: r.emoji == "♻️" and r.message.id == del_message.id and u.id == ctx.author.id
             )
         except asyncio.TimeoutError:
+            try:
+                wh = await self.bot.fetch_webhook(cache_data['webhook'])
+                await wh.delete()
+            except:
+                pass
             return await del_message.remove_reaction("♻️", self.bot.user)
 
         if Servers(ctx.guild.id).get():
+            try:
+                wh = await self.bot.fetch_webhook(cache_data['webhook'])
+                await wh.delete()
+            except:
+                pass
             return await ctx.send(embed=discord.Embed(
                 title="❎ Restore Failed",
                 description="The server seems to have been setup from the beginning, therefore restore is not possible.",