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
69
70
71
72
73
74
75
|
# -*- coding: utf-8-*-
import logging
import os
import pkgutil
import importlib
import jasperpath
class Brain(object):
def __init__(self, mic, profile):
"""
Instantiates a new Brain object, which cross-references user
input with a list of modules. Note that the order of brain.modules
matters, as the Brain will cease execution on the first module
that accepts a given input.
Arguments:
mic -- used to interact with the user (for both input and output)
profile -- contains information related to the user (e.g., phone number)
"""
self.mic = mic
self.profile = profile
self.modules = self.get_modules()
self._logger = logging.getLogger(__name__)
@classmethod
def get_modules(cls):
"""
Dynamically loads all the modules in the modules folder and sorts
them by the PRIORITY key. If no PRIORITY is defined for a given
module, a priority of 0 is assumed.
"""
logger = logging.getLogger(__name__)
module_locations = [jasperpath.PLUGIN_PATH]
logger.debug("Looking for modules in: %s", ', '.join(["'%s'" % location for location in module_locations]))
modules = []
for finder, name, ispkg in pkgutil.walk_packages(module_locations):
try:
loader = finder.find_module(name)
mod = loader.load_module(name)
except:
logger.warning("Skipped module '%s' due to an error.", name, exc_info=True)
else:
if hasattr(mod, 'WORDS'):
logger.debug("Found module '%s' with words: %r", name, mod.WORDS)
modules.append(mod)
else:
logger.warning("Skipped module '%s' because it misses the WORDS constant.", name)
modules.sort(key=lambda mod: mod.PRIORITY if hasattr(mod, 'PRIORITY') else 0, reverse=True)
return modules
def query(self, texts):
"""
Passes user input to the appropriate module, testing it against
each candidate module's isValid function.
Arguments:
text -- user input, typically speech, to be parsed by a module
"""
for module in self.modules:
for text in texts:
if module.isValid(text):
self._logger.debug("'%s' is a valid phrase for module '%s'", text, module.__name__)
try:
module.handle(text, self.mic, self.profile)
except:
self._logger.error('Failed to execute module', exc_info=True)
self.mic.say("I'm sorry. I had some trouble with that operation. Please try again later.")
else:
self._logger.debug("Handling of phrase '%s' by module '%s' completed", text, module.__name__)
finally:
return
self._logger.debug("No module was able to handle any of these phrases: %r", texts)
|
