aboutsummaryrefslogtreecommitdiffstats
path: root/cogs/web.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--cogs/web.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/cogs/web.py b/cogs/web.py
index bea684b..9431e84 100644
--- a/cogs/web.py
+++ b/cogs/web.py
@@ -5,6 +5,9 @@ import requests
5from bs4 import BeautifulSoup 5from bs4 import BeautifulSoup
6import urllib.parse 6import urllib.parse
7import pytube 7import pytube
8import logging
9
10logger = logging.getLogger(__name__)
8 11
9class Web(commands.Cog): 12class Web(commands.Cog):
10 def __init__(self, bot): 13 def __init__(self, bot):
@@ -24,6 +27,8 @@ class Web(commands.Cog):
24 url = f"https://duckduckgo.com/html/?q={query}" 27 url = f"https://duckduckgo.com/html/?q={query}"
25 response = requests.get(url, headers=headers) 28 response = requests.get(url, headers=headers)
26 29
30 logger.info(f"Query: {query}")
31
27 if response.status_code == 200: 32 if response.status_code == 200:
28 soup = BeautifulSoup(response.text, 'html.parser') 33 soup = BeautifulSoup(response.text, 'html.parser')
29 links = soup.find_all('a', class_='result__a') 34 links = soup.find_all('a', class_='result__a')
@@ -33,6 +38,7 @@ class Web(commands.Cog):
33 for i, link in enumerate(links[:5], start=1): 38 for i, link in enumerate(links[:5], start=1):
34 title = link.text 39 title = link.text
35 if link.get('href').startswith('https://www.youtube.com/'): 40 if link.get('href').startswith('https://www.youtube.com/'):
41 logger.info("youtube link")
36 try: 42 try:
37 parsed_url = urllib.parse.urlparse(link.get('href')) 43 parsed_url = urllib.parse.urlparse(link.get('href'))
38 video_id = urllib.parse.parse_qs(parsed_url.query).get('v', [None])[0] 44 video_id = urllib.parse.parse_qs(parsed_url.query).get('v', [None])[0]
@@ -48,10 +54,13 @@ class Web(commands.Cog):
48 button.callback = button_callback 54 button.callback = button_callback
49 view.add_item(button) 55 view.add_item(button)
50 await ctx.respond(embed=embed, view=view, ephemeral=True) 56 await ctx.respond(embed=embed, view=view, ephemeral=True)
57 logger.info("Search results sent")
51 else: 58 else:
52 await ctx.respond(content="No results found.", ephemeral=True) 59 await ctx.respond(content="No results found.", ephemeral=True)
60 logger.warning("No results found")
53 else: 61 else:
54 await ctx.respond(content=f"Error: {response.status_code}", ephemeral=True) 62 await ctx.respond(content=f"Error: {response.status_code}", ephemeral=True)
63 logger.error(f"Error: {response.status_code}")
55 64
56def setup(bot): 65def setup(bot):
57 bot.add_cog(Web(bot)) 66 bot.add_cog(Web(bot))