ɯeiKzddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZdd lmZdd lmZmZdd lmZmZdd lmZej:eZdZ dZ!dZ"edddgZ#edgdZ$edgdZ%GddZ&GddZ'e'jPjRe'jTjRe'jVjRe'jXjRe'jZjRfZ.Gdde%Z/Gdde/Z0Gdd e/Z1Gd!d"Z2y)#) annotationsN) namedtuple)Queue)Lock)Any)OK) CLIENT_NAMESNOWFLAKE_CONNECTOR_VERSION)SecretDetector)TelemetryFieldgenerate_telemetry_data_dict)ENABLE_TELEMETRY_LOGrt_plain_logger)requests  TelemetryAPIurlapi_keyTelemetryServer)namerrTelemetryEventBasertagsurgentvaluecHeZdZeddZeddZeddZy) TelemetryAPIEndpointz?https://sfctest.client-telemetry.snowflakecomputing.com/enqueue(rRNY3EPNsB4U89XYuqsZKa7TSxb9QVX93yNM4tS6)rrz>https://sfcdev.client-telemetry.snowflakecomputing.com/enqueue(kyTKLWpEZSaJnrzTZ63I96QXZHKsgfqbaGmAaIWfz7https://client-telemetry.snowflakecomputing.com/enqueue(wLpEKqnLOW9tGNwTjab5N611YQApOb3t9xOnE1rXN)__name__ __module__ __qualname__rSFCTESTSFCDEVPRODc/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/snowflake/connector/telemetry_oob.pyrr!s6 M:G L:F  E: Dr*rceZdZedej j ej jZedej j ej jZ edejj ejjZ edejj ejjZ edejj ejjZ y)TelemetryServerDeploymentsdevregqa1preprod3prodN)r#r$r%rrr&rrDEVREGr'QA1PREPROD3r(r)r*r+r-r-0s  #++//1E1M1M1U1U C  #++//1E1M1M1U1U C  #**..0D0K0K0S0S C(//335I5P5P5X5XH $))--/C/H/H/P/P Dr*r-c(eZdZdZdZdZdZdZy)TelemetryEventzBase class for log and metric telemetry events. This class has all of the logic except for the 'type' of the telemetry event. That must be defined by the child class. ct)zGets the telemetry event type.)NotImplementedErrorselfs r+get_typezTelemetryEvent.get_typeRs!!r*ct}|j|d<|j|d<|j|d<|j |d<|j t tjtjjtjjjdjd|jdd |S) z'Transform this event into a dictionary.NameUrgentValueTagsN)tzinfoz%Y-%m-%d %H:%M:%Sr)UUID Created_OnType SchemaVersion)dictrrr generate_tagsupdatestruuiduuid4datetimenowtimezoneutcreplacestrftimer=)r<events r+to_dictzTelemetryEvent.to_dictVs f ++hg**,f  DJJL)&//33H4E4E4I4IJ%-. !"    r*c~|j}|r/|D]*}|jdddk(s|jdcSy)z9Gets the deployment field specified in tags if it exists.r?N deploymentrAUnknown)rget)r<rtags r+get_deploymentzTelemetryEvent.get_deploymentisDyy  ,7764(L8777++ ,r*cNt}|jrbt|jdkDrJ|jjD]-\}}| t ||t |j </t j}t|tjj<t t|tjj<|jj|tj j<|j#|tj$j<|j&ret|j&dkDrM|j&jD]0\}}| t ||dt |j z<2|S)z^Generates the tags to send as part of the telemetry event. Parts of the tags are user defined.rctx_)rHrlenitemsrKlowerTelemetryService get_instancer r KEY_OOB_DRIVERrr KEY_OOB_VERSIONrWr#KEY_OOB_TELEMETRY_SERVER_DEPLOYMENTget_connection_stringKEY_OOB_CONNECTION_STRINGcontext)r<rkv telemetrys r+rIzTelemetryEvent.generate_tagsssMv 99TYY!+ ) 21=+.q6DQ( 2%113 4?^ * * 0 01589T5U^ + + 1 12  % % ^ ? ? E EF  + + - ^ 5 5 ; ;<   Y%6%6!7!!;!))//1 ;1=47FD#a&,,.01 ; r*N)r#r$r%__doc__r=rUr[rIr)r*r+r8r8Ks "&r*r8ceZdZddZy)TelemetryLogEventcy)NLogr)r;s r+r=zTelemetryLogEvent.get_typesr*NreturnrKr#r$r%r=r)r*r+rnrnsr*rnceZdZddZy)TelemetryMetricEventcy)NMetricr)r;s r+r=zTelemetryMetricEvent.get_typesr*Nrqrsr)r*r+rurusr*ruc(eZdZdZeZed dZd!dZd!dZ e d"dZ d!dZ d!dZ e dZe d#d Zej d!d Ze d#d Zej d!d Ze d Zej d!dZe d$dZej d!dZe d%dZej d!dZd"dZdZd!dZd!dZd!dZd!dZ d& d'dZ d( d'dZ d) d*dZd!dZdZd!dZ dZ!y)+raNc|j5|j|ddd|jS#1swY|jSxYw)zStatic access method.N)_TelemetryService__lock_init_TelemetryService__instance)clss r+rbzTelemetryService.get_instancesB__ ~~% ~~ ~~s 5A ctj td|t_d|_t |_t |_t|_ t|_ t|_ tj|_y)zVirtually private constructor.NzThis class is a singleton!F)rar{ Exception_enabledr_queueDEFAULT_BATCH_SIZE batch_size)DEFAULT_NUM_OF_RETRY_TO_TRIGGER_TELEMETRY!num_of_retry_to_trigger_telemetryrHrhconnection_paramsr-r(rWr;s r+__init__zTelemetryService.__init__sg  & & 289 9*.  ' g , 5 .v !%499r*cD |jy#t$rYywxYw)zDTries to flush all events left in the queue. Ignores all exceptions.N)closer~r;s r+__del__zTelemetryService.__del__s"  JJL   s  cy)z0Whether the Telemetry service is enabled or not.Fr)r;s r+enabledzTelemetryService.enabledsr*cd|_y)zEnable Telemetry Service.FNrr;s r+enablezTelemetryService.enable  r*cd|_y)zDisable Telemetry Service.FNrr;s r+disablezTelemetryService.disablerr*c|jS)z5Get the queue that holds all of the telemetry events.)rr;s r+queuezTelemetryService.queues{{r*c|jS)z.Returns the context of the current connection._contextr;s r+rhzTelemetryService.contexts}}r*c||_y)z+Sets the context of the current connection.Nrr<rs r+rhzTelemetryService.contexts  r*c|jS)z>Returns the connection parameters from the current connection._connection_paramsr;s r+rz"TelemetryService.connection_paramss&&&r*c||_y)z;Sets the connection parameters from the current connection.Nrrs r+rz"TelemetryService.connection_paramss #(r*c|jS)z-Returns the batch size for uploading results. _batch_sizer;s r+rzTelemetryService.batch_sizer*c||_y)z*Sets the batch size for uploading results.Nrrs r+rzTelemetryService.batch_size !r*c|jS)zFReturns the number of HTTP retries before we submit a telemetry event."_num_of_retry_to_trigger_telemetryr;s r+rz2TelemetryService.num_of_retry_to_trigger_telemetrys666r*c||_y)zCSets the number of HTTP retries before we submit a telemetry event.Nrrs r+rz2TelemetryService.num_of_retry_to_trigger_telemetrys 38/r*c|jS)zHReturns the deployment that we are sending the telemetry information to. _deploymentr;s r+rWzTelemetryService.deploymentrr*c||_y)zESets the deployment that we are sending the telemetry information to.Nrrs r+rWzTelemetryService.deploymentrr*c:|jjtvS)z2Returns whether or not this deployment is enabled.)rWrENABLED_DEPLOYMENTSr;s r+is_deployment_enabledz&TelemetryService.is_deployment_enableds##':::r*c|jjdddz|jjddzdzt|jjddzS)z-Returns the URL used to connect to Snowflake.protocolz://host:port)rrYrKr;s r+rfz&TelemetryService.get_connection_stringsn  " " & &z2 6 $$((4 5 $((,,VR89  : r*c|jsy|jj||jj|jkDs |j r%|j }|y|j|yy)zdAdds a telemetry event to the queue. If the event is urgent, upload all telemetry events right away.N)rrputqsizerrexport_queue_to_string_upload_payload)r<rTpayloads r+addzTelemetryService.addse||  u ::    /5<<113G   ) 4@r*c|jsy|jjs%|j}|y|j |yy)zJFlushes all telemetry events in the queue and submit them to the back-end.N)rremptyrr)r<rs r+flushzTelemetryService.flushsE|| zz!113G   ) "r*c|j|t|_|jD]#\}}d|vs d|vsd|vs||j|<%y)zKUpdates the telemetry service context. Remove any passwords or credentials.passwordpasscode privateKeyN)configure_deploymentrHrhr_)r<rkeyrs r+update_contextzTelemetryService.update_context%s\ !!"34v +113 *JC#%c) +$) S!  *r*c*||_|jjdr|jjdnd}|jjdr|jjdnd}|jjdd}tj}d|vsd|vr-tj}|dk(rLtj }||_yd |vsd |vrtj }||_yd |vrtj}||_y) zEDetermines which deployment we are sending Telemetry OOB messages to.accountrrrNr/localir0r1) rrYr-r(r4r3r5r6rW)r<rrrrrWs r+rz%TelemetryService.configure_deployment2s !2%%)))4  " " & &y 1 %%))&1  " " & &v . %%))&$7044 D=GtO377Jt|7;; % d]ew.377J%4 3<#K#K#Q#QR*$#&DDJJ<6D@J^66<<=-#$v^ #)* H LL7$L G HsBB.. CCc | t} |jr!d} ttjj |d|tj j |tjj |id}|r|j|tjj <t|j|tjj <|jr2t|j} | |tjj <|r&t||tjj <|r&t||tj j <| r&t| |tj"j <| r| |tj$j <| t} | | tjj <t|| tj j <|| tjj <t'|| || }|j)|yy#t*$rt,j/ddYywxYw) zCLogs an HTTP Request error and adds it to the queue to be uploaded.N T) from_dictis_oob_telemetryrrrrz Failed to log HTTP request errorr)rHrrr KEY_OOB_REQUESTrKEY_OOB_SQL_STATEKEY_OOB_ERROR_CODEjsonKEY_OOB_RESPONSErKreasonKEY_OOB_RESPONSE_STATUS_LINE status_codeKEY_OOB_RESPONSE_STATUS_CODEKEY_OOB_RETRY_TIMEOUTKEY_OOB_RETRY_COUNTrrrnrr~rr)r<rrmethodsqlstateerrnoresponse retry_timeout retry_countrrrrresponse_status_coderrs r+log_http_request_errorz'TelemetryService.log_http_request_errorts0 <6D9 L||')$!=&66<<#>O&88>>&99?? &* " #>#B#B#H#HI HOO,#&CCII ++/283G3G/H,1'*GGMM!QT%RN>#G#G#M#MNOR#PN>#E#E#K#KLI#>#K#K#Q#QR$#&DDJJ<6D)^@@FFG@C8}^55;;<@E^66<<=-#$nV #kl L LL;dL K LsH-H== I I c| t} |jr!t||||}|j|yy#t$rt j ddYywxYw)z2Sends any type of exception through OOB telemetry.NrzFailed to log general exceptionTr)rHrrnrr~rr)r<rrrrrs r+log_general_exceptionz&TelemetryService.log_general_exceptionsg <6D K||-#$nV #   K LL:TL J Ks,< AAcd}d} |jsBtjd|jj tjd|ytjd|t rt jd|tj5}d|jjd}|j|jj||t }|jtk(rWtj |j"j%d d tk(r!tjd |jn"tjd |jd}dddtjd|y#1swY xYw#t&$r,}tjd|t)|d}Yd}~Sd}~wwxYw#tjd|wxYw)z{Uploads the JSON-formatted string payload to the telemetry backend. Ignore any exceptions that may arise. TNz Skip the disabled deployment: %szTelemetry request success=%sz%Sending OOB telemetry data. Payload: z!OOB telemetry data being sent is zapplication/json)z Content-typez x-api-key)dataheaderstimeout statusCoderz$telemetry server request success: %dz"telemetry server request error: %dFz?Telemetry request failed, Exception response: %s, exception: %s)rrrrWrrrrSessionrpostrREQUEST_TIMEOUTrr rloadstextrYr~rK)r<rsuccessrsessionres r+rz TelemetryService._upload_payloads ) B--/ ?AUAUVL LL7 AK LL@ J K#%%(I'&ST!!# $w$6!%!8!8#<<OO'' #+ (((B. 8==155lAF"LLL>@T@TLLR>R$G- $> LL7 A? $ $.  LLQA  G   LL7 AsI:F A F "CF5F FF G)"G GGGG0ct}|jjsR|j|jj j |jjsR t j|}tj|\}}}|S#t$r(tjdt|zdd}YKwxYw)zQExports all events in the queue into a JSON formatted string with secrets masked.ziFailed to generate a JSON dump from the passed in telemetry OOB events. String representation of logs: %sTrN)listrrappendrYrUrdumpsr~rrrKr mask_secrets)r<logsr_ masked_texts r+rz'TelemetryService.export_queue_to_stringsv++##% KK )113 4++##% jj&G+77@;  LL{d)   G  s8B((.CCcD|j|jy)zCloses the telemetry service.N)rrr;s r+rzTelemetryService.closes  r*c6|jjS)zReturns the size of the queue.)rrr;s r+sizezTelemetryService.sizeszz!!r*)rrra)rrNone)rrbool)rrzdict[str, Any])rrint)r<rrrz Any | None)NNNF)rr rrr )NNNNNNF)NF) rrKrrHrz dict | Nonerz bool | Nonerrr )"r#r$r%r{rrz classmethodrbrrpropertyrrrrrhsetterrrrrWrrfrrrrrrrrrrr r)r*r+rarasJ&K:  ^^''((  !!77'--8.8  !!;  * * *%@ $H$H $HZ KLKL KLb!# KKK K  K  K(0Bd$ "r*ra)3 __future__rrNrloggingrL collectionsrrr threadingrtypingrcompatr descriptionr r secret_detectorr rkr r test_utilrrvendoredr getLoggerr#rrrrrrrrr-r3rr4r5r6r(rr8rnrurar)r*r+rs""  "A+C<   8 $,.).5)*<= .0JK=  &""''""''""'''',,##(( @'@F > H"H"r*