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
|
from html import unescape
from urllib.parse import quote_plus
from discord import Embed
from discord.ext.commands import Bot, Cog, Context, command, cooldown
BASE_URL = "https://api.stackexchange.com/2.2/search/advanced?order=desc&sort=activity&site=stackoverflow&q={query}"
SEARCH_URL = "https://stackoverflow.com/search?q={query}"
SO_COLOR = 0xF98036
class Stackoverflow(Cog):
"""A cog which returns the top 5 results of a query from stackoverflow."""
def __init__(self, bot: Bot):
self.bot = bot
@command(name="stackoverflow", aliases=["so"])
@cooldown(1, 15)
async def stackoverflow(self, ctx: Context, *, search_query: str) -> None:
"""Sends the top 5 results from stackoverflow based on a search query."""
async with self.bot.http_session.get(BASE_URL.format(query=quote_plus(search_query))) as response:
data = await response.json()
top5 = data["items"][:5]
embed = Embed(title=f"Search results for {search_query!r} - Stackoverflow",
url=SEARCH_URL.format(query=quote_plus(search_query)),
description=f"Here are the top {len(top5)} results:",
color=SO_COLOR)
for item in top5:
embed.add_field(
name=f"{unescape(item['title'])}",
value=(f"{item['score']} upvotes ┃ "
f"{item['view_count']} views ┃ "
f"{item['answer_count']} answers "
),
inline=False)
embed.set_footer()
await ctx.send(embed=embed)
def setup(bot: Bot) -> None:
"""Adds the cog to the bot."""
bot.add_cog(Stackoverflow(bot))
|
