discord_bot

Complexity

Total Complexity

12

Size/Duplication

Total Lines	55
Duplicated Lines	0 %

Test Coverage

Coverage

44.83%

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	DiscordBot.begin()	0	3	1
A	DiscordBot.on_message()	0	7	2
A	DiscordBot.__init__()	0	8	1
A	DiscordBot.on_message_edit()	0	3	1
B	DiscordBot.checkMarvinActions()	0	13	7

#! /usr/bin/env python3
# -*- coding: utf-8 -*-

"""
Module for the Discord bot.

Connecting, sending and receiving messages and doing custom actions.
"""

import discord

from bot import Bot


class DiscordBot(discord.Client, Bot):
    """Bot implementing the discord protocol"""
    def __init__(self):
        Bot.__init__(self)
        self.CONFIG = {
            "token": ""
        }
        intents = discord.Intents.default()
        intents.message_content = True
        discord.Client.__init__(self, intents=intents)

    def begin(self):
        """Start the bot"""
        self.run(self.CONFIG.get("token"))

    async def checkMarvinActions(self, message):
        """Check if Marvin should perform any actions"""
        words = self.tokenize(message.content)
        if self.user.mentioned_in(message) or self.user.name.lower() in words:
            for action in self.ACTIONS:
                response = action(words)
                if response:
                    await message.reply(response)
        else:
            for action in self.GENERAL_ACTIONS:
                response = action(words)
                if response:
                    await message.reply(response)

    async def on_message(self, message):
        """Hook run on every message"""
        self.MSG_LOG.debug("#%s <%s> %s", message.channel.name, message.author, message.content)
        if message.author.name == self.user.name:
            # don't react to own messages
            return
        await self.checkMarvinActions(message)

    async def on_message_edit(self, _, after):
        """Hook run on every edited message"""
        await self.on_message(after)