29 lines
1 KiB
Python
29 lines
1 KiB
Python
|
import Utils
|
||
|
|
|
||
|
|
URL_GOOGLESEARCH = "https://www.googleapis.com/customsearch/v1"
|
||
|
|
|
||
|
|
class Module(object):
|
||
|
|
def __init__(self, bot):
|
||
|
|
self.bot = bot
|
||
|
|
bot.events.on("received").on("command").on("google",
|
||
|
|
"g").hook(self.google, help="Google feeling lucky")
|
||
|
|
|
||
|
|
def google(self, event):
|
||
|
|
phrase = event["args"] or event["log"].get()
|
||
|
|
if phrase:
|
||
|
|
page = Utils.get_url(URL_GOOGLESEARCH, get_params={
|
||
|
|
"q": phrase, "key": self.bot.config[
|
||
|
|
"google-api-key"], "cx": self.bot.config[
|
||
|
|
"google-search-id"], "prettyPrint": "true",
|
||
|
|
"num": 1, "gl": "us"}, json=True)
|
||
|
|
if page:
|
||
|
|
if "items" in page and len(page["items"]):
|
||
|
|
event["stdout"].write(page["items"][0][
|
||
|
|
"link"])
|
||
|
|
else:
|
||
|
|
event["stderr"].write("No results found")
|
||
|
|
else:
|
||
|
|
event["stderr"].write("Failed to load results")
|
||
|
|
else:
|
||
|
|
event["stderr"].write("No phrase provided")
|