EmoteManager/bot.py

#!/usr/bin/env python3

# © lambda#0987 <lambda@lambda.dance>
# SPDX-License-Identifier: AGPL-3.0-or-later

import sys
import asyncio
import base64
import logging
import traceback

import nextcord
from bot_bin.bot import Bot
from discord.ext import commands

logging.basicConfig(level=logging.WARNING)
logging.getLogger('discord').setLevel(logging.INFO)
logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)

# SelectorEventLoop on windows doesn't support subprocesses lol
if sys.platform == 'win32':
    loop = asyncio.ProactorEventLoop()
    asyncio.set_event_loop(loop)


class Bot(Bot):
    startup_extensions = (
        'cogs.emote',
        'cogs.meta',
        'bot_bin.debug',
        'bot_bin.misc',
        'bot_bin.systemd',
        'jishaku',
    )

    def __init__(self, **kwargs):
        with open('data/config.py', encoding='utf-8') as f:
            config = eval(f.read(), {})

        super().__init__(config=config, **kwargs)
        # allow use of the bot's user ID before ready()
        token_part0 = self.config['tokens']['discord'].partition('.')[
            0].encode()
        self.user_id = int(base64.b64decode(
            token_part0 + b'=' * (3 - len(token_part0) % 3)))

    def process_config(self):
        """Load the emojis from the config to be used when a command fails or succeeds
        We do it this way so that they can be used anywhere instead of requiring a bot instance.
        """
        super().process_config()
        import utils.misc
        default = ('❌', '✅')
        utils.SUCCESS_EMOJIS = utils.misc.SUCCESS_EMOJIS = (
            self.config.get('response_emojis', {}).get('success', default))


def main():
    import sys

    if len(sys.argv) == 1:
        shard_count = None
        shard_ids = None
    elif len(sys.argv) < 3:
        print(
            'Usage:', sys.argv[0], '[<shard count> <hyphen-separated list of shard IDs>]', file=sys.stderr)
        sys.exit(1)
    else:
        shard_count = int(sys.argv[1])
        shard_ids = list(map(int, sys.argv[2].split('-')))

    Bot(
        intents=nextcord.Intents(
            guilds=True,
            # we hardly need DM support but it's helpful to be able to run the help/support commands in DMs
            messages=True,
            # we don't need DM reactions because we don't ever paginate in DMs
            guild_reactions=True,
            emojis=True,
            # everything else, including `members` and `presences`, is implicitly false.
        ),

        # the least stateful bot you will ever see 😎
        chunk_guilds_at_startup=False,
        member_cache_flags=nextcord.MemberCacheFlags.none(),
        # disable message cache
        max_messages=None,

        shard_count=shard_count,
        shard_ids=shard_ids,
    ).run()


if __name__ == '__main__':
    main()