o #f9@sdZddgZddlZddlZddlZzddlZWn ey#dZYnwddlmZm Z m Z m Z er7e dGdddej Z Gd ddejjZ dd d d ed edede de ejeff ddZde de de dejfddZdS)z Utilities. to_threadopen_connectionN) TYPE_CHECKINGAnyCallableTuple StreamWriterc@s$eZdZdZdejddfddZdS)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. writerreturnNcCs&|j}||_||_|ddu|_dS)zQReplace stream writer. Args: writer: Stream Writer. sslcontextN) transport_stream_writer _transportget_extra_info _over_ssl)selfr r rO/home/ubuntu/dev/venv/lib/python3.10/site-packages/mysql/connector/aio/utils.py_replace_writer:sz$StreamReaderProtocol._replace_writer)__name__ __module__ __qualname____doc__asynciorrrrrrr 3sr c @s4eZdZdZddddejdededdfdd ZdS) 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 csV|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) _protocol_client_connected_cbdrain_loop start_tlsrr)rrrrrprotocol new_transportrrrr#Ms   zStreamWriter.start_tls) rrrrssl SSLContextstrintr#rrrrrFs i)limithostportr*kwdsr c sdt}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)Nrrr$rrsz!open_connection..N)rget_running_loopstreams StreamReaderr create_connectionr) r+r,r*r-r.readerr _r rr/rrms (funcargskwargscs@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)rr1 contextvars copy_context functoolspartialrunrun_in_executor)r7r8r9r.ctx func_callrrrrs  )NN)r__all__rr:r<r& ImportErrortypingrrrrappendr r2rr(r)r3rFuturerrrrrs<   (