1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
"""Send motivating messages to the devops team."""
import re
from datetime import UTC, datetime, time
import discord
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"
GARF_URL = "https://www.gocomics.com/garfield/"
class Motivation(commands.Cog):
"""Motivation is the key to productivity."""
def __init__(self, bot: KingArthur) -> None:
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=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()
today_date_url_friendly = today_date.replace("-", "/")
async with self.bot.http_session.get(GARF_URL + today_date_url_friendly) as resp:
resp.raise_for_status()
raw_content = await resp.text()
image = MOTIVATION_IMAGE_RE.search(raw_content).group(1)
embed = discord.Embed(
title=f"Garfield: {today_date}",
url=GARF_URL + today_date_url_friendly,
colour=discord.Colour.orange(),
)
embed.set_author(name="GoComics.com", icon_url=THE_CAT, url=GARF_URL)
embed.set_image(url=image)
await self.devops_channel.send(embed=embed)
async def setup(bot: KingArthur) -> None:
"""Add cog to bot."""
await bot.add_cog(Motivation(bot))
|
