>ijgdZddgZddlZddlZddlZddlZddlmZmZ ddl Z n #e $rdZ YnwxYwddl m Z m Z mZmZe rddlmZedGd d ejZGd dejjZ dd d dededede deejeff dZdede de dejfdZdedefdZdefdZdS)z Utilities. to_threadopen_connectionN)ReadTimeoutErrorWriteTimeoutError) 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)selfrrs W/var/www/html/mpstechhub/venv/lib/python3.11/site-packages/mysql/connector/aio/utils.py_replace_writerz$StreamReaderProtocol._replace_writer?s9 $ $#"11,??tK)__name__ __module__ __qualname____doc__asyncior rrrrr8sK Lg&: Lt L L L L L Lrrc <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_tlsRsn9E >jjll"j22 O  #+"73             &&&&&r) rrrrssl SSLContextstrintr,r!rrr r Ksl $%) '''^' ' # '  ''''''ri)limithostportr3kwdsrc 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. )r3loop)r8cSNr!)r-srz!open_connection..srN)r get_running_loopstreams StreamReaderrcreate_connectionr ) r4r5r3r6r8readerr_rr-s @rrrrs.  # % %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)r r< contextvars copy_context functoolspartialrunrun_in_executor)rBrCrDr8ctx func_calls rrrsp  # % %D  " $ $C!#'4A$AAA&AAI%%dI66 6 6 6 6 6 66rreasonc0dtdtffd }|S)z&Use it to decorate deprecated methods.rBrcvtjdtdtdtffd }|S)NrCrDrcrKtjdjdtd|i|d{VS)NzCall to deprecated function z . Reason: )category stacklevel)warningswarnrDeprecationWarning)rCrDrBrNs rwrapperz.deprecated..decorator..wrappersd MPt}PPPP+     t.v........ .r)rHwrapsrr )rBrXrNs` r decoratorzdeprecated..decoratorsX    / / / / / / / / /   /rr )rNrZs` r deprecatedr\s7  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 vtjdddtdtdtffd }|S)Ncnxr rCrDrcK |g|Ri|d{VS#t$r=}t|ttfr|d{V|d}~wwxYwr:) Exception isinstancerrclose)r`rCrDerrr^s rhandle_cnx_methodzGhandle_read_write_timeout..decorator..handle_cnx_methods 'Z=d===f=========   c$46G#HII&))++%%%%%%%  s A8AA)rHrYr)r^rfs` rrZz,handle_read_write_timeout..decorators^  $ $ * 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@ ) *FFFFFFFFJJJJ CCC766666666666#EEEEEE NN>"""LLLLL77LLL&$'$'$'$'$'7?/$'$'$'P#'8= 25GJ 7  -.> 7( 73 7# 7'. 7 7 7 7 sx$8s %//