aboutsummaryrefslogtreecommitdiff
path: root/modules/define.py
blob: 5455f35f10239f36011fea6500e5887dc88a4d7e (about) (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#--require-config wordnik-api-key

import time
from src import ModuleManager, utils

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(ModuleManager.BaseModule):
    _last_called = 0

    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"]}, json=True)

        return page

    @utils.hook("received.command.define")
    def define(self, event):
        """
        :help: Define a provided term
        :usage: <phrase>
        """
        if event["args"]:
            word = event["args"]
        else:
            word = event["target"].buffer.get(from_self=False)
        word = word.replace(" ", "+")

        page = self._get_definition(word)
        if page:
            if len(page.data):
                event["stdout"].write("%s: %s" % (page.data[0]["word"],
                    page.data[0]["text"]))
            else:
                event["stderr"].write("No definitions found")
        else:
            raise utils.EventsResultsError()

    @utils.hook("received.command.randomword")
    def random_word(self, event):
        """
        :help: Define a random word
        """
        if not self._last_called or (time.time()-self._last_called >=
                RANDOM_DELAY_SECONDS):
            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=True)
            if page and len(page.data):
                definition = self._get_definition(page.data["word"])
                if definition and len(definition.data):
                    definition = definition.data[0]
                else:
                    raise utils.EventError("Try again in a couple of seconds")

                event["stdout"].write("Random Word: %s - Definition: %s" % (
                    page.data["word"], definition["text"]))
            else:
                raise utils.EventsResultsError()
        else:
            event["stderr"].write("Try again in a couple of seconds")