o g@s*dZddgZddlZddlZddlZddlZddlmZmZzddl Z Wn e y/dZ Ynwddl m Z m Z mZmZe rIddlmZedGd d d ejZGd ddejjZ dd d dededede deejeff ddZdede de dejfddZdedefddZdefddZdS)z Utilities. to_threadopen_connectionN)ReadTimeoutErrorWriteTimeoutError) 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 rrS/var/www/html/api-tag/env/lib/python3.10/site-packages/mysql/connector/aio/utils.py_replace_writer?sz$StreamReaderProtocol._replace_writer)__name__ __module__ __qualname____doc__asyncior rrrrrr 8sr 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_contextrrrcsV|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)rr rrr!protocol new_transportrrrr&Rs   zStreamWriter.start_tls) rrrrssl SSLContextstrintr&rrrrr Ks i)limithostportr-kwdsrc 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)r1csSNrrr'rrsz!open_connection..N)rget_running_loopstreams StreamReaderr create_connectionr ) r.r/r-r0r1readerr_r rr3rrrs (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)rr5 contextvars copy_context functoolspartialrunrun_in_executor)r;r<r=r1ctx func_callrrrrs  reasoncsdtdtffdd }|S)z&Use it to decorate deprecated methods.r;rcs*tdtdtdtffdd }|S)Nr<r=rcs4tjdjdtdd|i|IdHS)NzCall to deprecated function z . Reason: )category stacklevel)warningswarnrDeprecationWarning)r<r=)r;rFrrwrappersz.deprecated..decorator..wrapper)r@wrapsrr)r;rMrF)r;r decoratorszdeprecated..decoratorr)rFrPrrOr deprecateds rRcCsdtdtfdd}|S)z Decorator to close the current connection if a read or a write timeout is raised by the method passed via the func parameter. cnx_methodrc s,tdddtdtdtffdd }|S)Ncnxr r<r=rc sZz|g|Ri|IdHWSty,}zt|ttfr&|IdH|d}~wwr2) Exception isinstancerrclose)rTr<r=errrSrrhandle_cnx_methodszGhandle_read_write_timeout..decorator..handle_cnx_method)r@rNr)rSrZrrYrrPs z,handle_read_write_timeout..decoratorrQ)rPrrrhandle_read_write_timeoutsr[)NN)r__all__rr>r@rJmysql.connector.errorsrrr) ImportErrortypingrrrr mysql.connector.aio.abstractsr appendr r6r r+r,r7rFuturerrRr[rrrrsF    (