118 lines
3.6 KiB
Python
118 lines
3.6 KiB
Python
|
"""
|
||
|
|
Support for the google speech service.
|
||
|
|
|
||
|
|
For more details about this component, please refer to the documentation at
|
||
|
|
https://home-assistant.io/components/tts/google/
|
||
|
|
"""
|
||
|
|
import asyncio
|
||
|
|
import logging
|
||
|
|
import re
|
||
|
|
|
||
|
|
import aiohttp
|
||
|
|
import async_timeout
|
||
|
|
import yarl
|
||
|
|
|
||
|
|
from homeassistant.components.tts import Provider
|
||
|
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
||
|
|
|
||
|
|
REQUIREMENTS = ["gTTS-token==1.1.1"]
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
GOOGLE_SPEECH_URL = "http://translate.google.com/translate_tts"
|
||
|
|
MESSAGE_SIZE = 148
|
||
|
|
|
||
|
|
|
||
|
|
@asyncio.coroutine
|
||
|
|
def async_get_engine(hass, config):
|
||
|
|
"""Setup Google speech component."""
|
||
|
|
return GoogleProvider(hass)
|
||
|
|
|
||
|
|
|
||
|
|
class GoogleProvider(Provider):
|
||
|
|
"""Google speech api provider."""
|
||
|
|
|
||
|
|
def __init__(self, hass):
|
||
|
|
"""Init Google TTS service."""
|
||
|
|
self.hass = hass
|
||
|
|
self.headers = {
|
||
|
|
'Referer': "http://translate.google.com/",
|
||
|
|
'User-Agent': ("Mozilla/5.0 (Windows NT 10.0; WOW64) "
|
||
|
|
"AppleWebKit/537.36 (KHTML, like Gecko) "
|
||
|
|
"Chrome/47.0.2526.106 Safari/537.36")
|
||
|
|
}
|
||
|
|
|
||
|
|
@asyncio.coroutine
|
||
|
|
def async_get_tts_audio(self, message):
|
||
|
|
"""Load TTS from google."""
|
||
|
|
from gtts_token import gtts_token
|
||
|
|
|
||
|
|
token = gtts_token.Token()
|
||
|
|
websession = async_get_clientsession(self.hass)
|
||
|
|
message_parts = self._split_message_to_parts(message)
|
||
|
|
|
||
|
|
data = b''
|
||
|
|
for idx, part in enumerate(message_parts):
|
||
|
|
part_token = yield from self.hass.loop.run_in_executor(
|
||
|
|
None, token.calculate_token, part)
|
||
|
|
|
||
|
|
url_param = {
|
||
|
|
'ie': 'UTF-8',
|
||
|
|
'tl': self.language,
|
||
|
|
'q': yarl.quote(part),
|
||
|
|
'tk': part_token,
|
||
|
|
'total': len(message_parts),
|
||
|
|
'idx': idx,
|
||
|
|
'client': 'tw-ob',
|
||
|
|
'textlen': len(part),
|
||
|
|
}
|
||
|
|
|
||
|
|
request = None
|
||
|
|
try:
|
||
|
|
with async_timeout.timeout(10, loop=self.hass.loop):
|
||
|
|
request = yield from websession.get(
|
||
|
|
GOOGLE_SPEECH_URL, params=url_param,
|
||
|
|
headers=self.headers
|
||
|
|
)
|
||
|
|
|
||
|
|
if request.status != 200:
|
||
|
|
_LOGGER.error("Error %d on load url %s", request.code,
|
||
|
|
request.url)
|
||
|
|
return (None, None)
|
||
|
|
data += yield from request.read()
|
||
|
|
|
||
|
|
except (asyncio.TimeoutError, aiohttp.errors.ClientError):
|
||
|
|
_LOGGER.error("Timeout for google speech.")
|
||
|
|
return (None, None)
|
||
|
|
|
||
|
|
finally:
|
||
|
|
if request is not None:
|
||
|
|
yield from request.release()
|
||
|
|
|
||
|
|
return ("mp3", data)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _split_message_to_parts(message):
|
||
|
|
"""Split message into single parts."""
|
||
|
|
if len(message) <= MESSAGE_SIZE:
|
||
|
|
return [message]
|
||
|
|
|
||
|
|
punc = "!()[]?.,;:"
|
||
|
|
punc_list = [re.escape(c) for c in punc]
|
||
|
|
pattern = '|'.join(punc_list)
|
||
|
|
parts = re.split(pattern, message)
|
||
|
|
|
||
|
|
def split_by_space(fullstring):
|
||
|
|
"""Split a string by space."""
|
||
|
|
if len(fullstring) > MESSAGE_SIZE:
|
||
|
|
idx = fullstring.rfind(' ', 0, MESSAGE_SIZE)
|
||
|
|
return [fullstring[:idx]] + split_by_space(fullstring[idx:])
|
||
|
|
else:
|
||
|
|
return [fullstring]
|
||
|
|
|
||
|
|
msg_parts = []
|
||
|
|
for part in parts:
|
||
|
|
msg_parts += split_by_space(part)
|
||
|
|
|
||
|
|
return [msg for msg in msg_parts if len(msg) > 0]
|