Edm0nd/edmond/plugins/wikipedia.py

67 lines
2 KiB
Python

import time
try:
import wikipedia
DEPENDENCIES_FOUND = True
except ImportError:
DEPENDENCIES_FOUND = False
from edmond.plugin import Plugin
class WikipediaPlugin(Plugin):
REQUIRED_CONFIGS = [
"commands", "ambiguous_response", "empty_response", "lang",
]
NUM_RETRIES = 3
def __init__(self, bot):
super().__init__(bot)
if not self.is_ready:
return
wikipedia.set_lang(self.config["lang"])
def on_pubmsg(self, event):
if not self.should_handle_command(event.arguments[0]):
return False
if self.command.ident == "science":
self.tell_random_summary(event)
elif self.command.ident == "define":
self.tell_definition(event)
return True
def tell_random_summary(self, event):
summary = ""
retries = self.NUM_RETRIES
while retries > 0:
try:
summary = wikipedia.summary(wikipedia.random(), sentences=1)
break
except wikipedia.exceptions.WikipediaException:
pass
retries -= 1
if summary:
self.bot.say(event.target, summary)
def tell_definition(self, event):
summary = ""
retries = self.NUM_RETRIES
while retries > 0:
try:
summary = wikipedia.summary(self.command.content, sentences=1)
break
except wikipedia.exceptions.DisambiguationError:
summary = self.config["ambiguous_response"]
break
except wikipedia.exceptions.PageError:
summary = self.config["empty_response"]
break
except wikipedia.exceptions.WikipediaException:
summary = self.bot.config["error_message"]
# Keep trying after a slight delay.
time.sleep(1)
retries -= 1
if summary:
self.bot.say(event.target, summary)