def __init__(self, event_queue, tts_queue, session, recognizer, recognition_timeout, failed_message, confirmation_messages):
super(Controller, self).__init__()
self.event_queue = event_queue
self.tts_queue = tts_queue
self.session = session
self.recognizer = recognizer
self.recognition_timeout = recognition_timeout
self.failed_action = Action(tts=failed_message)
self.confirmation_messages = confirmation_messages
self._stop = False
def actions():
if request.method == 'GET':
equipment = request.args.get('equipment')
search = request.args.get('search')
q = Action.query
if equipment is not None:
q = q.filter(Action.equipmentId == equipment)
if search is not None:
q = q.filter(Action.name.like(search + '%'))
actions = q.order_by(Action.name).all()
return actions_schema.jsonify(actions)
elif request.method == 'POST':
equipment = request.json['equipmentId']
name = request.json['name']
action = Action.query.filter(Action.equipmentId == equipment,
Action.name.like(name)).first()
if action is not None:
abort(400)
action = Action(equipmentId=equipment, name=name)
session.add(action)
session.commit()
return action_schema.jsonify(action)
def from_json_obj(json_obj):
packet = Packet(
action=Action.from_json_obj(
json_obj=json_obj["action"]),
message=Message.from_json_obj(
json_obj=json_obj["message"])
)
if json_obj["data"]:
packet.data = Data.from_json_obj(json_obj=json_obj["data"])
return packet
class Controller(threading.Thread):
"""Threaded controller that holds the main logic of Dobby. It grabs events as they come and put corresponding
(according to the database) processed actions in the queue.
Error message and confirmation messages are customizable
:param Queue.Queue event_queue: where to listen for events
:param Queue.Queue tts_queue: where to put the tts from processed actions
:param Session session: Dobby database session
:param integer recognition_timeout: time to wait for a :class:`~dobby.models.command.Command` to be recognized once a :class:`RecognitionEvent` is received
:param string failed_message: error message to say when the recognized :class:`~dobby.models.command.Command` does not match anything in the database
:param list confirmation_messages: a random message to say is picked and sent to the action queue whenever a :class:`RecognitionEvent` is caught
"""
def __init__(self, event_queue, tts_queue, session, recognizer, recognition_timeout, failed_message, confirmation_messages):
super(Controller, self).__init__()
self.event_queue = event_queue
self.tts_queue = tts_queue
self.session = session
self.recognizer = recognizer
self.recognition_timeout = recognition_timeout
self.failed_action = Action(tts=failed_message)
self.confirmation_messages = confirmation_messages
self._stop = False
def stop(self):
"""Stop the thread"""
self._stop = True
def run(self):
while not self._stop:
# Get the event from the queue
try:
event = self.event_queue.get(timeout=1)
except Queue.Empty:
continue
# Fire a Scenario directly if the event is an CommandEvent
if isinstance(event, CommandEvent):
logger.debug(u'CommandEvent caught with command "%s"' % event.command)
scenario = self.session.query(Scenario).join(Command).filter(Command.text == event.command).first()
# Launch recognition and analyze the first voice command
elif isinstance(event, RecognitionEvent):
# Send a random confirmation message
if self.confirmation_messages:
self.tts_queue.put(random.sample(self.confirmation_messages, 1)[0])
# Monitor the recognized voice commands and catch the first one
recognition_queue = Queue.Queue()
self.recognizer.subscribe(recognition_queue)
try:
recognition = recognition_queue.get(timeout=self.recognition_timeout)
recognized_command = unicode(recognition)
logger.debug(u'RecognitionEvent caught with voice command "%s"' % recognized_command)
scenario = self.session.query(Scenario).join(Command).filter(Command.text == recognized_command).first()
except Queue.Empty:
scenario = None
pass
self.recognizer.unsubscribe(recognition_queue)
# Failed message if no scenario
if not scenario:
logger.debug(u'Could not find the scenario in database')
if self.failed_action.tts:
self.tts_queue.put(self.failed_action.format_tts())
continue
# Loop over the scenario's actions in the correct order and execute them
for action_number in sorted(scenario.actions.keys()):
self.tts_queue.put(scenario.actions[action_number].format_tts())
# Mark the task as done
self.event_queue.task_done()
# Expire session
self.session.expire_all()
logger.info(u'Terminating...')