a i @sdZddgZddlZddlZddlZz ddlZWneyFdZYn0ddlmZm Z m Z m Z erzddl m Z edGdd d ejZGd ddejjZdd d eeee e ejefd ddZe e e ejdddZdS)z Utilities. to_threadopen_connectionN) TYPE_CHECKINGAnyCallableTuple)MySQLConnectionAbstract 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}||_||_|ddu|_dS)zQReplace stream writer. Args: writer: Stream Writer. Z sslcontextN) transportZ_stream_writer _transportZget_extra_infoZ _over_ssl)selfr r rI/home/Claro/venv/lib/python3.9/site-packages/mysql/connector/aio/utils.py_replace_writer<sz$StreamReaderProtocol._replace_writer)__name__ __module__ __qualname____doc__asyncior rrrrrr 5sr c@s.eZdZdZdddejeeddddZdS)r zExtends 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|jjdu}|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_tlsrr)rrrrrprotocolZ new_transportrrrrOs  zStreamWriter.start_tls) rrrrssl SSLContextstrintrrrrrr Hs i)limit)hostportr#kwdsr c sbt}tjj||d}t||d|jfdd||fi|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(rros  ()funcargskwargsr cs>t}t}tj|j|g|Ri|}|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)r1r2r3r'ctxZ func_callrrrrs )NN)r__all__rr4r5r ImportErrortypingrrrrZmysql.connector.aio.abstractsrappendr r,r r!r"r-rZFuturerrrrr s,     (