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
84
|
import asyncio
import datetime
import logging
from typing import Optional
from discord import Color, Embed, Member
from discord.ext import commands
log = logging.getLogger(__name__)
SEARCH_API = "https://en.wikipedia.org/w/api.php?action=query&list=search&srsearch={search_term}&format=json"
WIKIPEDIA_URL = "https://en.wikipedia.org/wiki/{title}"
class WikipediaCog(commands.Cog):
"""Get info from wikipedia."""
def __init__(self, bot: commands.Bot):
self.bot = bot
self.http_session = bot.http_session
async def search_wikipedia(self, search_term: str) -> Optional[str]:
"""Search wikipedia and return the first page found."""
async with self.http_session.get(SEARCH_API.format(search_term=search_term)) as response:
data = await response.json()
page = []
search_result = data["query"]["search"]
if len(search_result) == 0:
return None
# we dont like "may refere to" pages.
for a in search_result:
if "may refer to" in a["snippet"]:
pass
else:
page.append(a["title"])
log.info("appening titles")
return page
@commands.command(name="wikipedia", aliases=["wiki"])
async def w_pedia(self, ctx: commands.Context, *, search: str) -> None:
"""Gives list of item."""
final = []
s = ''
title = await self.search_wikipedia(search)
def check(user: Member) -> bool:
return user.author.id == ctx.author.id
if title is None:
await ctx.send("Sorry, we could not find a wikipedia article using that search term")
return
for i in title:
t = i.replace(" ", "_") # wikipedia uses "_" as spaces
final.append(t)
async with ctx.typing():
for index, (a, j) in enumerate(zip(title, final), start=1):
s = s + f'`{index}` [{a}]({WIKIPEDIA_URL.format(title=j)})\n'
embed = Embed(colour=Color.blue(), title=f"Wikipedia results for `{search}`", description=s)
embed.timestamp = datetime.datetime.utcnow()
await ctx.send(embed=embed)
embed = Embed(colour=Color.green(), description="Enter number to choose")
msg = await ctx.send(embed=embed)
try:
user = await ctx.bot.wait_for('message', timeout=60.0, check=check)
await ctx.send(WIKIPEDIA_URL.format(title=final[int(user.content) - 1]))
except asyncio.TimeoutError:
embed = Embed(colour=Color.red(), description=f"Time's up {ctx.author.mention}")
await msg.edit(embed=embed)
except ValueError:
await ctx.send("I am sorry but that isn't int value")
def setup(bot: commands.Bot) -> None:
"""Uptime Cog load."""
bot.add_cog(WikipediaCog(bot))
|
