Added alarm, timer and error event handlers on assistant backend

This commit is contained in:
Fabio Manganiello 2019-01-19 01:55:32 +01:00
parent 03bdf5c547
commit 2b38f1b03f
2 changed files with 85 additions and 2 deletions

View file

@ -11,13 +11,15 @@ import time
import google.oauth2.credentials import google.oauth2.credentials
from google.assistant.library import Assistant from google.assistant.library import Assistant
from google.assistant.library.event import EventType from google.assistant.library.event import EventType, AlertType
from google.assistant.library.file_helpers import existing_file from google.assistant.library.file_helpers import existing_file
from platypush.backend import Backend from platypush.backend import Backend
from platypush.message.event.assistant import \ from platypush.message.event.assistant import \
ConversationStartEvent, ConversationEndEvent, ConversationTimeoutEvent, \ ConversationStartEvent, ConversationEndEvent, ConversationTimeoutEvent, \
ResponseEvent, NoResponseEvent, SpeechRecognizedEvent ResponseEvent, NoResponseEvent, SpeechRecognizedEvent, AlarmStartedEvent, \
AlarmEndEvent, TimerStartedEvent, TimerEndEvent, AlertStartedEvent, \
AlertEndEvent
class AssistantGoogleBackend(Backend): class AssistantGoogleBackend(Backend):
@ -40,6 +42,14 @@ class AssistantGoogleBackend(Backend):
when a conversation times out when a conversation times out
* :class:`platypush.message.event.assistant.ConversationEndEvent` \ * :class:`platypush.message.event.assistant.ConversationEndEvent` \
when a new conversation ends when a new conversation ends
* :class:`platypush.message.event.assistant.AlarmStartedEvent` \
when an alarm starts
* :class:`platypush.message.event.assistant.AlarmEndEvent` \
when an alarm ends
* :class:`platypush.message.event.assistant.TimerStartedEvent` \
when a timer starts
* :class:`platypush.message.event.assistant.TimerEndEvent` \
when a timer ends
Requires: Requires:
@ -96,6 +106,25 @@ class AssistantGoogleBackend(Backend):
phrase = event.args['text'].lower().strip() phrase = event.args['text'].lower().strip()
self.logger.info('Speech recognized: {}'.format(phrase)) self.logger.info('Speech recognized: {}'.format(phrase))
self.bus.post(SpeechRecognizedEvent(phrase=phrase)) self.bus.post(SpeechRecognizedEvent(phrase=phrase))
elif event.type == EventType.ON_ALERT_STARTED:
if event.args.get('alert_type') == AlertType.ALARM:
self.bus.post(AlarmStartedEvent())
elif event.args.get('alert_type') == AlertType.TIMER:
self.bus.post(TimerStartedEvent())
else:
self.bus.post(AlertStartedEvent())
elif event.type == EventType.ON_ALERT_FINISHED:
if event.args.get('alert_type') == AlertType.ALARM:
self.bus.post(AlarmEndEvent())
elif event.args.get('alert_type') == AlertType.TIMER:
self.bus.post(TimerEndEvent())
else:
self.bus.post(AlertEndEvent())
elif event.type == EventType.ON_ASSISTANT_ERROR:
if event.args.get('is_fatal'):
self.logger.error('Fatal assistant error, restart the application')
else:
self.logger.warning('Assistant error')
def start_conversation(self): def start_conversation(self):

View file

@ -118,5 +118,59 @@ class HotwordDetectedEvent(AssistantEvent):
super().__init__(*args, hotword=hotword, **kwargs) super().__init__(*args, hotword=hotword, **kwargs)
class AlertStartedEvent(AssistantEvent):
"""
Event triggered when an alert starts on the assistant
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
class AlertEndEvent(AssistantEvent):
"""
Event triggered when an alert ends on the assistant
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
class AlarmStartedEvent(AlertStartedEvent):
"""
Event triggered when an alarm starts on the assistant
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
class AlarmEndEvent(AlertEndEvent):
"""
Event triggered when an alarm ends on the assistant
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
class TimerStartedEvent(AlertStartedEvent):
"""
Event triggered when a timer starts on the assistant
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
class TimerEndEvent(AlertEndEvent):
"""
Event triggered when a timer ends on the assistant
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
# vim:sw=4:ts=4:et: # vim:sw=4:ts=4:et: