bot.cogs.tasks

Complexity

Total Complexity

5

Size/Duplication

Total Lines	54
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	TaskCog.report_errors()	0	24	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"):
    """
    Task Cog

        Cog that holds tasks for the bot

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

    """

    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) -> None:
        """
        Report error

        Every 24 hours, all errors for the current day are send to the admin channels.

        :return: None
        """
        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))