mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 03:08:55 -05:00
9fc4e54ece
After some considerations, while the flusher was an interesting experiment, it doesn't seem to be reliable enough. It's been removed in favor of the helpers autosaving on any change in a threadsafe way.
93 lines
2.8 KiB
Python
93 lines
2.8 KiB
Python
from core.bot import Red, ExitCodes
|
|
from core.global_checks import init_global_checks
|
|
from core.events import init_events
|
|
from core.settings import parse_cli_flags
|
|
import asyncio
|
|
import discord
|
|
import logging.handlers
|
|
import logging
|
|
import os
|
|
import sys
|
|
|
|
#
|
|
# Red - Discord Bot v3
|
|
#
|
|
# Made by Twentysix, improved by many
|
|
#
|
|
|
|
|
|
def init_loggers(cli_flags):
|
|
dpy_logger = logging.getLogger("discord")
|
|
dpy_logger.setLevel(logging.WARNING)
|
|
console = logging.StreamHandler()
|
|
console.setLevel(logging.WARNING)
|
|
dpy_logger.addHandler(console)
|
|
|
|
logger = logging.getLogger("red")
|
|
|
|
red_format = logging.Formatter(
|
|
'%(asctime)s %(levelname)s %(module)s %(funcName)s %(lineno)d: '
|
|
'%(message)s',
|
|
datefmt="[%d/%m/%Y %H:%M]")
|
|
|
|
stdout_handler = logging.StreamHandler(sys.stdout)
|
|
stdout_handler.setFormatter(red_format)
|
|
|
|
if cli_flags.debug:
|
|
os.environ['PYTHONASYNCIODEBUG'] = '1'
|
|
logger.setLevel(logging.DEBUG)
|
|
else:
|
|
logger.setLevel(logging.WARNING)
|
|
|
|
fhandler = logging.handlers.RotatingFileHandler(
|
|
filename='red.log', encoding='utf-8', mode='a',
|
|
maxBytes=10**7, backupCount=5)
|
|
fhandler.setFormatter(red_format)
|
|
|
|
logger.addHandler(fhandler)
|
|
logger.addHandler(stdout_handler)
|
|
|
|
return logger
|
|
|
|
if __name__ == '__main__':
|
|
cli_flags = parse_cli_flags()
|
|
log = init_loggers(cli_flags)
|
|
description = "Red v3 - Alpha"
|
|
red = Red(cli_flags, description=description, pm_help=None)
|
|
init_global_checks(red)
|
|
init_events(red, cli_flags)
|
|
red.load_extension('core')
|
|
if cli_flags.dev:
|
|
pass # load dev cog here?
|
|
|
|
token = os.environ.get("RED_TOKEN", red.db.get_global("token", None))
|
|
|
|
if token is None:
|
|
log.critical("No token to login with")
|
|
sys.exit(1)
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
try:
|
|
loop.run_until_complete(red.start(token, bot=not cli_flags.not_bot))
|
|
except discord.LoginFailure:
|
|
log.critical("This token doesn't seem to be valid. If it belongs to "
|
|
"a user account, remember that the --not-bot flag "
|
|
"must be used. For self-bot functionalities instead, "
|
|
"--self-bot")
|
|
except KeyboardInterrupt:
|
|
log.info("Keyboard interrupt detected. Quitting...")
|
|
loop.run_until_complete(red.logout())
|
|
red._shutdown_mode = ExitCodes.SHUTDOWN
|
|
except Exception as e:
|
|
log.critical("Fatal exception", exc_info=e)
|
|
loop.run_until_complete(red.logout())
|
|
finally:
|
|
pending = asyncio.Task.all_tasks(loop=red.loop)
|
|
gathered = asyncio.gather(*pending, loop=red.loop)
|
|
gathered.cancel()
|
|
red.loop.run_until_complete(gathered)
|
|
gathered.exception()
|
|
sys.exit(red._shutdown_mode.value)
|
|
|