Add .randomword command, and change the api to use https

This commit is contained in:
dngfx 2018-09-23 09:52:53 +01:00
parent 0cb2c43ee3
commit fdcf3e45c7

View file

@ -1,24 +1,42 @@
#--require-config wordnik-api-key #--require-config wordnik-api-key
import Utils import Utils
import time
URL_WORDNIK = "http://api.wordnik.com:80/v4/word.json/%s/definitions" 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): class Module(object):
def __init__(self, bot, events, exports): def __init__(self, bot, events, exports):
self.bot = bot self.bot = bot
self.last_called = 0
self.events = events
events.on("received.command.define").hook(self.define, events.on("received.command.define").hook(self.define,
help="Define a provided term", usage="<phrase>") 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, event):
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): def define(self, event):
if event["args"]: if event["args"]:
word = event["args"] word = event["args"]
else: else:
word = event["buffer"].get(from_self=False) word = event["buffer"].get(from_self=False)
page = Utils.get_url(URL_WORDNIK % event["args"], get_params={ page = self.get_definition(event)
"useCanonical": "true", "limit": 1,
"sourceDictionaries": "wiktionary", "api_key": self.bot.config[
"wordnik-api-key"]}, json=True)
if page: if page:
if len(page): if len(page):
event["stdout"].write("%s: %s" % (page[0]["word"], event["stdout"].write("%s: %s" % (page[0]["word"],
@ -27,3 +45,36 @@ class Module(object):
event["stderr"].write("No definitions found") event["stderr"].write("No definitions found")
else: else:
event["stderr"].write("Failed to load results") 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:
if len(page):
definition = self.get_definition(page["word"])
if len(definition):
definition = definition[0]
else:
self.events.on("send.stderr").call(module_name="Random",
target=event["target"],
message="Try again in a couple of seconds")
return
event["stdout"].set_prefix("Random")
event["stdout"].write("Random Word: %s - Definition: %s" % (
page["word"], definition["text"]))
else:
event["stderr"].write("Something has gone terribly wrong")
else:
event["stderr"].write("Failed to load results")
else:
self.events.on("send.stderr").call(module_name="Random",
target=event["target"],
message="Try again in a couple of seconds")