home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-17 19:41:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

* made supported_options const

Browse Source 
* fixed direct env variable access
This commit is contained in:
lufton
2019-05-03 08:51:43 +03:00
parent 398704c8b6
commit 4b84ad42ff
1 changed files with 22 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
homeassistant/components/google_cloud_tts/tts.py
View File

@@ -1,11 +1,11 @@
"""Support for the Google Cloud TTS service.""" """Support for the Google Cloud TTS service."""
import logging
import os
import asyncio
import async_timeout
import voluptuous as vol
from google.cloud import texttospeech from google.cloud import texttospeech
from homeassistant.components.tts import CONF_LANG, PLATFORM_SCHEMA, Provider from homeassistant.components.tts import CONF_LANG, PLATFORM_SCHEMA, Provider
import logging
import os
import async_timeout
import asyncio
import voluptuous as vol
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@@ -24,7 +24,10 @@ DEFAULT_LANG = SUPPORT_LANGUAGES[0]
CONF_GENDER = 'gender' CONF_GENDER = 'gender'
CONF_VOICE = 'voice' CONF_VOICE = 'voice'
CONF_KEY_FILE = 'key_file' CONF_KEY_FILE = 'key_file'
GOOGLE_APPLICATION_CREDENTIALS = 'GOOGLE_APPLICATION_CREDENTIALS'
SUPPORTED_OPTIONS = [
CONF_VOICE, CONF_GENDER
]
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Optional(CONF_LANG, default=DEFAULT_LANG): vol.In(SUPPORT_LANGUAGES), vol.Optional(CONF_LANG, default=DEFAULT_LANG): vol.In(SUPPORT_LANGUAGES),
@@ -56,14 +59,16 @@ class GoogleCloudTTSProvider(Provider):
self._voice = voice self._voice = voice
self.name = 'Google Cloud TTS' self.name = 'Google Cloud TTS'
path = hass.config.path(key_file) path = hass.config.path(key_file)
if key_file and os.path.isfile(path): if key_file:
os.environ[GOOGLE_APPLICATION_CREDENTIALS] = path if os.path.isfile(path):
if GOOGLE_APPLICATION_CREDENTIALS not in os.environ: self._client = texttospeech \
.TextToSpeechClient.from_service_account_json(path)
else:
_LOGGER.error( _LOGGER.error(
"You need to specify valid" "GOOGLE_APPLICATION_CREDENTIALS file doesn't exist!"
" GOOGLE_APPLICATION_CREDENTIALS file location."
) )
self.client = texttospeech.TextToSpeechClient() else:
self._client = texttospeech.TextToSpeechClient()
self.audio_config = texttospeech.types.AudioConfig( self.audio_config = texttospeech.types.AudioConfig(
audio_encoding=texttospeech.enums.AudioEncoding.MP3 audio_encoding=texttospeech.enums.AudioEncoding.MP3
) )
@@ -81,7 +86,7 @@ class GoogleCloudTTSProvider(Provider):
@property @property
def supported_options(self): def supported_options(self):
"""Return a list of supported options.""" """Return a list of supported options."""
return ["voice", "gender"] return SUPPORTED_OPTIONS
async def async_get_tts_audio(self, message, language, options=None): async def async_get_tts_audio(self, message, language, options=None):
"""Load TTS from google.""" """Load TTS from google."""
@@ -107,7 +112,7 @@ class GoogleCloudTTSProvider(Provider):
synthesis_input = texttospeech.types.SynthesisInput( synthesis_input = texttospeech.types.SynthesisInput(
text=message text=message
) )
response = self.client.synthesize_speech( response = self._client.synthesize_speech(
synthesis_input, synthesis_input,
voice, voice,
self.audio_config self.audio_config
@@ -115,7 +120,7 @@ class GoogleCloudTTSProvider(Provider):
return "mp3", response.audio_content return "mp3", response.audio_content
except asyncio.TimeoutError as ex: except asyncio.TimeoutError as ex:
_LOGGER.error("Timeout for Google Cloud TTS: %s", ex) _LOGGER.error("Timeout for Google Cloud TTS call: %s", ex)
except Exception as ex: except Exception as ex:
_LOGGER.error("Error occured during Google Cloud TTS call: %s", ex) _LOGGER.error("Error occured during Google Cloud TTS call: %s", ex)