diff options
| -rw-r--r-- | arthur/config.py | 1 | ||||
| -rw-r--r-- | arthur/exts/fun/motivation.py | 37 |
2 files changed, 37 insertions, 1 deletions
diff --git a/arthur/config.py b/arthur/config.py index 146db73..59014a8 100644 --- a/arthur/config.py +++ b/arthur/config.py @@ -18,6 +18,7 @@ class Config( prefixes: tuple[str, ...] = ("arthur ", "M-x ") cloudflare_token: pydantic.SecretStr + youtube_api_key: pydantic.SecretStr | None = None grafana_url: str = "https://grafana.pydis.wtf" grafana_token: pydantic.SecretStr github_token: pydantic.SecretStr diff --git a/arthur/exts/fun/motivation.py b/arthur/exts/fun/motivation.py index d256bd4..4da7460 100644 --- a/arthur/exts/fun/motivation.py +++ b/arthur/exts/fun/motivation.py @@ -8,6 +8,7 @@ from discord.ext import commands, tasks from arthur.bot import KingArthur from arthur.config import CONFIG +from arthur.log import logger MOTIVATION_IMAGE_RE = re.compile(r"data-image=\"(https://assets\.amuniversal\.com/.+?)\"") THE_CAT = "https://avatar.amuniversal.com/feature_avatars/ubadge_images/features/ga/mid_u-201701251612.png" @@ -21,8 +22,42 @@ class Motivation(commands.Cog): self.bot = bot self.devops_channel = bot.get_channel(CONFIG.devops_channel_id) self.send_daily_motivation.start() + if CONFIG.youtube_api_key: + self.send_daily_mission.start() - @tasks.loop(time=time(hour=9)) + @tasks.loop(time=time(hour=12, minute=30, tzinfo=UTC)) + async def send_daily_mission(self) -> None: + """Send the daily mission to the team.""" + params = { + "part": "snippet", + "channelId": "UC4CoHBR01SHu6fMy2EaeWcg", + "key": CONFIG.youtube_api_key.get_secret_value(), + "order": "date", + "maxResults": "2", + } + async with self.bot.http_session.get( + "https://www.googleapis.com/youtube/v3/search", + params=params, + ) as resp: + resp.raise_for_status() + data = await resp.json() + todays_date = datetime.now(UTC).date() + for video in data["items"]: + title: str = video["snippet"]["title"] + if not title.startswith("TODAY'S MISSION:"): + continue + date = datetime.fromisoformat(video["snippet"]["publishedAt"]).date() + if date != todays_date: + continue + + await self.devops_channel.send( + f"[Today's mission](https://www.youtube.com/shorts/{video["id"]["videoId"]})" + ) + break + else: + logger.warning("No mission found for %s", todays_date) + + @tasks.loop(time=time(hour=9, tzinfo=UTC)) async def send_daily_motivation(self) -> None: """Send motivation to the people who need it most.""" today_date = datetime.now(UTC).date().isoformat() |