bitbot-3.11-fork/modules/define.py

75 lines
2.5 KiB
Python
Raw Normal View History

#--depends-on commands
#--require-config wordnik-api-key
import time
from src import ModuleManager, utils
2016-03-29 11:56:58 +00:00
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
2016-03-29 11:56:58 +00:00
class Module(ModuleManager.BaseModule):
_last_called = 0
2018-09-24 12:23:59 +00:00
def _get_definition(self, word):
page = utils.http.request(URL_WORDNIK % word, get_params={
"useCanonical": "true", "limit": 1,
"sourceDictionaries": "wiktionary",
"api_key": self.bot.config["wordnik-api-key"]})
if page:
if page.code == 200:
return True, page.json()[0]
else:
return True, None
else:
return False, None
@utils.hook("received.command.define")
2016-03-29 11:56:58 +00:00
def define(self, event):
"""
:help: Define a provided term
:usage: <phrase>
"""
2016-03-29 11:56:58 +00:00
if event["args"]:
word = event["args"]
else:
word = event["target"].buffer.get(from_self=False)
word = word.replace(" ", "+")
2018-09-24 12:23:59 +00:00
success, definition = self._get_definition(word)
if success:
if not definition == None:
text = utils.http.strip_html(definition["text"])
event["stdout"].write("%s: %s" % (definition["word"], text))
2016-03-29 11:56:58 +00:00
else:
event["stderr"].write("No definitions found")
else:
raise utils.EventResultsError()
@utils.hook("received.command.randomword")
def random_word(self, event):
"""
:help: Define a random word
"""
2018-09-24 12:24:36 +00:00
if not self._last_called or (time.time()-self._last_called >=
2018-09-24 12:23:59 +00:00
RANDOM_DELAY_SECONDS):
2018-09-24 12:24:36 +00:00
self._last_called = time.time()
page = utils.http.request(URL_WORDNIK_RANDOM, get_params={
"api_key": self.bot.config["wordnik-api-key"],
"min_dictionary_count": 1}).json()
if page:
success, definition = self._get_definition(page["word"])
if not success:
raise utils.EventError("Try again in a couple of seconds")
text = utils.http.strip_html(definition["text"])
2018-09-24 12:23:59 +00:00
event["stdout"].write("Random Word: %s - Definition: %s" % (
page["word"], text))
else:
raise utils.EventResultsError()
else:
2018-09-24 12:23:59 +00:00
event["stderr"].write("Try again in a couple of seconds")