import duckduckgo
import urllib
from bs4 import BeautifulSoup

def xkcd(query):
    res = duckduckgo.get_zci('site:xkcd.com ' + query);
    title = "";
    try: #ddg returns a url for these searches. i want a title as well
        title = BeautifulSoup(urllib.urlopen(res).read(), 'html.parser').title.text
    except:
        pass #just swallow the error. maybe the result wasn't a url or something else bad happened
    return [(((title + ' - ') if title else '') + res).encode('ascii', 'ignore')]

# never mind, blocked by ddg
#def xkcd_links(query):
#    url = "https://duckduckgo.com/html/?q=site%3Axkcd.com+" + query.replace(' ', '+')
#    soup = BeautifulSoup(urllib.urlopen(url).read(), 'html.parser')
#    items = soup.find_all("a", class_="result__a")
#    print items
#    items = filter(lambda i: i[0:8] == 'xkcd.com', [i.find(class_="result__title").text.strip() for i in items])
#    print items
#    def pretty_link(item):
#         url = item.find(class_="result__url").text.strip()
#         title = item.find(class_="result__title").text.strip()
#         return (title + ' - ' + url) if title else url
#
#    links = map(lambda url: pretty_link(url), items)
#    return links