73 lines
2.1 KiB
Python
Executable File
73 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
search.py - Phenny Web Search Module
|
|
Copyright 2008-9, Sean B. Palmer, inamidst.com
|
|
Modified by sfan5 2012
|
|
Licensed under the Eiffel Forum License 2.
|
|
|
|
http://inamidst.com/phenny/
|
|
"""
|
|
|
|
import web
|
|
import re
|
|
|
|
def google_ajax(query):
|
|
"""Search using AjaxSearch, and return its JSON."""
|
|
uri = 'http://ajax.googleapis.com/ajax/services/search/web'
|
|
args = '?v=1.0&safe=off&q=' + web.urlencode(query)
|
|
data, sc = web.get(uri + args)
|
|
data = str(data, 'utf-8')
|
|
return web.json(data)
|
|
|
|
def google_search(query):
|
|
results = google_ajax(query)
|
|
try: return results['responseData']['results'][0]['unescapedUrl']
|
|
except IndexError: return None
|
|
except TypeError:
|
|
return False
|
|
|
|
def g(phenny, input):
|
|
"""Queries Google for the specified input."""
|
|
query = input.group(2)
|
|
if not query:
|
|
return phenny.reply('.g what?')
|
|
log.log("event", "%s searched Google for '%s'" % (log.fmt_user(input), query), phenny)
|
|
uri = google_search(query)
|
|
if uri:
|
|
phenny.reply(uri)
|
|
phenny.bot.last_seen_uri = uri
|
|
elif uri is False: phenny.reply("Problem getting data from Google.")
|
|
else: phenny.reply("No results found for '%s'." % query)
|
|
g.commands = ['g']
|
|
g.priority = 'high'
|
|
g.example = '.g minetest'
|
|
|
|
def gc(phenny, input):
|
|
query = input.group(2)
|
|
if not query:
|
|
return phenny.reply("No query term.")
|
|
log.log("event", "%s searched Google for '%s'" % (log.fmt_user(input), query), phenny)
|
|
result = new_gc(query)
|
|
if result:
|
|
phenny.say(query + ": " + result)
|
|
else: phenny.reply("Sorry, couldn't get a result.")
|
|
|
|
def new_gc(query):
|
|
uri = 'https://www.google.com/search?hl=en&q='
|
|
uri = uri + web.urlencode(query).replace('+', '%2B')
|
|
if '"' in query: uri += '&tbs=li:1'
|
|
data, sc = web.get(uri)
|
|
data = str(data, 'utf-8')
|
|
if "did not match any documents" in data:
|
|
return "0"
|
|
for result in re.compile(r'(?ims)([0-9,]+) results?').findall(data):
|
|
return result
|
|
return None
|
|
|
|
gc.commands = ['gc']
|
|
gc.priority = 'high'
|
|
gc.example = '.gc minetest'
|
|
|
|
if __name__ == '__main__':
|
|
print(__doc__.strip())
|