main.py

import asyncio
import os
import traceback
from collections import OrderedDict
from importlib import import_module
import logging

import discord

from config import DISCORD_TOKEN, PREFIX


class LimitedSizeDict(OrderedDict):


    def __init__(self, *args, **kwds):

        self.size_limit = kwds.pop("size_limit", None)
        OrderedDict.__init__(self, *args, **kwds)
        self._check_size_limit()

    def __setitem__(self, key, value):

        OrderedDict.__setitem__(self, key, value)
        self._check_size_limit()

    def _check_size_limit(self):

        if self.size_limit is not None:
            while len(self) > self.size_limit:
                self.popitem(last=False)


client = discord.Client()


@client.event
async def on_ready():

    global message_id_to_author_id
    message_id_to_author_id = LimitedSizeDict(size_limit=100)

    global user_message_id_to_bot_message
    user_message_id_to_bot_message = LimitedSizeDict(size_limit=100)

    print('起動しました')


@client.event
async def on_message(message):
    await reply(message)
    if 'にゃーん' in message.content:
        message.add_reaction('😿')


@client.event
async def on_message_edit(before, after):

    if before.id in user_message_id_to_bot_message:
        try:
            await globals()['user_message_id_to_bot_message'][before.id].delete()
        except discord.errors.NotFound:
            pass

    await reply(after)


@client.event
async def on_reaction_add(reaction, user):

    if user == client.user:
        return

    if reaction.message.author == client.user \
        and reaction.message.id in message_id_to_author_id \
        and user.id == message_id_to_author_id[reaction.message.id]:

        if str(reaction.emoji) in ('🚮', '✖️', '🗑️'):
            await reaction.message.delete()


async def reply(message):

    # message主がbotならば無視
    if message.author.bot and not message.author.id == 614130545227726849:
        return

    if message.content.startswith(PREFIX):

        command = message.content[len(PREFIX):].split()[0]
        arg = message.content[len(PREFIX) + len(command):].lstrip()

        if os.path.exists(f'commands/command_{command}.py'):

            tmp_module = import_module(f'commands.command_{command}')
            async with message.channel.typing():
                try:
                    sent_message = await tmp_module.main(message, arg)
                except discord.Forbidden:
                    return
                except Exception as e:
                    logging.exception(e)
                    embed = discord.Embed(
                        title='内部エラーが発生しました',
                        description='エラーが続く場合は公式サーバで報告してください\n'
                            + 'https://discord.gg/7gypE3Q',
                        color=0xff0000
                        )
                    embed.set_author(name=message.author.name, icon_url=message.author.avatar_url)
                    sent_message = await message.channel.send(embed=embed)
                message_id_to_author_id[sent_message.id] = message.author.id
                user_message_id_to_bot_message[message.id] = sent_message
                await sent_message.add_reaction('🚮')


client.run(DISCORD_TOKEN)