38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
import random
|
|
import urllib
|
|
import urllib2
|
|
import re
|
|
import json
|
|
|
|
from util import hook
|
|
|
|
@hook.command
|
|
def suggest(inp, inp_unstripped=''):
|
|
".suggest [#n] <phrase> -- gets a random/the nth suggested google search"
|
|
if not inp:
|
|
return suggest.__doc__
|
|
|
|
inp = inp_unstripped
|
|
m = re.match('^#(\d+) (.+)$', inp)
|
|
if m:
|
|
num, inp = m.groups()
|
|
num = int(num)
|
|
if num > 10:
|
|
return 'can only get first ten suggestions'
|
|
else:
|
|
num = 0
|
|
|
|
url = 'http://google.com/complete/search?q=' + urllib.quote(inp, safe='')
|
|
page = urllib2.urlopen(url).read()
|
|
page_json = page.split('(', 1)[1][:-1]
|
|
suggestions = json.loads(page_json)[1]
|
|
if not suggestions:
|
|
return 'no suggestions found'
|
|
if num:
|
|
if len(suggestions) + 1 <= num:
|
|
return 'only got %d suggestions' % len(suggestions)
|
|
out = suggestions[num - 1]
|
|
else:
|
|
out = random.choice(suggestions)
|
|
return '#%d: %s (%s)' % (int(out[2][0]) + 1, out[0], out[1])
|