G)fIUdZddlmZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZmZe rddlmZmZdd lmZdd lmZ dd lmZd d lmZddlmZm Z ddl!m!Z!ddl"m#Z# ddl$m%Z%ddl&m'Z'ddl(m)Z)dZ*ddlm0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;ejxdZ=de>d<d+dZ?d,dZ@d-dZA d. d/dZBd0d ZC d1d!ZDGd"d#eZEGd$d%eEZFGd&d'eEZG d. d2d(ZHGd)d*ZIy#e$rYwxYw#e$r< ddl+m%Z%ddl,m)Z)dZ*n)#e$r!Z-ej\j_de-dZ-[-wwxYwYwxYw)3z1MySQL instrumentation supporting mysql-connector.) annotationsN)ABCabstractmethod) nullcontext) TYPE_CHECKINGAnyCallable CollectionDictOptionalUnion)MySQLConnectionAbstractMySQLCursorAbstract)MySQLConnection)PooledMySQLConnection)CMySQLConnection) connector) CNX_POOL_ARGSDEFAULT_CONFIGURATION)logger) VERSION_TEXT)trace)TracerProvider)SpanAttributesTFzBundled installation has missing dependencies. Please use `pip install mysql-connector-python[opentelemetry]`, or for an editable install use `pip install -e '.[opentelemetry]'`, to install the dependencies required by the bundled opentelemetry package.) CONNECTION_SPAN_NAME DB_SYSTEMDEFAULT_THREAD_IDDEFAULT_THREAD_NAMEFIRST_SUPPORTED_VERSIONNET_SOCK_FAMILYNET_SOCK_HOST_ADDRNET_SOCK_HOST_PORTNET_SOCK_PEER_ADDRNET_SOCK_PEER_PORTOPTION_CNX_SPANOPTION_CNX_TRACERz ^/\*.*?\*/z re.Patternleading_comment_removerc|r|jr|sy|jtjtjj |j |y)zRecords an exeception event.N) is_recording set_statusrStatus StatusCodeERRORrecord_exception)spanexcs j/var/www/html/flask-app/venv/lib/python3.12/site-packages/mysql/connector/opentelemetry/instrumentation.pyrecord_exception_eventr5OsD t((*#OOELL!1!1!7!789#cJ|r|jsy|jy)z Ends span.N)r,end)r2s r4end_spanr9Xs t((*HHJr6cv|r7t|tr'tjd|j dSy)z&Parse query to extract operation name.r) isinstancestrr*subsplit) operations r4get_operation_namerA`s2Z 3/&**2y9??A!DD r6c|r|jsy|i}|r |j nd|v}tjttj|rdndt |rdndi}|r |r |j n|jdtd|tj<|r |jn|jdtd|tj<t|d r|jr |jjj\|t <}|jjj#\|t$<|t&<|tj|k7rt||t(<nj|r |jn|jd|t <t|d r8|jr, |jjj#|t$<|j1|y#t*$r }t-j.d |Yd}~6d}~wwxYw#t*$r }t-j.d |Yd}~bd}~wwxYw) aDefines connection span attributes. If `cnx` is None then we use `cnx_kwargs` to get basic net information. Basic net attributes are defined such as: * DB_SYSTEM * NET_TRANSPORT * NET_SOCK_FAMILY Socket-level attributes [*] are also defined [**]. [*]: Socket-level attributes identify peer and host that are directly connected to each other. Since instrumentations may have limited knowledge on network information, instrumentations SHOULD populate such attributes to the best of their knowledge when populate them at all. [**]: `CMySQLConnection` connections have no access to socket-level details so socket-level attributes aren't included. `MySQLConnection` connections, on the other hand, do include socket-level attributes. References: [1]: https://github.com/open-telemetry/opentelemetry-specification/blob/main/ specification/trace/semantic_conventions/span-general.md N unix_socketip_tcpinprocinetunixhostport_socketzConnection socket is down %s)r, _unix_socketrr NET_TRANSPORTr#_hostgetr NET_PEER_NAME_port NET_PEER_PORThasattrrJsock getpeernamer& getsocknamer$r%r' Exceptionrwarningset_attributes)cnxcnx_span cnx_kwargsis_tcpattrssock_peer_portsock_errs r4set_connection_span_attrsr`hs8 8002 %(!! !m:.MF   )$$&hh6vECII*..9Nv9V"W n**+CII*..9Nv9V"W n**+ 3 "s{{ ?KK$$002,-" KK$$002,-,- 556.H1?E,- !$C   )F  ! 3 "s{{ I,/KK,<,<,H,H,J() E") I=xHH I" I=xHH Is1+A#G*-+H* H3HH H?H::H?c$ dfd }|S)z?Attach the connection span while executing a connection method.c|jr;|jjr!tj|jdn t 5|g|i|cdddS#1swYyxYw)z#Connection span attacher decorator.F end_on_exitN)_spanr,ruse_spanr)rYargskwargsmethods r4wrapperz.with_connection_span_attached..wrappers` YY399113^^ II5 9D 0#/// 0 0 0s  A))A2)rYz,Union['MySQLConnection', 'CMySQLConnection']rgrrhrreturnr)rirjs` r4with_connection_span_attachedrms/0 90BE0QT0 0 Nr6c t|dr t|dn t|d}tjttj|j tj ttjtd|jji}|jt|dxsdtjj |g|5||i|cdddS#1swYyxYw)zInstruments the execution of `query_method`. A query span with a link to the corresponding connection span is generated. _connection_cnx cursor_typerz SQL statement)namekindlinks attributesN)rRgetattrrrDB_USER_user THREAD_IDr THREAD_NAMEr! __class____name__start_as_current_spanrArSpanKindCLIENT) query_methodtracerconnection_span_linkwrappedrgrh connectionquery_span_attributess r4_instrument_executionrs 7M * ' Wf %   ) 0 0  "3""$7w((11 #  % % Q ( ;O ^^ " "#$( & - T,V, ---s C##C,cVeZdZdZeddZd dZd fd Zd dZd dZ d dZ xZ S)BaseMySQLTracerz||_|jj}yr)rsql_mode)rr_s r4rzTracedMySQLConnection.__init__Ts3:  MM " "r6czt|jj|i||j|jS)Wraps the object method.)rrr)rrcursorrrers r4rzTracedMySQLConnection.cursor\s6 (DMM(($9&9<< JJ  r6c:|jj|i|S)r)rcmd_change_userrs r4rz%TracedMySQLConnection.cmd_change_userds -t}},,d=f==r6N)r'MySQLConnectionAbstract'rkr)rgrrhrrkrr)r|rrrrrrmrrlr6r4rrQs#M# #>#>r6rcTtj dfd }|S)z.Retrurn the instrumented version of `connect`.ctfdtDrtjd|iSt j dt }|jttjjt<|t<t jtd5}td||i}t||t| cdddS#1swYyxYw) zWraps the connection object returned by the method `connect`. Instrumentation for PooledConnections is not supported. c3&K|]}|v ywNrl).0keyrhs r4 z7_instrument_connect..wrapper..ys6sf}6sz4Instrumentation for pooled connections not supportedMySQL Connector/Pythoninstrumenting_module_nameinstrumenting_library_versiontracer_providerrrrsFrcNr)anyrrrWr get_tracerr start_spanrr~rr(r)rfr`r)rgrhrrZrYconnectrs ` r4rjz$_instrument_connect..wrapperqs 6 6 6 NNQ RD+F+ +!!&>*6+ #)"3"3%ENN,A,A#4# %+ !^^F?3 G 8 %dHf =4*6*C &c8V <(   s 1.C))C2)rgrrhrrkz9Union['MySQLConnectionAbstract', 'PooledMySQLConnection']) functoolswraps)rrrjs`` r4_instrument_connectrjs@__W&&!& B&&P Nr6cfeZdZUdZdZded<d dZd dZd dZ d ddZ d d Z dd Z y)MySQLInstrumentorz8MySQL instrumentation supporting mysql-connector-python.NzOptional[MySQLInstrumentor] _instancec|jCtj|g|i||_t|jdtj |jS)zQSinglenton. Restricts the instantiation to a singular instance. _original_connect)robject__new__setattrrr)clsrgrhs r4rzMySQLInstrumentor.__new__sI == "NN3@@@CM CMM#6 8I8I J}}r6cdtgS)zoReturn a list of python packages with versions that the will be instrumented (e.g., versions >= 8.1.0).zmysql-connector-python >= )r"rs r4instrumentation_dependenciesz.MySQLInstrumentor.instrumentation_dependenciess--D,EFGGr6c tjt|dk7rtjdyt t|d|j dt_y)aInstrument the library. Args: trace_module: reference to the 'trace' module from opentelemetry. tracer_provider (optional): TracerProvider instance. NOTE: Instrumentation for pooled connections not supported. rz3MySQL Connector/Python module already instrumented.Nr)rr)rrrvrrWrrNrrhs r4 instrumentzMySQLInstrumentor.instrumentsO   .A B B NNP Q /D"56"JJ'89 r6ct|trtjd|St |dr t |ds,tjd|j j |Stjdt|}|jttjj|_||_t!||jt|S) aEnable instrumentation in a MySQL connection. Args: connection: uninstrumented connection instance. trace_module: reference to the 'trace' module from opentelemetry. tracer_provider (optional): TracerProvider instance. Returns: connection: instrumented connection instace. NOTE: Instrumentation for pooled connections not supported. z Connection already instrumented.rerz+Instrumentation for class %s not supported.rrrr)r<rrrWrRr{r|rrrrrr~rrerr`)rrrrs r4instrument_connectionz'MySQLInstrumentor.instrument_connections" j"7 8 NN= > z7+7:y3Q NN=$$--  !!&>*6+  ",,%ENN,A,A- $ !*j.>.>?$Z88r6c tjt|dk(rtjdyt|dt_y)zUninstrument the library.rz5MySQL Connector/Python module already uninstrumented.N)rrrvrrWrs r4 uninstrumentzMySQLInstrumentor.uninstruments:   .A B B NNR S #D*=> r6cft|ds,tjd|jj|St |t stjd|S|jr;|jjr!|jjd|_|jS)aDisable instrumentation in a MySQL connection. Args: connection: instrumented connection instance. Returns: connection: uninstrumented connection instace. NOTE: Instrumentation for pooled connections not supported. rez-Uninstrumentation for class %s not supported.z"Connection already uninstrumented.N) rRrrWr{r|r<rrer,r8r)rrs r4uninstrument_connectionz)MySQLInstrumentor.uninstrument_connectionsz7+ NN?$$--  *&;< NN? @     0 0 = = ?    "#J """r6)rgrrhrrkr)rkzCollection[str])rhrrkrr)rrrOptional[trace.TracerProvider]rkr)rrrkr) r|rrrr__annotations__rrrrrrrlr6r4rrsaB-1I*1 H  (;?(9-(98(9 # (9T?#3# "#r6r)r2rr3zOptional[Exception]rkr)r2rrkr)r@r=rkr=r)rYz#Optional['MySQLConnectionAbstract']rZrr[zOptional[Dict[str, Any]]rkr)rir rkr )rr rrrz trace.Linkrrrgrrhrrkr )rHCallable[..., Union['MySQLConnectionAbstract', 'PooledMySQLConnection']]rrrkr)Jr __future__rrreabcrr contextlibrtypingrrr r r r r abstractsrrrrpoolingrconnection_cextr ImportErrorr;r constantsrrrversionr opentelemetryropentelemetry.sdk.traceropentelemetry.semconv.tracerOTEL_SYSTEM_AVAILABLEmysql.opentelemetry!mysql.opentelemetry.semconv.tracemissing_dependencies_errerrorsProgrammingErrorrrr r!r"r#r$r%r&r'r(r)compiler*rr5r9rAr`rmrrrrrrrlr6r4rs7# #"RRRI,/ 6 <"($6: "    '1bjj&??,0R# ,R#R#)R# R#j "-"- "-%"-# "-  "-  "-"-J+0c+0\/ / d>O>67;0 U030N0fv#v#C    " ( (-D % (// Y (  ((  (sH D*D"DD"E#(D76E#7E<EEE#"E#