Add Pico communication service
This commit is contained in:
@@ -0,0 +1,48 @@
|
||||
from .protocol import (
|
||||
make_status_request,
|
||||
make_set_relay,
|
||||
is_status_response,
|
||||
is_relay_response,
|
||||
is_error,
|
||||
)
|
||||
|
||||
|
||||
class CommunicationService:
|
||||
def __init__(self, uart_client, app_state):
|
||||
self.uart_client = uart_client
|
||||
self.app_state = app_state
|
||||
self.last_messages = []
|
||||
|
||||
def request_status(self):
|
||||
self.uart_client.send_message(make_status_request())
|
||||
|
||||
def set_relay(self, relay, enabled):
|
||||
self.uart_client.send_message(make_set_relay(relay, enabled))
|
||||
|
||||
def poll(self):
|
||||
messages = self.uart_client.read_available_messages()
|
||||
self.last_messages = messages
|
||||
|
||||
for message in messages:
|
||||
self.handle_message(message)
|
||||
|
||||
return messages
|
||||
|
||||
def handle_message(self, message):
|
||||
if is_status_response(message):
|
||||
self.app_state.update_from_status(message)
|
||||
return
|
||||
|
||||
if is_relay_response(message):
|
||||
self.app_state.update_from_relay_response(message)
|
||||
return
|
||||
|
||||
if is_error(message):
|
||||
self.app_state.set_error(message)
|
||||
return
|
||||
|
||||
self.app_state.set_error({
|
||||
"type": "error",
|
||||
"message": "unknown_message_type",
|
||||
"raw": message,
|
||||
})
|
||||
Reference in New Issue
Block a user