qrm2/main.py

94 lines
2.1 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
"""
2019-10-04 11:39:12 -04:00
qrm, a bot for Discord
---
2019-10-05 02:23:11 -04:00
Copyright (C) 2019 Abigail Gold, 0x5c
2019-10-05 02:23:11 -04:00
This file is part of discord-qrmbot and is released under the terms of the GNU
General Public License, version 2.
"""
from types import SimpleNamespace
import discord
from discord.ext import commands, tasks
import info
import options as opt
import keys
# --- Global settings ---
debug_mode = opt.debug # Separate assignement in-case we define an override (ternary operator goes here)
class GlobalSettings(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.opt = opt
self.keys = keys
self.info = info
self.colours = SimpleNamespace(good=0x2dc614, neutral=0x2044f7, bad=0xc91628)
self.debug = debug_mode
# --- Bot setup ---
bot = commands.Bot(command_prefix=opt.prefix, description=info.description, help_command=commands.MinimalHelpCommand())
# --- Commands ---
# --- Events ---
@bot.event
async def on_ready():
print(f"Logged in as: {bot.user} - {bot.user.id}")
print("------")
# --- Tasks ---
@tasks.loop(minutes=5)
async def _ensure_activity():
await bot.change_presence(activity=discord.Game(name="with lids on 7.200"))
@_ensure_activity.before_loop
async def _before_ensure_activity():
await bot.wait_until_ready()
# --- Run ---
bot.add_cog(GlobalSettings(bot))
2019-10-04 18:16:47 -04:00
bot.load_extension("cogs.basecog")
bot.load_extension("cogs.morsecog")
2019-10-04 20:19:31 -04:00
bot.load_extension("cogs.funcog")
_ensure_activity.start()
try:
bot.run(keys.discord_token)
except discord.LoginFailure as ex: # Miscellaneous authentications errors: borked token and co
if debug_mode:
raise
raise SystemExit("Error: Failed to authenticate: {}".format(ex))
except discord.ConnectionClosed as ex: # When the connection the the gateway (websocket) is closed
if debug_mode:
raise
raise SystemExit("Error: Discord gateway connection closed: [Code {}] {}".format(ex.code, ex.reason))
except ConnectionResetError as ex: # More generic connection reset error
if debug_mode:
raise
raise SystemExit("ConnectionResetError: {}".format(ex))