ɯei#ddlmZddlZddlmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZdd lmZmZe r dd lmZdd lmZej.eZeGd d eZGddZGddZ d ddZy)) annotationsN)Enumunique)Lock) TYPE_CHECKINGAny) CLIENT_NAMESNOWFLAKE_CONNECTOR_VERSION)SecretDetector)ENABLE_TELEMETRY_LOGrt_plain_logger)SnowflakeConnection)SnowflakeRestfulceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5Z7d6Z8y7)8TelemetryField client_time_consume_first_resultclient_time_consume_last_resultclient_time_downloading_chunksclient_time_parsing_chunksclient_sql_exceptionclient_ocsp_exceptionclient_http_exceptionclient_get_partitions_used'client_pyformat_empty_seq_interpolationclient_fetch_pandas_allclient_fetch_pandas_batchesclient_fetch_arrow_allclient_fetch_arrow_batchesclient_write_pandasclient_imported_packagesmini_core_importclient_multi_statement_querytypesourcequery_id sql_state driver_typedriver_versionreasonvalue exception ErrorNumber ErrorMessage StacktracedriverversiontelemetryServerDeploymentconnectionStringexceptionMessage errorMessageexceptionStackTrace eventType errorCodesqlStaterequestresponseresponseStatusLineresponseStatusCode retryTimeout retryCount eventSubType sfcPeerHostcertIdocspRequestBase64ocspResponderURL insecureModefailOpen cacheEnabledcacheHitN)9__name__ __module__ __qualname__TIME_CONSUME_FIRST_RESULTTIME_CONSUME_LAST_RESULTTIME_DOWNLOADING_CHUNKSTIME_PARSING_CHUNKS SQL_EXCEPTIONOCSP_EXCEPTIONHTTP_EXCEPTIONGET_PARTITIONS_USEDEMPTY_SEQ_INTERPOLATIONPANDAS_FETCH_ALLPANDAS_FETCH_BATCHESARROW_FETCH_ALLARROW_FETCH_BATCHES PANDAS_WRITEIMPORTED_PACKAGES CORE_IMPORTMULTI_STATEMENTKEY_TYPE KEY_SOURCE KEY_SFQID KEY_SQLSTATEKEY_DRIVER_TYPEKEY_DRIVER_VERSION KEY_REASON KEY_VALUE KEY_EXCEPTIONKEY_ERROR_NUMBERKEY_ERROR_MESSAGEKEY_STACKTRACEKEY_OOB_DRIVERKEY_OOB_VERSION#KEY_OOB_TELEMETRY_SERVER_DEPLOYMENTKEY_OOB_CONNECTION_STRINGKEY_OOB_EXCEPTION_MESSAGEKEY_OOB_ERROR_MESSAGEKEY_OOB_EXCEPTION_STACK_TRACEKEY_OOB_EVENT_TYPEKEY_OOB_ERROR_CODEKEY_OOB_SQL_STATEKEY_OOB_REQUESTKEY_OOB_RESPONSEKEY_OOB_RESPONSE_STATUS_LINEKEY_OOB_RESPONSE_STATUS_CODEKEY_OOB_RETRY_TIMEOUTKEY_OOB_RETRY_COUNTKEY_OOB_EVENT_SUB_TYPEKEY_OOB_SFC_PEER_HOSTKEY_OOB_CERT_IDKEY_OOB_OCSP_REQUEST_BASE64KEY_OOB_OCSP_RESPONDER_URLKEY_OOB_INSECURE_MODEKEY_OOB_FAIL_OPENKEY_OOB_CACHE_ENABLEDKEY_OOB_CACHE_HIT_/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/snowflake/connector/telemetry.pyrrs5!C@>6*M,N,N6G08.O6(L2$K4OHJIL#O)JIM$&!NNO*E' 2 2*$9!$$"O!#7 #7 *&+)O"5!3*"*"rrcVeZdZdZdZdZd dZe d d dZd dZ d dZ y) TelemetryDataz>An instance of telemetry data which can be sent to the server.r rc ||_||_yNmessage timestamp)selfrrs r__init__zTelemetryData.__init__^s "rNc4|t|xsi|||S)ab Generate telemetry data with driver info from given dict and timestamp. It takes an optional connection object to read data from. It also takes a boolean is_oob_telemetry to indicate whether it's for out-of-band telemetry, as naming of keys for driver and version is different from the ones of in-band telemetry.  from_dict connectionis_oob_telemetry)generate_telemetry_data_dict)clsrrrrs rfrom_telemetry_data_dictz&TelemetryData.from_telemetry_data_dictbs+ ($?%!1     rcF|jt|jdS)Nr)rstrrrs rto_dictzTelemetryData.to_dictys<""rreturnNoneNF)rdictrintrSnowflakeConnection | Nonerbool)rdict[str, Any])rr) rIrJrK__doc__TRUEFALSEr classmethodrrrrrrrrXsaH D E# 26!&    /     ,K#rrcleZdZdZdZdZdddZddZddZdddZ e dd Z ddd Z dd Z d Zd Zy)TelemetryClientzFClient to enqueue and send metrics to the telemetry endpoint in batch.z/telemetry/senddNc~||_g|_|xstj|_t |_d|_y)NT)_rest _log_batchrDEFAULT_FORCE_FLUSH_SIZE _flush_sizer_lock_enabled)rrest flush_sizes rrzTelemetryClient.__init__s3.2 %Q)Q)QV  rcT|jr td|jstj dy|j 5|j j|dddt|j |jk\r|jyy#1swY=xYw)Nz3Attempted to add log when TelemetryClient is closedz'TelemetryClient disabled. Ignoring log.) is_closed Exceptionrloggerdebugrrappendlenr send_batchrtelemetry_datas radd_log_to_batchz TelemetryClient.add_log_to_batchs >>QR R LLB C  ZZ 3 OO " "> 2 3 t 4#3#3 3 OO  4 3 3s BB'ct |j|y#t$rtjddYywxYw)NzFailed to add log to telemetry.Texc_info)rrrwarningrs rtry_add_log_to_batchz$TelemetryClient.try_add_log_to_batchs8 M  ! !. 1 M NN>TU U LLF G  ZZ !ooG DO ! LL8 9 g667 7 I  ' 'D 21 5  ! !$H"O P R**$$11#) %Cy> +H !&  KLE ! !76 R!DM LLADL Q Rs+E*?E78AE<E<*E4<'F&%F&c|jduSr)rrs rrzTelemetryClient.is_closedszzT!!rcz|js/tjd|j|d|_yy)NzClosing telemetry client.)r)rrrrr)rrs rclosezTelemetryClient.closes0~~ LL4 5 OO%O (DJrcd|_yrrrs rdisablezTelemetryClient.disables  rc|jSrrrs r is_enabledzTelemetryClient.is_enableds }}rc,t|jSr)rrrs r buffer_sizezTelemetryClient.buffer_sizes4??##rr)rrrr)rrrr)F)rrrr)rrr)rIrJrKrrrrrrrpropertyrrrrrrrrrrsPP)" M ,R\"" $rrcp|xsi}|sotjjttjjt tj j|r|ji|Sti|Stjjttjjt i|S)a^ Generate telemetry data with driver info. The method also takes an optional dict to update from and optional connection object to read data from. It also takes a boolean is_oob_telemetry to indicate whether it's for out-of-band telemetry, as naming of keys for driver and version is different from the ones of in-band telemetry. ) rrar+r rbr r^ applicationrirjrs rrrsRI   * * 0 0+  - - 3 35P  % % + +*4 &&     ;F      ) ) / /  * * 0 02M  r)NNF)rz dict | Nonerrrrrr) __future__rloggingenumrr threadingrtypingrr descriptionr r secret_detectorr test_utilr rrrnetworkr getLoggerrIrrrrrrrrrs"%A+</)   8 $@#T@#@#F%#%#P_$_$F"-1"* r