AahZdZddgZddlZddlZddlZ ddlZn #e$rdZYnwxYwddlmZm Z m Z m Z erddl m Z edGdd ejZGd dejjZ dd d d ededede de ejeff dZde de de dejfdZdS)z Utilities. to_threadopen_connectionN) TYPE_CHECKINGAnyCallableTuple)MySQLConnectionAbstract StreamWriterc,eZdZdZdejddfdZdS)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. writerreturnNch|j}||_||_|ddu|_dS)zQReplace stream writer. Args: writer: Stream Writer. sslcontextN) transport_stream_writer _transportget_extra_info _over_ssl)selfr rs K/var/www/lms/venv/lib/python3.11/site-packages/mysql/connector/aio/utils.py_replace_writerz$StreamReaderProtocol._replace_writer<s9 $ $#"11,??tK)__name__ __module__ __qualname____doc__asyncior rrrr r 5sK Lg&: Lt L L L L L Lrr c <eZdZdZddddejdededdfdZdS) 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_contextr!r"rcK|jjdu}|j}|d{V|j|j|||||d{V}||_||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_sider!r") _protocol_client_connected_cbdrain_loop start_tlsrr)rr#r!r"r%protocol new_transports rr*zStreamWriter.start_tlsOsn9E >jjll"j22 O  #+"73             &&&&&r) rrrrssl SSLContextstrintr*rrrr r Hsl $%) '''^' ' # '  ''''''ri)limithostportr1kwdsrc Ktj}tj||}t || |j fd||fi|d{V\}}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. )r1loop)r6cS)Nr)r+srz!open_connection..srN)rget_running_loopstreams StreamReaderr create_connectionr ) r2r3r1r4r6readerr_r r+s @rrros.  # % %D _ ) )D ) A AF#F666H//0@0@0@0@$UUPTUUUUUUUULIq )Xvt < rfuncargskwargscKtj}tj}t j|j|g|Ri|}|d|d{VS)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)rr9 contextvars copy_context functoolspartialrunrun_in_executor)r?r@rAr6ctx func_calls rrrsp  # % %D  " $ $C!#'4A$AAA&AAI%%dI66 6 6 6 6 6 66r)NN)r__all__rrCrEr- ImportErrortypingrrrrmysql.connector.aio.abstractsr appendr r:r r/r0r;rFuturerrrrrQs@ ) *JJJJ CCC766666666666#EEEEEE NN>"""LLLLL77LLL&$'$'$'$'$'7?/$'$'$'P#'8= 25GJ 7  -.> 7( 73 7# 7'. 7 7 7 7 7 7s ##