Archived
generated from imcompiling/discord-bot-template
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
"""Defines the daily loop that runs the Dakiny bot functions."""
|
|
|
|
from datetime import UTC, datetime, time
|
|
from typing import TYPE_CHECKING
|
|
|
|
from nextcord.ext import commands, tasks
|
|
from pytz import timezone
|
|
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from logger import ocd_logger
|
|
from main import birthday_list
|
|
from settings import settings
|
|
|
|
if TYPE_CHECKING:
|
|
import nextcord
|
|
|
|
|
|
class DailyLoop(commands.Cog, name="Daily Loop"):
|
|
"""Class for performing daily functions."""
|
|
|
|
def __init__(self, bot: commands.Bot) -> None:
|
|
self.bot = bot
|
|
self.current_election = None
|
|
self.update_birthdays()
|
|
|
|
# pylint doesn't recognize that daily_loop is subclassed from tasks.loop.
|
|
self.daily_loop.start() # pylint: disable=no-member
|
|
|
|
@tasks.loop(time=time(hour=0, tzinfo=timezone("UTC")))
|
|
async def daily_loop(self) -> None:
|
|
"""Run daily repeated functions."""
|
|
self.update_birthdays()
|
|
|
|
await self.cakedays()
|
|
|
|
@daily_loop.before_loop
|
|
async def before_loop(self) -> None:
|
|
"""Actions run before the loop starts. Want to wait for bot to be ready and get the current cycle."""
|
|
await self.bot.wait_until_ready()
|
|
|
|
def update_birthdays(self) -> None:
|
|
"""Goes through database to collect all birthdays that are today. Adds user's id to birthday list."""
|
|
birthday_list.clear()
|
|
|
|
today = datetime.now(tz=UTC).date()
|
|
birthday = f"{today.month}/{today.day}"
|
|
|
|
stmt = select(UserBirthday.UserId).where(UserBirthday.Birthday == birthday)
|
|
with Session(engine) as session:
|
|
to_add = session.scalars(stmt)
|
|
birthday_list.extend(to_add.all())
|
|
|
|
async def cakedays(self) -> None:
|
|
"""
|
|
Loops through the guild's humans checking for joined_at dates that match today's date.
|
|
|
|
Sends a message saying happy cakeday.
|
|
"""
|
|
today = datetime.now(tz=UTC).date()
|
|
if (guild := self.bot.get_guild(settings.params.guild_id)) is None:
|
|
ocd_logger.warning("Guild not found in bot.")
|
|
return
|
|
if (announce_channel := guild.get_channel(settings.channel_ids.general)) is None or not isinstance(
|
|
announce_channel, nextcord.TextChannel
|
|
):
|
|
ocd_logger.warning("Set general channel isn't found or isn't a text channel.")
|
|
return
|
|
|
|
cakedays: list[nextcord.Member] = [
|
|
human
|
|
for human in guild.humans
|
|
if human.joined_at is not None and human.joined_at.month == today.month and human.joined_at.day == today.day
|
|
]
|
|
|
|
for member in cakedays:
|
|
years = today.year - member.joined_at.year
|
|
year_str = "years" if years > 1 else "year"
|
|
msg = f"Happy Cake day! <@{member.id}> joined the discord {years} {year_str} ago!"
|
|
await announce_channel.send(msg)
|