j bdZddlZddlZddlmZmZmZddlZddlZddl m Z ddl m Z m Z mZddlmZddlmZmZmZej(ej* Zd ej.d ej0d eed eefdZddddeej6d eed eefdZGddZy)z|This module contains functions to list all available voices and a class to find the correct voice based on their attributes.N)AnyListOptional)Unpack)SEC_MS_GEC_VERSION VOICE_HEADERS VOICE_LIST)DRM)VoiceVoicesManagerFindVoicesManagerVoice)cafilesessionssl_ctxproxyreturncK|jtdtjdttj t ||d4d{}tj|jd{}dddd{D]*}d|vri|d<d|dvrg|dd<d|dvs#g|dd<,|S7p7K79#1d{7swYIxYww) a Private function that makes the request to the voice list URL and parses the JSON response. This function is used by list_voices() and makes it easier to handle client response errors related to clock skew. Args: session (aiohttp.ClientSession): The aiohttp session to use for the request. ssl_ctx (ssl.SSLContext): The SSL context to use for the request. proxy (Optional[str]): The proxy to use for the request. Returns: List[Voice]: A list of voices and their attributes. z &Sec-MS-GEC=z&Sec-MS-GEC-Version=T)headersrsslraise_for_statusNVoiceTagContentCategoriesVoicePersonalities) getr r generate_sec_ms_gecrheaders_with_muidr jsonloadstext)rrrurldatavoices C  C"C (C"; C" C  C" CC CC") connectorrr&cKtj|d4d{} t|t|d{}dddd{|S707#tj$rI}|j dk7rt j|t|t|d{7}Yd}~md}~wwxYw7i#1d{7swYSxYww)a List all available voices and their attributes. This pulls data from the URL used by Microsoft Edge to return a list of all available voices. Args: connector (Optional[aiohttp.BaseConnector]): The connector to use for the request. proxy (Optional[str]): The proxy to use for the request. Returns: List[Voice]: A list of voices and their attributes. T)r& trust_envNi)aiohttp ClientSessionr%_SSL_CTXClientResponseErrorstatusr handle_client_response_error)r&rrr"es r$ list_voicesr0:s $$yDIIW A&w%@@DJI KJ@** Axx3  , ,Q /&w%@@@D  AJIII KsC AC B4AAA C B2 C AB/&9B*B" B*%B4*B//B42C 4C:B= ;CC c^eZdZdZd dZe d deeeddfdZ de e dee fdZ y) VoicesManagerzF A class to find the correct voice based on their attributes. rNc g|_d|_y)NF)voices called_create)selfs r$__init__zVoicesManager.__init__[s02 #( custom_voicescKt}|td{n|}|Dcgc]}i|d|djddi!c}|_d|_|S7?cc}ww)z\ Creates a VoicesManager object and populates it with all available voices. NLanguageLocale-rT)r2r0splitr4r5)clsr9r6r4r#s r$createzVoicesManager.create_sx(5(={}$$=NT NTU @u @j%/"7"7"rWs, &&$DD@@ %3 % %]W]]_ =$  " "$-0^^$DLSM$ %[$P59QU7001AI# %[8$$r8