core/homeassistant/components/tts/helper.py

24 lines
587 B
Python

"""Provide helper functions for the TTS."""
from __future__ import annotations
from typing import TYPE_CHECKING
from homeassistant.core import HomeAssistant
from .const import DATA_COMPONENT, DATA_TTS_MANAGER
if TYPE_CHECKING:
from . import TextToSpeechEntity
from .legacy import Provider
def get_engine_instance(
hass: HomeAssistant, engine: str
) -> TextToSpeechEntity | Provider | None:
"""Get engine instance."""
if entity := hass.data[DATA_COMPONENT].get_entity(engine):
return entity
return hass.data[DATA_TTS_MANAGER].providers.get(engine)