ʯei? ddlmZddlmZddlZddlmZddl Z ddl m Z d dZ e eddf d d Ze eddf dd Ze e j dd Zy)) annotations) timedeltaN)extracdtjvr6tjd}|jtjd<tjdS)z Get or create a Snowpark session from the Streamlit connection. Returns: sp.Session: The Snowpark session object snowpark_session snowflake)st session_state connectionsession)r s g/var/www/html/glpi_dashboard/venv/lib/python3.12/site-packages/streamlit_extras/snowflake/connection.py _get_sessionr sL!1!11]];/ /9/A/A/C+,  . //)hoursTctj|dd}||}|r)|jjj |_|S)a Execute a SQL query and cache the results. Args: query (str): The SQL query to execute ttl (timedelta | int | None): Time-to-live for the cache. Defaults to 2 hours. Set to None to use the default cache invalidation. lowercase_columns (bool): Whether to convert column names to lowercase. Defaults to True. Returns: pd.DataFrame: The query results as a pandas DataFrame ttlcPtj|jSN)rsql to_pandas)querys r_run_sqlzrun_sql.._run_sql,s~!!%(2244r)rstrreturn pd.DataFrame)r cache_datacolumnsrlower)rrlowercase_columnsrdfs rrun_sqlr#sJ&]]s55 %BZZ^^))+ Irctj| dd}|jjdj}||||S)a Convert a Snowpark DataFrame to a pandas DataFrame and cache the result. Args: df (sp.DataFrame): The Snowpark DataFrame to convert ttl (timedelta | int | None): Time-to-live for the cache. Defaults to 2 hours. Set to None to use the default cache invalidation. lowercase_columns (bool): Whether to convert column names to lowercase. Defaults to True. Returns: pd.DataFrame: The converted pandas DataFrame with cached results rc|}|j}|r)|jjj|_|Sr)rrrr )_dfrr!_r"s r _run_snowparkz#run_snowpark.._run_snowparkKs4  ]]_ --/BJ rr)r!)r& sp.DataFramerrr!boolrr)r r_planqueriesr)r"rr!r(rs r run_snowparkr-8sk&]]s   "% :>     HH  Q  # #E U6G HHrc4tj|S)a3 Get a Snowpark table for use in building a query. Args: table_name (str): Name of the table to retrieve Returns: sp.Table: A cached Snowpark Table object that can be used for querying. The result is cached so that metadata is not re-fetched from the database. )rtable) table_names r get_tabler1\s >   ++r)rz sp.Session)rrrtimedelta | int | Noner!r*rr)r"r)rr2r!r*rr)r0rrzsp.Table) __future__rdatetimerpandaspdsnowflake.snowparksnowparksp streamlitr rrr#r-cache_resourcer1rrr>s" 0#,!"4"   <#,!"4" I I  I I I IF , ,r