import urllib from lxml import html from util import hook, yaml def api_get(kind, query): req_url = 'http://ajax.googleapis.com/ajax/services/search/%s?' \ 'v=1.0&safe=off&q=%s' url = req_url % (kind, urllib.quote(query, safe='')) json = urllib.urlopen(url).read() return yaml.load(json) @hook.command def gis(inp): '''.gis -- returns first google image result (safesearch off)''' if not inp: return gis.__doc__ parsed = api_get('images', inp) if not 200 <= parsed['responseStatus'] < 300: raise IOError('error searching for images: %d: %s' % ( parsed['responseStatus'], '')) if not parsed['responseData']['results']: return 'no images found' return parsed['responseData']['results'][0]['unescapedUrl'] @hook.command @hook.command('g') @hook.command('goog') def google(inp): '''.g/.goog/.google -- returns first google search result''' if not inp: return google.__doc__ parsed = api_get('web', inp) if not 200 <= parsed['responseStatus'] < 300: raise IOError('error searching for pages: %d: %s' % ( parsed['responseStatus'], '')) if not parsed['responseData']['results']: return 'no results found' result = parsed['responseData']['results'][0] title, content = map(lambda x: html.fromstring(x).text_content(), (result['titleNoFormatting'], result['content'])) out = '%s -- \x02%s\x02: "%s"' % (result['unescapedUrl'], title, content) out = ' '.join(out.split()) if len(out) > 300: out = out[:out.rfind(' ')] + '..."' return out