&e0fUddlmZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z ddl mZmZddlmZmZmZmZddlmZddlmZdd lmZdd lmZdd lmZeeed Zej d Z!dddddddZ"de#d<e deeZ$ed d9d:d"Z%d;d#Z&e dd+Z'e d.__create_connectionOs  ??????z% is not a subclass of BaseConnection!._zRunning `st.connection(...)`.)r$ show_spinnerr&)r r!r"r#r(r) issubclassr rr!replace __qualname__r)r r"r$r&r-r/ttl_strs r._create_connectionr8?s @@@@ & 7 7 # F F F   #hh SG  +EEgEE EE$.4    t%5 @ @ @ @@r0cb|tvr t|Std|dt)NzInvalid connection 'z!'. Supported connection classes: )FIRST_PARTY_CONNECTIONSr)r"s r._get_first_party_connectionr;ksP222&'788  C/ C C)@ C C  r0FLiteral['sql'] autocommitboolrc dSNr,r r$r&r=r-s r.connection_factoryrBu  Dr0typec dSr@r,r rDr$r&r=r-s r.rBrB  Dr0Literal['snowflake']r c dSr@r,rAs r.rBrBrCr0c dSr@r,rFs r.rBrBrGr0Literal['snowpark']rc dSr@r,)r r$r&r-s r.rBrBs  Dr0c dSr@r,r rDr$r&r-s r.rBrBrCr0c dSr@r,rNs r.rBrBrCr0 str | NoneBaseConnection[Any]c dSr@r,rNs r.rBrBrCr0c d}||r)|t|d}tj|}|E|tvrt |}n,t jt d|d}|}t|trtd|vra| d}| } tj d|} t| | }nt |} t!||f||d|} t| t"rt%| ddd } | S#t&$r} t| } t)jt,| }d }|r5t.|d }|rd |d }t't| d|d} ~ wwxYw)aZ Create a new connection to a data store or API, or return an existing one. Config options, credentials, secrets, etc. for connections are taken from various sources: - Any connection-specific configuration files. - An app's ``secrets.toml`` files. - The kwargs passed to this function. Parameters ---------- name : str The connection name used for secrets lookup in ``[connections.]``. Type will be inferred from passing ``"sql"``, ``"snowflake"``, or ``"snowpark"``. type : str, connection class, or None The type of connection to create. It can be a keyword (``"sql"``, ``"snowflake"``, or ``"snowpark"``), a path to an importable class, or an imported class reference. All classes must extend ``st.connections.BaseConnection`` and implement the ``_connect()`` method. If the type kwarg is None, a ``type`` field must be set in the connection's section in ``secrets.toml``. max_entries : int or None The maximum number of connections to keep in the cache, or None for an unbounded cache. (When a new entry is added to a full cache, the oldest cached entry will be removed.) The default is None. ttl : float, timedelta, or None The maximum number of seconds to keep results in the cache, or None if cached results should not expire. The default is None. **kwargs : any Additional connection specific kwargs that are passed to the Connection's ``_connect()`` method. Learn more from the specific Connection's documentation. Returns ------- Connection object An initialized Connection object of the specified type. Examples -------- The easiest way to create a first-party (SQL, Snowflake, or Snowpark) connection is to use their default names and define corresponding sections in your ``secrets.toml`` file. >>> import streamlit as st >>> conn = st.connection("sql") # Config section defined in [connections.sql] in secrets.toml. Creating a SQLConnection with a custom name requires you to explicitly specify the type. If type is not passed as a kwarg, it must be set in the appropriate section of ``secrets.toml``. >>> import streamlit as st >>> conn1 = st.connection("my_sql_connection", type="sql") # Config section defined in [connections.my_sql_connection]. >>> conn2 = st.connection("my_other_sql_connection") # type must be set in [connections.my_other_sql_connection]. Passing the full module path to the connection class that you want to use can be useful, especially when working with a custom connection: >>> import streamlit as st >>> conn = st.connection("my_sql_connection", type="streamlit.connections.SQLConnection") Finally, you can pass the connection class to use directly to this function. Doing so allows static type checking tools such as ``mypy`` to infer the exact return type of ``st.connection``. >>> import streamlit as st >>> from streamlit.connections import SQLConnection >>> conn = st.connection("my_sql_connection", type=SQLConnection) zenv:N connectionsrDr1)r$r&zconnection("snowpark")zconnection("snowflake")z 2024-04-01z@You may be missing a dependency required to use this connection.zYou need to install the 'z!' package to use this connection.z. ) startswithlenosenvironr:r;rload_if_toml_exists isinstancer!splitpop importlib import_modulejoingetattrr8rrModuleNotFoundErrorresearchMODULE_EXTRACTION_REGEXrgetgroup)r rDr$r&r-USE_ENV_PREFIX envvar_namer"parts classnameconnection_moduleconne err_stringmissing_module extra_info pypi_packages r.rBrBs!TN ~&&'3~..001 z+& | * * */t44DD  1 3 3 3$]3D9&AD "C(( M " " "$**3//E I ) 7 H H &'8)DD  :;KLL =! " 0;  HN   d. / / %() D   = = =VV #:JGGW  i3778L8LQ8O8OPPL ihhhh !SVV";";z";";<<< =s9E G BGG )NN) r r!r"r#r$r%r&r'r(r)r"r!)NNF) r r<r$r%r&r'r=r>r(r) r r!rDr<r$r%r&r'r=r>r(r) r rHr$r%r&r'r=r>r(r ) r r!rDrHr$r%r&r'r=r>r(r )r rKr$r%r&r'r(r) r r!rDrKr$r%r&r'r(r) r r!rDr#r$r%r&r'r(r)NNN) r r!rDrPr$r%r&r'r(rQ)( __future__rr^rXrcdatetimertypingrrrrr typing_extensionsr r streamlit.connectionsr r rrstreamlit.deprecation_utilrstreamlit.errorsrstreamlit.runtime.cachingrstreamlit.runtime.metrics_utilrstreamlit.runtime.secretsrr:compilerer__annotations__rr8r;rBr,r0r.rs#"""""" 55555555555555,,,,,,,, :99999222222444444999999777777%"  %"*%@AA!-75 33'+>#3FGGG #$( (A(A(A(A(AV #$(        #$(        #$(        #$(        #$(       #$(        #$(        "$(         I=I=I=I=I=I=r0