ɯei-ddlmZddlmZmZmZmZddlZGddeZ GddZ Gdd e Z Gd d e Z y) )abstractmethod)DictList NamedTupleTupleNcNeZdZUdZeed<eed<dZeed<dZe ed<defdZ y) QueryRecordz[Contains the query information returned from the Snowflake database after the query is run.query_idsql_textN is_describe thread_idreturnc |j)|jd|jd|jdS|j6|j*d|jd|jd|jdS|j6|j*d|jd|jd|jdSd|jd|jd|jd|jd S)NzQueryRecord(query_id=z , sql_text=)z, is_describe=z , thread_id=)r r r r selfs b/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/snowflake/snowpark/query_history.py__repr__zQueryRecord.__repr__s    #(>*4==/T]]OSTU U    )dnn.D*4==/T]]OSabfbrbrasstu u    %$..*D*4==/T]]OS_`d`n`n_oopq q*4==/T]]OSabfbrbrassAEAOAO@PPQR R) __name__ __module__ __qualname____doc__str__annotations__r boolr intrrrr r s1eMMKIsR#Rrr c&eZdZededdfdZy) QueryListener query_recordrNc y)z notify query listener of a query event Args: query_record: record of the query to notify the listener of **kwargs: optional keyword arguments Returns: None Nrrr!kwargss r_notifyzQueryListener._notifys r)rrrrr r%rrrr r s"  K  d    rr c eZdZdZ ddddededededd f d Zd Zd Zd edd fdZ e de efdZ e defdZ e defdZe defdZe deee effdZy ) QueryHistoryzA context manager that listens to and records SQL queries that are pushed down to the Snowflake database. See also: :meth:`snowflake.snowpark.Session.query_history`. session"snowflake.snowpark.session.Sessioninclude_describeinclude_thread_id include_errorinclude_dataframe_profilingrNct||_g|_||_||_||_||_i|_i|_yN)r(_queries_include_describe_include_thread_id_include_error_include_dataframe_profiling_dataframe_queries_describe_queries)rr(r*r+r,r-s r__init__zQueryHistory.__init__2sB +- !1"3+,G)8:EGrc|Sr/rrs r __enter__zQueryHistory.__enter__E rcN|jjj|yr/r(_connremove_query_listenerrexc_typeexc_valexc_tbs r__exit__zQueryHistory.__exit__H 006rr!c |jrNd|vrJ|d}||jvrg|j|<|j|j|j|jj|y)Ndataframe_uuid)r4r5appendr r0)rr!r$df_uuids rr%zQueryHistory._notifyKsm  , ,6) !12$"9"9979D++G4''077 8M8MN \*rc|jSr/)r0rs rquerieszQueryHistory.queriesTs }}rc|jS)zKWhen True, QueryRecords for describe queries are recorded by this listener.)r1rs rr*zQueryHistory.include_describeXs%%%rc|jS)z@When True, thread id of the query are recorded by this listener.)r2rs rr+zQueryHistory.include_thread_id]&&&rc|jS)zRWhen True, queries that have error during execution are recorded by this listener.)r3rs rr,zQueryHistory.include_errorbs"""rc|jS)z2Returns a map of dataframe plan uuid to query ids.)r5rs rdataframe_querieszQueryHistory.dataframe_queriesgrMr)FFFF)rrrrrr7r9rCr r%propertyrrJr*r+r,rrrrPrrrr'r'+s"'"'#,1 H5HH H  H &* H H&7+K+d+k*&$&&'4''#t##'4T#Y#7''rr'creZdZ d dddeddfdZdZdZd eddfd Ze defd Z e de e fd Z y) AstListenerr(r)include_failuresrNc.||_g|_||_y)a  Initializes the AstListener. Args: session: The session to listen to. include_failures: When True, the listener will include failed queries in the history. This can be useful for debugging and testing. N)r( _ast_batches_include_failures)rr(rTs rr7zAstListener.__init__ns ')!1rc|Sr/rrs rr9zAstListener.__enter__r:rcN|jjj|yr/r<r?s rrCzAstListener.__exit__rDrr!c Jd|vr|jj|dyy)N dataframeAst)rVrGr#s rr%zAstListener._notifys( V #    $ $VN%; < $rc|jSr/)rWrs rrTzAstListener.include_failuress%%%rc|jSr/)rVrs rbase64_batcheszAstListener.base64_batchess   r)F)rrrrr7r9rCr r%rQrTrrr^rrrrSrSms"'2522  2"7=K=d=&$&&!S !!rrS) abcrtypingrrrrsnowflake.snowpark snowflaker r r'rSrrrrcsD00R*R&   ?'=?'D"!-"!r