modules/command_suggestions.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

#--depends-on commands

import difflib
from src import ModuleManager, utils

class Module(ModuleManager.BaseModule):
    def _all_command_hooks(self):
        return self.events.on("received.command").get_children()

    @utils.hook("unknown.command")
    def unknown_command(self, event):
        all_commands = self._all_command_hooks()
        match = difflib.get_close_matches(event["command"], all_commands,
            cutoff=0.7)
        if match:
            nickname = ""
            if event["is_channel"]:
                nickname = "%s: " % event["user"].nickname

            event["target"].send_message(
                "%sUnknown command. Did you mean %s%s?" % (
                nickname, event["command_prefix"], match[0]))