o m=8j @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_statusNVoiceTagContentCategoriesVoicePersonalities) getr r generate_sec_ms_gecrheaders_with_muidrjsonloadstext)rrrurldatavoicer"V/home/nk/hobo-godmode/plappi-mvp/.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_CTXClientResponseErrorstatusr handle_client_response_error)r%rrr er"r"r# 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)selfr"r"r#__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)LanguageLocale-r)split.0r!r"r"r# hsz(VoicesManager.create..T)r/r.r0r1)clsr4r2r0r"r"r#create_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")itemsr9r>r"r#r;wsz&VoicesManager.find..)r1 RuntimeErrorr0)r2r>matching_voicesr"r@r#findns zVoicesManager.find)rN)N)__name__ __module__ __qualname____doc__r3 classmethodrrr r=rr r rCr"r"r"r#r/Vs   r/)rGrrtypingrrrr'certifityping_extensionsr constantsrrr drmr r r r create_default_contextwherer)r( SSLContextstrr$ BaseConnectorr.r/r"r"r"r#s:   (