Add strict type annotations to amazon_polly (#50697)

* add strict type annotations * apply suggestions Co-authored-by: Erik Montnemery <erik@montnemery.com>
2021-05-17 14:09:52 +02:00 · 2021-05-17 14:09:52 +02:00 · df6862a519
commit df6862a519
parent 9e86602950
6 changed files with 226 additions and 141 deletions
--- a/homeassistant/components/amazon_polly/const.py
+++ b/homeassistant/components/amazon_polly/const.py
@ -0,0 +1,131 @@
+"""Constants for the Amazon Polly text to speech service."""
+from __future__ import annotations
+
+from typing import Final
+
+CONF_REGION: Final = "region_name"
+CONF_ACCESS_KEY_ID: Final = "aws_access_key_id"
+CONF_SECRET_ACCESS_KEY: Final = "aws_secret_access_key"
+
+DEFAULT_REGION: Final = "us-east-1"
+SUPPORTED_REGIONS: Final[list[str]] = [
+    "us-east-1",
+    "us-east-2",
+    "us-west-1",
+    "us-west-2",
+    "ca-central-1",
+    "eu-west-1",
+    "eu-central-1",
+    "eu-west-2",
+    "eu-west-3",
+    "ap-southeast-1",
+    "ap-southeast-2",
+    "ap-northeast-2",
+    "ap-northeast-1",
+    "ap-south-1",
+    "sa-east-1",
+]
+
+CONF_ENGINE: Final = "engine"
+CONF_VOICE: Final = "voice"
+CONF_OUTPUT_FORMAT: Final = "output_format"
+CONF_SAMPLE_RATE: Final = "sample_rate"
+CONF_TEXT_TYPE: Final = "text_type"
+
+SUPPORTED_VOICES: Final[list[str]] = [
+    "Olivia",  # Female, Australian, Neural
+    "Zhiyu",  # Chinese
+    "Mads",
+    "Naja",  # Danish
+    "Ruben",
+    "Lotte",  # Dutch
+    "Russell",
+    "Nicole",  # English Australian
+    "Brian",
+    "Amy",
+    "Emma",  # English
+    "Aditi",
+    "Raveena",  # English, Indian
+    "Joey",
+    "Justin",
+    "Matthew",
+    "Ivy",
+    "Joanna",
+    "Kendra",
+    "Kimberly",
+    "Salli",  # English
+    "Geraint",  # English Welsh
+    "Mathieu",
+    "Celine",
+    "Lea",  # French
+    "Chantal",  # French Canadian
+    "Hans",
+    "Marlene",
+    "Vicki",  # German
+    "Aditi",  # Hindi
+    "Karl",
+    "Dora",  # Icelandic
+    "Giorgio",
+    "Carla",
+    "Bianca",  # Italian
+    "Takumi",
+    "Mizuki",  # Japanese
+    "Seoyeon",  # Korean
+    "Liv",  # Norwegian
+    "Jacek",
+    "Jan",
+    "Ewa",
+    "Maja",  # Polish
+    "Ricardo",
+    "Vitoria",  # Portuguese, Brazilian
+    "Cristiano",
+    "Ines",  # Portuguese, European
+    "Carmen",  # Romanian
+    "Maxim",
+    "Tatyana",  # Russian
+    "Enrique",
+    "Conchita",
+    "Lucia",  # Spanish European
+    "Mia",  # Spanish Mexican
+    "Miguel",  # Spanish US
+    "Penelope",  # Spanish US
+    "Lupe",  # Spanish US
+    "Astrid",  # Swedish
+    "Filiz",  # Turkish
+    "Gwyneth",  # Welsh
+]
+
+SUPPORTED_OUTPUT_FORMATS: Final[list[str]] = ["mp3", "ogg_vorbis", "pcm"]
+
+SUPPORTED_ENGINES: Final[list[str]] = ["neural", "standard"]
+
+SUPPORTED_SAMPLE_RATES: Final[list[str]] = ["8000", "16000", "22050", "24000"]
+
+SUPPORTED_SAMPLE_RATES_MAP: Final[dict[str, list[str]]] = {
+    "mp3": ["8000", "16000", "22050", "24000"],
+    "ogg_vorbis": ["8000", "16000", "22050"],
+    "pcm": ["8000", "16000"],
+}
+
+SUPPORTED_TEXT_TYPES: Final[list[str]] = ["text", "ssml"]
+
+CONTENT_TYPE_EXTENSIONS: Final[dict[str, str]] = {
+    "audio/mpeg": "mp3",
+    "audio/ogg": "ogg",
+    "audio/pcm": "pcm",
+}
+
+DEFAULT_ENGINE: Final = "standard"
+DEFAULT_VOICE: Final = "Joanna"
+DEFAULT_OUTPUT_FORMAT: Final = "mp3"
+DEFAULT_TEXT_TYPE: Final = "text"
+
+DEFAULT_SAMPLE_RATES: Final[dict[str, str]] = {
+    "mp3": "22050",
+    "ogg_vorbis": "22050",
+    "pcm": "16000",
+}
+
+AWS_CONF_CONNECT_TIMEOUT: Final = 10
+AWS_CONF_READ_TIMEOUT: Final = 5
+AWS_CONF_MAX_POOL_CONNECTIONS: Final = 1