wikify.py 文件源码

def wikify3(phrase, description=None):
    ret = {}
    urlBase = "https://en.wikipedia.org/w/api.php?action=opensearch&format=json&formatversion=2&search={}&namespace=0&limit=10&suggest=true"
    url = urlBase.format(re.sub("\s+","%20",phrase))
    r = requests.get(url)
    jsonData = json.loads(r.content)
    #logging.info(items)
    ret = {}
    for idx, label in enumerate(jsonData[1][0:MAX_RESULT]):
        description = jsonData[2][idx]
        url = jsonData[3][idx]
        #if "refer to:" in description:
        #    continue

        if idx > 0:
            prefix = "wikipedia{}".format(idx+1)
        else:
            prefix = "wikipedia"
        ret["{}Label".format(prefix)] = label
        ret["{}Description".format(prefix)] = description
        ret["{}Url".format(prefix)] = url
    return ret