o i @sdZddlZddlZddlmZmZmZddlZddlZddl m Z ddl m Z m Z mZddlmZddlmZmZmZejed Zd ejd ejd eed eefddZddddeejd eed eefddZGdddZdS)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_ctxproxyreturnc s|jtdtdttt||dd4IdH}t| IdH}WdIdHn 1IdHs7wY|D]"}d|vrHi|d<d|dvrTg|dd<d|dvr`g|dd<q>|S) 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_statusNZVoiceTagZContentCategoriesZVoicePersonalities) getr r Zgenerate_sec_ms_gecrZheaders_with_muidrjsonloadstext)rrrurldatavoicerI/home/kim/smarthome/.venv/lib/python3.10/site-packages/edge_tts/voices.py __list_voicess,(     r) connectorrr c stj|dd4IdHJ}z t|t|IdH}Wn'tjy?}z|jdkr't|t|t|IdH}WYd}~nd}~wwWdIdH|SWdIdH|S1IdHs\wY|S)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_CTXZClientResponseErrorstatusr Zhandle_client_response_error)r rrrerrr list_voices:s&     r'c@sVeZdZdZd ddZe d deeeddfddZ d e e dee fd d Z dS) VoicesManagerzF A class to find the correct voice based on their attributes. rNcCsg|_d|_dS)NF)voices called_create)selfrrr__init__[s zVoicesManager.__init__ custom_voicescs:t}|durtIdHn|}dd|D|_d|_|S)z\ Creates a VoicesManager object and populates it with all available voices. NcSs*g|]}i|d|dddiqS)ZLanguageZLocale-r)split.0rrrr hsz(VoicesManager.create..T)r(r'r)r*)clsr-r+r)rrrcreate_szVoicesManager.createkwargsc s&|jstdfdd|jD}|S)zM Finds all matching voices based on the provided attributes. z9VoicesManager.find() called before VoicesManager.create()cs g|] }|kr|qSr)itemsr0r5rrr2wsz&VoicesManager.find..)r* RuntimeErrorr))r+r5Zmatching_voicesrr7rfindns zVoicesManager.find)rN)N)__name__ __module__ __qualname____doc__r, classmethodrrr r4rr r r9rrrrr(Vs   r()r=rrtypingrrrr"certifiZtyping_extensionsr constantsrrr Zdrmr r r r create_default_contextwherer$r# SSLContextstrrZ BaseConnectorr'r(rrrrs:   (