104 lines
2.8 KiB
Python
104 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
"""Controls the LEDs on the ReSpeaker Mic Array v2.0 (USB) ."""
|
|
import argparse
|
|
import asyncio
|
|
import logging
|
|
import time
|
|
from functools import partial
|
|
|
|
from wyoming.event import Event
|
|
from wyoming.satellite import (
|
|
SatelliteConnected,
|
|
SatelliteDisconnected,
|
|
StreamingStarted,
|
|
StreamingStopped,
|
|
)
|
|
from wyoming.snd import Played
|
|
from wyoming.server import AsyncEventHandler, AsyncServer
|
|
from wyoming.vad import VoiceStarted, VoiceStopped
|
|
from wyoming.wake import Detection
|
|
|
|
from pixel_ring import pixel_ring
|
|
|
|
_LOGGER = logging.getLogger()
|
|
|
|
async def main() -> None:
|
|
"""Main entry point."""
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--uri", required=True, help="unix:// or tcp://")
|
|
parser.add_argument("--debug", action="store_true", help="Log DEBUG messages")
|
|
args = parser.parse_args()
|
|
|
|
logging.basicConfig(level=logging.DEBUG if args.debug else logging.INFO)
|
|
_LOGGER.debug(args)
|
|
|
|
_LOGGER.info("Ready")
|
|
|
|
# Turn on power to LEDs
|
|
pixel_ring.set_color_palette(0x0080ff, 0x007a37)
|
|
pixel_ring.think()
|
|
await asyncio.sleep(3)
|
|
pixel_ring.off()
|
|
|
|
# Start server
|
|
server = AsyncServer.from_uri(args.uri)
|
|
|
|
try:
|
|
await server.run(partial(LEDsEventHandler, args))
|
|
except KeyboardInterrupt:
|
|
pass
|
|
finally:
|
|
pixel_ring.off()
|
|
|
|
|
|
class LEDsEventHandler(AsyncEventHandler):
|
|
"""Event handler for clients."""
|
|
|
|
def __init__(
|
|
self,
|
|
cli_args: argparse.Namespace,
|
|
*args,
|
|
**kwargs,
|
|
) -> None:
|
|
super().__init__(*args, **kwargs)
|
|
|
|
self.cli_args = cli_args
|
|
self.client_id = str(time.monotonic_ns())
|
|
|
|
_LOGGER.debug("Client connected: %s", self.client_id)
|
|
|
|
async def handle_event(self, event: Event) -> bool:
|
|
_LOGGER.debug(event)
|
|
|
|
if Detection.is_type(event.type):
|
|
_LOGGER.debug("Detection")
|
|
pixel_ring.wakeup()
|
|
elif VoiceStarted.is_type(event.type):
|
|
_LOGGER.debug("VoiceStarted")
|
|
pixel_ring.speak()
|
|
elif VoiceStopped.is_type(event.type):
|
|
_LOGGER.debug("VoiceStopped")
|
|
pixel_ring.spin()
|
|
elif StreamingStopped.is_type(event.type):
|
|
_LOGGER.debug("StreamingStopped")
|
|
pixel_ring.off()
|
|
elif SatelliteConnected.is_type(event.type):
|
|
_LOGGER.debug("SatelliteConnected")
|
|
pixel_ring.think()
|
|
await asyncio.sleep(2)
|
|
pixel_ring.off()
|
|
elif Played.is_type(event.type):
|
|
_LOGGER.debug("Played")
|
|
pixel_ring.off()
|
|
elif SatelliteDisconnected.is_type(event.type):
|
|
_LOGGER.debug("SatelliteDisconnected")
|
|
pixel_ring.mono(0xff0000)
|
|
|
|
return True
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
asyncio.run(main())
|
|
except KeyboardInterrupt:
|
|
pass
|