o ah@sdZddgZddlZddlZddlZzddlZWn ey#dZYnwddlmZm Z m Z m Z er=ddl m Z edGdd d ejZGd ddejjZ dd d d ededede de ejeff ddZde de de dejfddZdS)z Utilities. to_threadopen_connectionN) TYPE_CHECKINGAnyCallableTuple)MySQLConnectionAbstract 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 rrQ/home/air/sos/back/venv/lib/python3.10/site-packages/mysql/connector/aio/utils.py_replace_writer<sz$StreamReaderProtocol._replace_writer)__name__ __module__ __qualname____doc__asyncior rrrrrr 5sr c @s4eZdZdZddddejdededdfdd ZdS) 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 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$Os   zStreamWriter.start_tls) rrrrssl SSLContextstrintr$rrrrr Hs 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 rr0rros (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)rr2 contextvars copy_context functoolspartialrunrun_in_executor)r8r9r:r/ctx func_callrrrrs  )NN)r__all__rr;r=r' ImportErrortypingrrrrmysql.connector.aio.abstractsrappendr r3r r)r*r4rFuturerrrrrs>    (