async def reload(self, ctx: CommandContext, *extensions): for extension in (extensions or self.bot.extensions.copy().keys()): self.bot.unload_extension(extension) messages: List[Message] = [ await ctx.send(embed=Embed(color=Color.red(), description='Unloaded extensions')), ctx.message ] if len(extensions) == 0: load_all_modules(self.bot) else: for extension in extensions: try: self.bot.load_extension(extension) except: messages.append(await ctx.send(embed=Embed( title=f"Failed to load module `{extension}`", color=Color.red()))) messages.append(await ctx.send(embed=Embed( title= f"Reloaded {len(extensions) or len(self.bot.extensions)} extension(s)", color=Color.green()))) await asyncio.sleep(10) for mes in messages: await mes.delete()
def test_login(): from main import client client.add_cog(Cog(client)) load_all_modules(client, module_folder='../modules', module_package='modules') client.run(os.environ.get('test_token', 'no_token'))
async def reload(self, ctx: CommandContext, *extensions): for extension in (extensions or self.bot.extensions.copy().keys()): self.bot.unload_extension(extension) await ctx.send( embed=Embed( color=Color.red(), description='Unloaded extensions', )) if len(extensions) == 0: load_all_modules(self.bot) else: for extension in extensions: try: self.bot.load_extension(extension) except: await ctx.send( embed=Embed( title=f"Failed to load module `{extension}`", color=Color.red())) await ctx.send( embed=Embed( title=f"Reloaded {len(extensions) or len(self.bot.extensions)} extension(s)", color=Color.green()))
import discord from discord.ext import commands from discord.ext.commands import Bot, when_mentioned_or from database import GuildModel from utils import load_all_modules def get_server_prefix(bot: commands.Bot, message: discord.Message): prefixes = ['-'] if message.guild: prefixes = [GuildModel.get_or_insert(message.guild).prefix] return when_mentioned_or(*prefixes)(bot, message) def get_guild(id): return client.get_guild(id) client = Bot(command_prefix=get_server_prefix) if __name__ == '__main__': from config import config load_all_modules(client) client.run(config.token)
from discord.ext.commands import Bot, when_mentioned_or from config import config from utils import load_all_modules bot = Bot( command_prefix=when_mentioned_or('~'), self_bot=True, ) load_all_modules(bot) if __name__ == '__main__': bot.run(config.token, bot=False)