U jn @sdZddgZddlZddlZddlZz ddlZWnek rHdZYnXddlmZm Z m Z m Z erpe dGdddej Z Gd ddejjZdd d eeee e ejefd d dZe e e ejdddZdS)z Utilities. to_threadopen_connectionN) TYPE_CHECKINGAnyCallableTuple StreamWriterc@s"eZdZdZejddddZdS)StreamReaderProtocolzExtends asyncio.streams.StreamReaderProtocol for adding start_tls(). The ``start_tls()`` is based on ``asyncio.streams.StreamWriter`` introduced in Python 3.11. It provides the same functionality for older Python versions. N)writerreturncCs&|j}||_||_|ddk |_dS)zQReplace stream writer. Args: writer: Stream Writer. Z sslcontextN) transportZ_stream_writer _transportZget_extra_infoZ _over_ssl)selfr r r=/tmp/pip-unpacked-wheel-d6bt0v6z/mysql/connector/aio/utils.py_replace_writer:sz$StreamReaderProtocol._replace_writer)__name__ __module__ __qualname____doc__asynciorrrrrrr 3sr c@s.eZdZdZdddejeeddddZdS)rzExtends asyncio.streams.StreamWriter for adding start_tls(). The ``start_tls()`` is based on ``asyncio.streams.StreamWriter`` introduced in Python 3.11. It provides the same functionality for older Python versions. N)server_hostnamessl_handshake_timeout) ssl_contextrrr csT|jjdk }|j}|IdH|jj|j|||||dIdH}||_||dS)zUpgrade an existing stream-based connection to TLS. Args: ssl_context: Configured SSL context. server_hostname: Server host name. ssl_handshake_timeout: SSL handshake timeout. N) server_siderr) _protocolZ_client_connected_cbZdrainZ_loop start_tlsr r)rrrrrprotocolZ new_transportrrrrMs  zStreamWriter.start_tls) rrrrssl SSLContextstrintrrrrrrFs i)limit)hostportr"kwdsr c s^t}tjj||d}t||d|jfdd||f|IdH\}}t|||}||fS)aA wrapper for create_connection() returning a (reader, writer) pair. This function is based on ``asyncio.streams.open_connection`` and adds a custom stream reader. MySQL expects TLS negotiation to happen in the middle of a TCP connection, not at the start. This function in conjunction with ``_StreamReaderProtocol`` and ``_StreamWriter`` allows the TLS negotiation on an existing connection. Args: host: Server host name. port: Server port. limit: The buffer size limit used by the returned ``StreamReader`` instance. By default the limit is set to 64 KiB. Returns: tuple: Returns a pair of reader and writer objects that are instances of ``StreamReader`` and ``StreamWriter`` classes. )r"loop)r&csS)Nrrrrrz!open_connection..N)rget_running_loopstreams StreamReaderr create_connectionr) r#r$r"r%r&readerr _r rr'rrms  $)funcargskwargsr cs8t}t}tj|j|f||}|d|IdHS)aQAsynchronously run function ``func`` in a separate thread. This function is based on ``asyncio.to_thread()`` introduced in Python 3.9, which provides the same functionality for older Python versions. Returns: coroutine: A coroutine that can be awaited to get the eventual result of ``func``. N)rr* contextvarsZ copy_context functoolspartialrunZrun_in_executor)r0r1r2r&ctxZ func_callrrrrs )NN)r__all__rr3r4r ImportErrortypingrrrrappendr r+rr r!r,rZFuturerrrrr s2   (