h/plugins/google.py

import random

from util import hook, http


def api_get(query, key, is_image=None, num=1):
    url = ('https://www.googleapis.com/customsearch/v1?cx=007629729846476161907:ud5nlxktgcw'
            '&fields=items(title,link,snippet)&safe=off' + ('&searchType=image' if is_image else ''))
    return http.get_json(url, key=key, q=query, num=num)


@hook.api_key('google')
@hook.command
def gis(inp, api_key=None):
    '''.gis <term> -- finds an image using google images (safesearch off)'''

    parsed = api_get(inp, api_key, is_image=True, num=10)
    if 'items' not in parsed:
        return 'no images found'
    return random.choice(parsed['items'])['link']


@hook.api_key('google')
@hook.command('g')
@hook.command
def google(inp, api_key=None):
    '''.g/.google <query> -- returns first google search result'''

    parsed = api_get(inp, api_key)
    if 'items' not in parsed:
        return 'no results found'

    out = u'{link} -- \x02{title}\x02: "{snippet}"'.format(**parsed['items'][0])
    out = ' '.join(out.split())

    if len(out) > 300:
        out = out[:out.rfind(' ')] + '..."'

    return out