import json import random import re from util import hook, http @hook.command def suggest(inp, inp_unstripped=''): ".suggest [#n] -- 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 page = http.get('http://google.com/complete/search', q=inp) 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])