Added custom prefix & info
This commit is contained in:
18
cogs/bump.py
18
cogs/bump.py
@@ -1,6 +1,7 @@
|
||||
import discord, io, traceback, json, os
|
||||
import discord, io, traceback, json, os, asyncio
|
||||
|
||||
from colorama import Fore, Style, init
|
||||
from dateparser import parse
|
||||
|
||||
init(autoreset=True)
|
||||
|
||||
@@ -23,10 +24,11 @@ class Bumps(commands.Cog):
|
||||
async def bump(self, ctx):
|
||||
server = Servers(ctx.guild.id)
|
||||
guild = ctx.guild
|
||||
prefix = Servers(guild.id).getPrefix() if Servers(guild.id).hasPrefix else self.config["prefix"]
|
||||
|
||||
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!"))
|
||||
return await ctx.send(embed=Embeds(f"You must setup this server first! Use `{prefix}setup` to do so!").error())
|
||||
|
||||
servers = Servers().get_all()
|
||||
|
||||
@@ -89,12 +91,22 @@ class Bumps(commands.Cog):
|
||||
|
||||
#os.remove("cache_data.json")
|
||||
|
||||
return await ctx.send(embed=discord.Embed(
|
||||
done_message = 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']}"))
|
||||
|
||||
if settings["show_motd"]:
|
||||
await asyncio.sleep(settings["show_motd_wait"])
|
||||
return await done_message.edit(embed=discord.Embed(
|
||||
title="🗞️ Message Of The Day 🗞️",
|
||||
description=Data("motd").read(),
|
||||
color=discord.Color.green()
|
||||
))
|
||||
else:
|
||||
return
|
||||
|
||||
def setup(bot):
|
||||
bot.add_cog(Bumps(bot))
|
||||
38
cogs/info.py
Normal file
38
cogs/info.py
Normal file
@@ -0,0 +1,38 @@
|
||||
import discord
|
||||
|
||||
from core.database import Servers
|
||||
from core.files import Data
|
||||
|
||||
commands = discord.ext.commands
|
||||
|
||||
class Info(commands.Cog):
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
self.config = Data("config").yaml_read()
|
||||
|
||||
@commands.guild_only()
|
||||
@commands.command(aliases=['about'])
|
||||
async def info(self, ctx):
|
||||
return await ctx.send(embed=discord.Embed(
|
||||
title=f"{self.config['bot_name']} | Information",
|
||||
color=discord.Color.blurple()
|
||||
)
|
||||
.add_field(name="Version", value=self.config['version'])
|
||||
.add_field(name="Library", value=f"Discord.py v{discord.__version__}")
|
||||
.add_field(name="Latency", value=f"{round(self.bot.latency*1000)}ms")
|
||||
.add_field(name="Servers", value=len(self.bot.guilds))
|
||||
.add_field(name="Active Servers", value=len([i for i in Servers().get_all()]))
|
||||
.add_field(name="Open Source", value="[View](https://github.com/Nemika-Haj/BytesBump)") # DO NOT REMOVE
|
||||
.set_footer(text="Made by • " + ', '.join([str((await self.bot.fetch_user(i))) for i in self.config['managers']])))
|
||||
|
||||
@commands.guild_only()
|
||||
@commands.command(aliases=['support'])
|
||||
async def help(self, ctx):
|
||||
return await ctx.send(embed=discord.Embed(
|
||||
title=f"{self.config['bot_name']} | Help",
|
||||
description=Data("help").read(),
|
||||
color=discord.Color.blurple()
|
||||
))
|
||||
|
||||
def setup(bot):
|
||||
bot.add_cog(Info(bot))
|
||||
37
cogs/prefix.py
Normal file
37
cogs/prefix.py
Normal file
@@ -0,0 +1,37 @@
|
||||
import discord
|
||||
|
||||
from core.database import Servers
|
||||
from core.files import Data
|
||||
from core import embeds
|
||||
|
||||
commands = discord.ext.commands
|
||||
|
||||
def getPrefix(bot, message):
|
||||
prefix = [Data('config').yaml_read()['prefix']]
|
||||
if message.guild:
|
||||
server = Servers(message.guild.id)
|
||||
prefix = [server.getPrefix()]
|
||||
return commands.when_mentioned_or(*prefix)(bot, message)
|
||||
|
||||
class SetPrefix(commands.Cog):
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
|
||||
@commands.guild_only()
|
||||
@commands.has_permissions(manage_guild=True)
|
||||
@commands.command()
|
||||
async def setprefix(self, ctx, *, prefix=None):
|
||||
server = Servers(ctx.guild.id)
|
||||
if not prefix or prefix == "p!":
|
||||
if server.hasPrefix: server.deletePrefix
|
||||
return await ctx.send(embed=embeds.Embeds(f"The prefix was reset to default! `({Data('config').yaml_read()['prefix']})`").success())
|
||||
server.setPrefix(prefix)
|
||||
return await ctx.send(embed=embeds.Embeds(f"The prefix was set to `{prefix}`!").success())
|
||||
|
||||
@commands.Cog.listener()
|
||||
async def on_guild_remove(self, guild):
|
||||
server = Servers(guild.id)
|
||||
if server.hasPrefix: server.deletePrefix
|
||||
|
||||
def setup(bot):
|
||||
bot.add_cog(SetPrefix(bot))
|
||||
@@ -13,15 +13,20 @@ class BumpSetup(commands.Cog):
|
||||
global setting_up
|
||||
setting_up = []
|
||||
|
||||
@commands.Cog.listener('on_guild_remove')
|
||||
async def remove_guild(self, guild):
|
||||
Servers(guild.id).delete()
|
||||
|
||||
@commands.guild_only()
|
||||
@commands.has_permissions(manage_guild=True)
|
||||
@commands.check(lambda ctx: ctx.guild not in setting_up)
|
||||
@commands.command()
|
||||
async def setup(self, ctx):
|
||||
guild = ctx.guild
|
||||
prefix = Servers(guild.id).getPrefix() if Servers(guild.id).hasPrefix else self.config["prefix"]
|
||||
|
||||
if Servers(guild.id).get():
|
||||
return await ctx.send(embed=Embeds(f"This server was already setup! Use `{self.config['prefix']}delete` to initialize another setup!").error())
|
||||
return await ctx.send(embed=Embeds(f"This server was already setup! Use `{prefix}delete` to initialize another setup!").error())
|
||||
|
||||
embed = discord.Embed(
|
||||
title="🔄 Setting Up...",
|
||||
|
||||
Reference in New Issue
Block a user