mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-10 21:50:59 -05:00
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
# coding=utf8
|
|
from __future__ import unicode_literals, absolute_import, division, print_function
|
|
"""A way to search google"""
|
|
|
|
import requests
|
|
from fake_useragent import UserAgent
|
|
|
|
# TODO add a cache flush
|
|
|
|
|
|
class Google():
|
|
|
|
def __init__(self):
|
|
self.cache = {
|
|
"info": {},
|
|
"maps": {}
|
|
}
|
|
|
|
def search(self, searchterm, searchtype="info"):
|
|
|
|
# losercase searching
|
|
searchterm = searchterm.lower()
|
|
|
|
# check cache
|
|
if searchtype not in list(self.cache.keys()):
|
|
self.cache[searchtype] = dict()
|
|
if searchterm in self.cache[searchtype]:
|
|
return self.cache[searchtype][searchterm]
|
|
|
|
header = {'User-Agent': str(UserAgent().chrome)}
|
|
data = searchterm.replace(' ', '+')
|
|
lookfor = data.replace(':', '%3A')
|
|
try:
|
|
if searchtype == 'maps':
|
|
var = requests.get(r'http://www.google.com/maps/place/' + lookfor, headers=header)
|
|
elif searchtype == 'info':
|
|
var = requests.get(r'http://www.google.com/search?q=' + lookfor + '&btnI', headers=header)
|
|
else:
|
|
var = requests.get(r'http://www.google.com/search?q=' + lookfor + '&btnI', headers=header)
|
|
except Exception as e:
|
|
var = e
|
|
var = None
|
|
|
|
if not var or not var.url:
|
|
return None
|
|
else:
|
|
self.cache[searchtype][searchterm] = str(var.url)
|
|
return var.url
|
|
|
|
|
|
google = Google()
|