70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
#--require-config wordnik-api-key
|
|
|
|
import Utils
|
|
import time
|
|
|
|
URL_WORDNIK = "https://api.wordnik.com/v4/word.json/%s/definitions"
|
|
URL_WORDNIK_RANDOM = "https://api.wordnik.com/v4/words.json/randomWord"
|
|
|
|
RANDOM_DELAY_SECONDS = 3
|
|
|
|
class Module(object):
|
|
def __init__(self, bot, events, exports):
|
|
self.bot = bot
|
|
self.last_called = 0
|
|
self.events = events
|
|
|
|
events.on("received.command.define").hook(self.define,
|
|
help="Define a provided term", usage="<phrase>")
|
|
|
|
events.on("received.command.randomword").hook(self.random_word,
|
|
help="Generate a random word!")
|
|
|
|
def _get_definition(self, word):
|
|
word = event["args"] if "args" in event else event
|
|
|
|
page = Utils.get_url(URL_WORDNIK % word, get_params={
|
|
"useCanonical": "true", "limit": 1,
|
|
"sourceDictionaries": "wiktionary", "api_key": self.bot.config[
|
|
"wordnik-api-key"]}, json=True)
|
|
|
|
return page
|
|
|
|
def define(self, event):
|
|
if event["args"]:
|
|
word = event["args"]
|
|
else:
|
|
word = event["buffer"].get(from_self=False)
|
|
|
|
page = self._get_definition(word)
|
|
if page:
|
|
if len(page):
|
|
event["stdout"].write("%s: %s" % (page[0]["word"],
|
|
page[0]["text"]))
|
|
else:
|
|
event["stderr"].write("No definitions found")
|
|
else:
|
|
event["stderr"].write("Failed to load results")
|
|
|
|
def random_word(self, event):
|
|
if not self.last_called or (time.time()-self.last_called >=
|
|
RANDOM_DELAY_SECONDS):
|
|
self.last_called = time.time()
|
|
|
|
page = Utils.get_url(URL_WORDNIK_RANDOM, get_params={
|
|
"api_key":self.bot.config["wordnik-api-key"],
|
|
"min_dictionary_count":1},json=True)
|
|
if page and len(page):
|
|
definition = self._get_definition(page["word"])
|
|
if len(definition):
|
|
definition = definition[0]
|
|
else:
|
|
event["stderr"].write("Try again in a couple of "
|
|
"seconds")
|
|
return
|
|
event["stdout"].write("Random Word: %s - Definition: %s" % (
|
|
page["word"], definition["text"]))
|
|
else:
|
|
event["stderr"].write("Failed to load results")
|
|
else:
|
|
event["stderr"].write("Try again in a couple of seconds")
|