GgdZddgZddlZddlZddlZddlZddlmZmZ ddl Z ddl m Z m Z mZmZe rddlmZej%dGd d ej&ZGd dej(j*Z dd d dededede deej0eff dZdede de dej4fdZdedefdZdefdZy#e $rdZ YwxYw)z Utilities. to_threadopen_connectionN)ReadTimeoutErrorWriteTimeoutError) TYPE_CHECKINGAnyCallableTuple)MySQLConnectionAbstract StreamWriterc4eZdZdZdej ddfdZy)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}||_||_|jddu|_y)zQReplace stream writer. Args: writer: Stream Writer. sslcontextN) transport_stream_writer _transportget_extra_info _over_ssl)selfrrs v/home/kushmeetdev/apache_webroot/langgraph_flaskproject/venv/lib/python3.12/site-packages/mysql/connector/aio/utils.py_replace_writerz$StreamReaderProtocol._replace_writer?s5 $$ $#"11,?tK)__name__ __module__ __qualname____doc__asyncior rrrrr8s# Lg&:&: Lt Lrrc DeZdZdZddddej dededdfdZy) 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$rc"K|jjdu}|j}|jd{|jj |j |||||d{}||_|j |y7P7w)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_tlsRsnn99E >>jjl"jj22 OO  #+"73       &  s!8BB 2B-B .B B) rrrrssl SSLContextstrintr,r!rrr r Ks> $%) '^^' ' # '  'ri)limithostportr3kwdsrc Ktj}tjj||}t || |j fd||fi|d{\}}t | ||}||fS7w)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. )r3loop)r8cSNr!)r-srz!open_connection..srN)r get_running_loopstreams StreamReaderrcreate_connectionr ) r4r5r3r6r8readerr_rr-s @rrrrs.  # # %D __ ) )D ) AF#F6H///0@$UPTUULIq )Xvt VsAA=!A;"A=funcargskwargscKtj}tj}t j |j |g|i|}|jd|d{S7w)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)r r< contextvars copy_context functoolspartialrunrun_in_executor)rBrCrDr8ctx func_calls rrrs]  # # %D  " " $C!!#''4A$A&AI%%dI6 66 6sA"A+$A)%A+reasonc,dtdtffd }|S)z&Use it to decorate deprecated methods.rBrcltjdtdtdtffd }|S)NrCrDrcKtjdjdtd|i|d{S7w)NzCall to deprecated function z . Reason: )category stacklevel)warningswarnrDeprecationWarning)rCrDrBrNs rwrapperz.deprecated..decorator..wrappersE MM.t}}oZxP+  t.v.. ..s8AAA)rHwrapsrr )rBrXrNs` r decoratorzdeprecated..decorators:   / / / /  /rr )rNrZs` r deprecatedr\s  X  rc&dtdtfd}|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 ltjdddtdtdtffd }|S)Ncnxr rCrDrcK |g|i|d{S7#t$r6}t|ttfr|j d{7|d}~wwxYwwr:) Exception isinstancerrclose)r`rCrDerrr^s rhandle_cnx_methodzGhandle_read_write_timeout..decorator..handle_cnx_methods[ '=d=f==== c$46G#HI))+%%  s<AA A)A AAAA)rHrYr)r^rfs` rrZz,handle_read_write_timeout..decoratorsF  $ * 36 BE    % ! rr[)rZs rhandle_read_write_timeoutrgs !h !8 ! r)NN)r__all__r rFrHrUmysql.connector.errorsrrr/ ImportErrortypingrrr r mysql.connector.aio.abstractsr appendrr=r r1r2r>rFuturerr\rgr!rrros@ ) *F76E NN>"L777L&$'7??//$'P#'8= 25GJ 7   -.> 7( 73 7# 7'.. 7 sx$8M CsCC C