bot.cogs.tasks

Complexity

Total Complexity

5

Size/Duplication

Total Lines	55
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	TaskCog.report_errors()	0	21	3
A	TaskCog.__init__()	0	3	1

"""Cog for tasks that are scheduled to run"""
import logging
from datetime import datetime
from discord.ext import commands, tasks
from bot import utils

log = logging.getLogger("bot")


class TaskCog(commands.Cog, name="Tasks"):
    """TaskCog
    ---

    Cog that holds tasks for the bot

    Tasks
    ---
        :ref:`error_report`: Output all errors for a day to admin channels.

    Arguments:
    ---
        bot {discord.commands.Bot} -- The bot
    """

    def __init__(self, bot):
        self.bot = bot
        self.report_errors.start()  # pylint: disable=no-member

    @tasks.loop(hours=24.0)
    async def report_errors(self):
        """report_errors
        ---
        Every 24 hours, all errors for the current day are send to the admin channels.
        """
        date = datetime.utcnow().strftime("%Y-%m-%d")
        error_record = [
            error
            for error in await utils.select(
                "errors", "id, message, error, ack", "date_trunc('day', time)", date
            )
            if error[-1] is False
        ]
        if not error_record:
            errors = "No errors found for {}".format(date)
        else:
            errors = "Errors for {}.\n".format(date)
            for error in error_record:
                errors += f"- {error[0]}: {error[1]}; {error[2]}\n\n"
        await utils.admin_log(self.bot, errors, True)


def setup(bot):
    """Needed for extension loading"""
    bot.add_cog(TaskCog(bot))