phenny/modules/ddg.py

#!/usr/bin/python2
"""
ddg.py - duck duck go module
author: mutantmonkey <mutantmonkey@gmail.com>
portions based on search.py by sean b palmer
"""

import random

from urllib import quote as urlquote
from urllib2 import urlopen, HTTPError
import lxml.html

import web

def search(query):
	uri = 'https://api.duckduckgo.com/'
	args = '?q=%s&o=json' % web.urllib.quote(query.encode('utf-8'))
	bytes = web.get(uri + args)
	return web.json(bytes)

def result(query):
	results = search(query)
	try:
		return results['Results'][0]['FirstURL']
	except IndexError:
		return None

def ddg(phenny, input):
	""".ddg <query> - Search Duck Duck Go for the specified query."""

	query = input.group(2)
	if not query:
		return phenny.reply(".ddg what?")

	uri = result(query)
	resultsuri = "https://duckduckgo.com/?q=" + web.urllib.quote(query.encode('utf-8'))
	if uri:
		phenny.reply("%s - Results from %s" % (uri, resultsuri))
	else:
		phenny.reply("No results found for '%s'; try %s" % (query, resultsuri))
ddg.rule = (['ddg'], r'(.*)')

if __name__ == '__main__':
	print __doc__.strip()