mirror of https://github.com/stijndcl/didier
33 lines
914 B
Python
33 lines
914 B
Python
|
from requests import get
|
||
|
|
from urllib.parse import urlencode
|
||
|
|
from bs4 import BeautifulSoup
|
||
|
|
|
||
|
|
# TODO add Football requests in here as well
|
||
|
|
|
||
|
|
|
||
|
|
def google_search(query):
|
||
|
|
"""
|
||
|
|
Function to get Google search results
|
||
|
|
"""
|
||
|
|
headers = {
|
||
|
|
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/74.0.3729.169 Safari/537.36'
|
||
|
|
}
|
||
|
|
|
||
|
|
query = urlencode({"q": query})
|
||
|
|
|
||
|
|
resp = get("https://www.google.com/search?{}&num=10&hl=en".format(query), headers=headers)
|
||
|
|
print("got here")
|
||
|
|
if resp.status_code != 200:
|
||
|
|
return None, resp.status_code
|
||
|
|
|
||
|
|
bs = BeautifulSoup(resp.text, "html.parser")
|
||
|
|
|
||
|
|
def getContent(element):
|
||
|
|
link = element.find('a', href=True)
|
||
|
|
title = element.find('h3')
|
||
|
|
return link, title
|
||
|
|
|
||
|
|
divs = bs.find_all('div', attrs={'class': 'g'})
|
||
|
|
print(divs)
|
||
|
|
return list(getContent(d) for d in divs), 200
|