opensourcegames/code/utils/osg_wikipedia.py
2021-01-14 11:35:12 +01:00

29 lines
645 B
Python

"""
Everything specific to search in Wikipedia.
Using https://github.com/goldsmith/Wikipedia
"""
import wikipedia
wikipedia.set_lang('en') # just in case that isn't so already
def search(search_term, results=3):
"""
:param search_term:
:param max_results:
:return:
"""
return wikipedia.search(search_term, results=results)
def pages(titles):
pages = []
for title in titles:
try:
page = wikipedia.page(title, auto_suggest=False)
except wikipedia.exceptions.DisambiguationError:
continue # here we silently eat the exception
pages.append(page)
return pages