91 lines
3.2 KiB
Python
91 lines
3.2 KiB
Python
from __future__ import annotations
|
||
|
||
import queue
|
||
import subprocess
|
||
import sys
|
||
import threading
|
||
from typing import Any
|
||
|
||
from loguru import logger
|
||
|
||
class RepAnnouncer:
|
||
"""运动次数语音播报器"""
|
||
|
||
def __init__(self, *, enabled: bool = True, rate: int = 185, volume: float = 1.0) -> None:
|
||
"""初始化TTS引擎（macOS用say，其他系统用pyttsx3）"""
|
||
self.enabled = enabled
|
||
self.rate = rate
|
||
self.volume = volume
|
||
self._queue: queue.Queue[str | None] = queue.Queue()
|
||
self._thread: threading.Thread | None = None
|
||
self._engine: Any | None = None
|
||
self._use_macos_say = False
|
||
self._current_process: subprocess.Popen | None = None
|
||
|
||
if self.enabled:
|
||
self._start()
|
||
|
||
def announce_count(self, count: int) -> None:
|
||
"""将次数放入队列进行异步语音播报"""
|
||
if not self.enabled or count <= 0:
|
||
return
|
||
while True:
|
||
try:
|
||
self._queue.get_nowait()
|
||
except queue.Empty:
|
||
break
|
||
self._queue.put(str(count))
|
||
|
||
def close(self) -> None:
|
||
"""停止播报线程并释放资源"""
|
||
if not self.enabled:
|
||
return
|
||
self._queue.put(None)
|
||
if self._thread is not None:
|
||
self._thread.join(timeout=1.0)
|
||
if self._current_process is not None and self._current_process.poll() is None:
|
||
self._current_process.terminate()
|
||
|
||
def _start(self) -> None:
|
||
"""根据平台初始化TTS引擎并启动后台播报线程"""
|
||
if sys.platform == "darwin":
|
||
self._use_macos_say = True
|
||
logger.info("Rep announcer initialized with macOS say")
|
||
else:
|
||
try:
|
||
import pyttsx3
|
||
|
||
self._engine = pyttsx3.init()
|
||
self._engine.setProperty("rate", self.rate)
|
||
self._engine.setProperty("volume", self.volume)
|
||
logger.info("Rep announcer initialized with pyttsx3")
|
||
except Exception as exc:
|
||
self.enabled = False
|
||
logger.warning("Rep announcer disabled, pyttsx3 unavailable: {}", exc)
|
||
return
|
||
|
||
self._thread = threading.Thread(target=self._run, name="RepAnnouncer", daemon=True)
|
||
self._thread.start()
|
||
|
||
def _run(self) -> None:
|
||
"""后台线程：从队列读取文本并调用TTS播放"""
|
||
while True:
|
||
text = self._queue.get()
|
||
if text is None:
|
||
return
|
||
|
||
try:
|
||
if self._use_macos_say:
|
||
if self._current_process is not None and self._current_process.poll() is None:
|
||
self._current_process.terminate()
|
||
self._current_process = subprocess.Popen(
|
||
["say", "-r", str(self.rate), text],
|
||
stdout=subprocess.DEVNULL,
|
||
stderr=subprocess.DEVNULL,
|
||
)
|
||
elif self._engine is not None:
|
||
self._engine.say(text)
|
||
self._engine.runAndWait()
|
||
except Exception as exc:
|
||
logger.warning("Failed to announce rep count {}: {}", text, exc)
|