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
|
import logging
from random import randint
from typing import Dict, Optional, Union
from discord import Embed
from discord.ext import tasks
from discord.ext.commands import Cog, Context, command
from bot.bot import Bot
log = logging.getLogger(__name__)
URL = "https://xkcd.com/{0}/info.0.json"
LATEST = "https://xkcd.com/info.0.json"
class XKCD(Cog):
"""Retrieving XKCD comics."""
def __init__(self, bot: Bot) -> None:
self.bot = bot
self.latest_comic_info: Dict[str, Union[str, int]] = {}
self.get_latest_comic_info.start()
def cog_unload(self) -> None:
"""Cancels refreshing of the task for refreshing the most recent comic info."""
self.get_latest_comic_info.cancel()
@tasks.loop(minutes=30)
async def get_latest_comic_info(self) -> None:
"""Refreshes latest comic's information ever 30 minutes. Also used for finding a random comic."""
async with self.bot.http_session.get(LATEST) as resp:
if resp.status == 200:
self.latest_comic_info = await resp.json()
else:
log.debug(f"Failed to get latest XKCD comic information. Status code {resp.status}")
@command(name="xkcd")
async def fetch_xkcd_comics(self, ctx: Context, comic: Optional[str]) -> None:
"""
Getting an xkcd comic's information along with the image.
To get a random comic, don't type any number as an argument. To get the latest, enter 0.
"""
embed = Embed()
comic = comic or randint(1, self.latest_comic_info['num'])
if comic == "latest":
info = self.latest_comic_info
else:
async with self.bot.http_session.get(URL.format(comic)) as resp:
if resp.status == 200:
info = await resp.json()
else:
embed.description = f"{resp.status}: Could not retrieve xkcd comic #{comic}."
log.debug(f"Retrieving xkcd comic #{comic} failed with status code {resp.status}.")
await ctx.send(embed=embed)
return
embed.set_image(url=info["img"])
date = f"{info['year']}/{info['month']}/{info['day']}"
embed.set_footer(text=f"{date} - #{comic}, \'{info['safe_title']}\'")
await ctx.send(embed=embed)
def setup(bot: Bot) -> None:
"""Loading the XKCD cog."""
bot.add_cog(XKCD(bot))
|
