mirror of
https://github.com/UiO-Gaming/ShiteMusicBot.git
synced 2026-11-05 13:56:51 +01:00
149 lines
4.4 KiB
Python
149 lines
4.4 KiB
Python
import discord
|
|
import codecs
|
|
import time
|
|
import sys
|
|
import traceback
|
|
import yaml
|
|
import aiohttp
|
|
|
|
from discord.ext import commands
|
|
from discord.ext.commands.view import StringView
|
|
from cogs.utils.settingsmanager import Settings
|
|
from cogs.utils.localizer import Localizer
|
|
from cogs.utils.localizer import LocalizerWrapper
|
|
from cogs.utils.alias import Aliaser
|
|
from cogs.utils.context import Context
|
|
from cogs.helpformatter import commandhelper
|
|
from cogs.utils.paginator import Scroller
|
|
|
|
|
|
with codecs.open("data/config.yaml", 'r', encoding='utf8') as f:
|
|
conf = yaml.load(f, Loader=yaml.SafeLoader)
|
|
|
|
|
|
initial_extensions = [
|
|
'cogs.cogs',
|
|
'cogs.settings',
|
|
'cogs.misc',
|
|
'cogs.helpformatter'
|
|
]
|
|
|
|
|
|
on_ready_extensions = [
|
|
'cogs.music',
|
|
'cogs.musicevents'
|
|
]
|
|
|
|
|
|
def _get_prefix(bot, message):
|
|
if not message.guild:
|
|
prefix = bot.settings.default_prefix
|
|
return commands.when_mentioned_or(prefix)(bot, message)
|
|
prefixes = bot.settings.get(message.guild, 'prefixes', 'default_prefix')
|
|
return commands.when_mentioned_or(*prefixes)(bot, message)
|
|
|
|
|
|
class Bot(commands.Bot):
|
|
def __init__(self, debug: bool=False):
|
|
super().__init__(command_prefix=_get_prefix,
|
|
description=conf["bot"]["description"])
|
|
|
|
self.settings = Settings(**conf['default server settings'])
|
|
self.APIkeys = conf.get('APIkeys', {})
|
|
|
|
self.localizer = Localizer(conf.get('locale path', "./localization"), conf.get('locale', 'en_en'))
|
|
self.aliaser = Aliaser(conf.get('locale path', "./localization"), conf.get('locale', 'en_en'))
|
|
|
|
self.session = aiohttp.ClientSession(loop=self.loop)
|
|
|
|
self.debug = debug
|
|
|
|
for extension in initial_extensions:
|
|
try:
|
|
self.load_extension(extension)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
async def on_command_error(self, ctx, err):
|
|
if not self.debug:
|
|
if (isinstance(err, commands.MissingRequiredArgument) or
|
|
isinstance(err, commands.BadArgument)):
|
|
paginator = commandhelper(ctx, ctx.command, ctx.invoker, include_subcmd=False)
|
|
scroller = Scroller(ctx, paginator)
|
|
await scroller.start_scrolling()
|
|
|
|
if isinstance(err, commands.CommandInvokeError):
|
|
pass
|
|
|
|
elif isinstance(err, commands.NoPrivateMessage):
|
|
await ctx.send('That command is not available in DMs')
|
|
|
|
elif isinstance(err, commands.CheckFailure):
|
|
print('check failure')
|
|
pass
|
|
|
|
elif isinstance(err, commands.CommandNotFound):
|
|
pass
|
|
else:
|
|
tb = err.__traceback__
|
|
traceback.print_tb(tb)
|
|
print(err)
|
|
|
|
async def on_message(self, message):
|
|
if message.author.bot:
|
|
return
|
|
await self.process_commands(message)
|
|
|
|
async def process_commands(self, message):
|
|
ctx = await self.get_context(message, cls=Context)
|
|
|
|
# Replace aliases with commands
|
|
ctx = self.aliaser.get_command(ctx)
|
|
|
|
# Add the localizer
|
|
if ctx.command and ctx.command.cog_name:
|
|
ctx.localizer = LocalizerWrapper(self.localizer, ctx.locale, ctx.command.cog_name.lower())
|
|
else:
|
|
ctx.localizer = LocalizerWrapper(self.localizer, ctx.locale, None)
|
|
|
|
await self.invoke(ctx)
|
|
|
|
async def on_ready(self):
|
|
if not hasattr(self, 'uptime'):
|
|
self.uptime = time.time()
|
|
if self.debug:
|
|
print('\n\nDebug mode')
|
|
|
|
for extension in on_ready_extensions:
|
|
try:
|
|
self.load_extension(extension)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
print(f'\nLogged in as: {self.user.name}' +
|
|
f' in {len(self.guilds)} servers.')
|
|
print(f'Version: {discord.__version__}\n')
|
|
|
|
await self.change_presence(activity=discord.Game(type=0,
|
|
name=conf["bot"]["playing status"]),
|
|
status=discord.Status.online)
|
|
|
|
def run(self):
|
|
try:
|
|
super().run(conf["bot"]["token"], reconnect=True)
|
|
except Exception as e:
|
|
tb = e.__traceback__
|
|
traceback.print_tb(tb)
|
|
print(e)
|
|
|
|
|
|
def run_bot(debug: bool=False):
|
|
bot = Bot(debug=debug)
|
|
bot.run()
|
|
|
|
if __name__ == '__main__':
|
|
if 'debug' in sys.argv:
|
|
run_bot(debug=True)
|
|
else:
|
|
run_bot(debug=False)
|