bitbot-3.11-fork/modules/define.py

67 lines
2.3 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")
@utils.kwarg("help", "Define a provided term")
@utils.spec("!<term>lstring")
2016-03-29 11:56:58 +00:00
def define(self, event):
word = event["spec"][0].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")
@utils.kwarg("help", "Define a random word")
def random_word(self, event):
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")