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)