%ex/ddlmZddlmZddlmZddlmZddlm Z m Z m Z m Z m Z ddlZddlmZddlmZdd lmZdd lmZe rdd lmZdd lmZdd lmZhdZhdZ GddedZ!dS)) annotations)ChainMap)deepcopy) timedelta) TYPE_CHECKINGListOptionalUnioncastN)BaseConnection)extract_from_dict)StreamlitAPIException) cache_data) Connection)EngineSession>urlhostportdriverdialectdatabasepasswordusername>rrrceZdZdZdddZdd d d d d d dZd!dZed"dZed#dZ ed$dZ d#dZ d S)% SQLConnectionaA connection to a SQL database using a SQLAlchemy Engine. Initialize using ``st.connection("", type="sql")``. SQLConnection provides the ``query()`` convenience method, which can be used to run simple read-only queries with both caching and simple error handling/retries. More complex DB interactions can be performed by using the ``.session`` property to receive a regular SQLAlchemy Session. SQLConnections should always be created using ``st.connection()``, **not** initialized directly. Connection parameters for a SQLConnection can be specified using either ``st.secrets`` or ``**kwargs``. Some frequently used parameters include: - **url** or arguments for `sqlalchemy.engine.URL.create() `_. Most commonly it includes a dialect, host, database, username and password. - **create_engine_kwargs** can be passed via ``st.secrets``, such as for `snowflake-sqlalchemy `_ or `Google BigQuery `_. These can also be passed directly as ``**kwargs`` to connection(). - **autocommit=True** to run with isolation level ``AUTOCOMMIT``. Default is False. Example ------- >>> import streamlit as st >>> >>> conn = st.connection("sql") >>> df = conn.query("select * from pet_owners") >>> st.dataframe(df) F autocommitboolreturn'Engine'c Tddl}t|}tt|}t ||j}t|stdd|vr!|j |d}ntD]}||vrtd||dd|vr d|dndz}|j j ||d |d |d d |vrt|d nd|d }t ||jdi} |j|fi| } |r$t#d| dSt#d| S)NrzvMissing SQL DB connection configuration. Did you forget to set this in `secrets.toml` or as kwargs to `st.connection`?rz!Missing SQL DB connection param: rr+rrrrr) drivernamerrrrrcreate_engine_kwargsr AUTOCOMMIT)isolation_level) sqlalchemyrr _ALL_CONNECTION_PARAMSr_secretsto_dictlenrenginemake_url_REQUIRED_CONNECTION_PARAMSURLcreategetint create_enginer execution_options) selfrkwargsr)conn_param_kwargs conn_paramsrpr%r&engs Dlib/python3.11/site-packages/streamlit/connections/sql_connection.py_connectzSQLConnection._connectOs&!!-.DfMM0$-2G2G2I2IJJ ; '`  K  #,,[-?@@CC0 Y YK''/0WTU0W0WXXX(%Y//7;/F/F+K)+++BJ#'..%$Z0$44 (17;1F1FSV,---D$44 /C ( DM%%&`_, is supported. Default is None. **kwargs: dict Additional keyword arguments are passed to |pandas.read_sql|_. .. |pandas.read_sql| replace:: ``pandas.read_sql`` .. _pandas.read_sql: https://pandas.pydata.org/docs/reference/api/pandas.read_sql.html Returns ------- pandas.DataFrame The result of running the query, formatted as a pandas DataFrame. Example ------- >>> import streamlit as st >>> >>> conn = st.connection("sql") >>> df = conn.query("select * from pet_owners where owner = :owner", ttl=3600, params={"owner":"barbara"}) >>> st.dataframe(df) r)text) DatabaseError InternalErrorOperationalError)retryretry_if_exception_typestop_after_attempt wait_fixedc,S)N)reset)_r7s r=z%SQLConnection.query..sDJJLLr?T)afterstopreraiserQwaitNrErFr rKcvj}tj||f|||d|S)NrBrCrD) _instanceconnectpdread_sql)rErBrCrDr8instancer7rMs r=_queryz#SQLConnection.query.._querysY ~--//H;S $#    r?.rW)r@rAr`)NNN)rErFr rK)r)rMsqlalchemy.excrNrOrPtenacityrQrRrSrTrFreplace __qualname___connection_namer)r7rEr@rArBrCrDr8rNrOrPrQrRrSrTrfttl_strrMs` @r=queryzSQLConnection.queryzsH $#####QQQQQQQQQQ             ((((##A&&)) /?@A              *   '#s   "(!4XXt7LXXwXX %     v         r?'SQLAlchemyConnection'c4|jS)a_Call ``.connect()`` on the underlying SQLAlchemy Engine, returning a new ``sqlalchemy.engine.Connection`` object. Calling this method is equivalent to calling ``self._instance.connect()``. NOTE: This method should not be confused with the internal ``_connect`` method used to implement a Streamlit Connection. )rarbr7s r=rbzSQLConnection.connects~%%'''r?c|jS)zgThe underlying SQLAlchemy Engine. This is equivalent to accessing ``self._instance``. )rarqs r=r.zSQLConnection.engines ~r?c|jjS)zThe name of the driver used by the underlying SQLAlchemy Engine. This is equivalent to accessing ``self._instance.driver``. )rarrqs r=rzSQLConnection.driver s ~$$r? 'Session'c.ddlm}||jS)aPReturn a SQLAlchemy Session. Users of this connection should use the contextmanager pattern for writes, transactions, and anything more complex than simple read queries. See the usage example below, which assumes we have a table ``numbers`` with a single integer column ``val``. The `SQLAlchemy `_ docs also contain much more information on the usage of sessions. Example ------- >>> import streamlit as st >>> conn = st.connection("sql") >>> n = st.slider("Pick a number") >>> if st.button("Add the number!"): ... with conn.session as session: ... session.execute("INSERT INTO numbers (val) VALUES (:n);", {"n": n}) ... session.commit() rr)sqlalchemy.ormrra)r7rs r=sessionzSQLConnection.sessions(, +*****wt~&&&r?c .t|dd}t|j}t|jr d|jdnd}|j5}|j|jjj nd}dddn #1swxYwYd|jd|d|d |d |d S) N __module__z - Configured from `[connections.z]`r$unknownz --- **st.connection z built from `rgz`** z - Dialect: `z%` - Learn more using `st.help()` --- ) getattrtype__name__r-r+rlrwbindrname)r7 module_name class_namecfgsrs r= _repr_html_zSQLConnection._repr_html_.s,dL$77 $ZZ( 4=!!  Ht/D H H H H \ OQ-.V-?afn))YG O O O O O O O O O O O O O O O&5@CM  sA55A9<A9)F)rrr r!) rErFr@rGrArHrBrIrCrJr rK)r ro)r r!)r rF)r rt) r}ryrk__doc__r>rnrbpropertyr.rrwrr?r=rr/s>)')')')')'^$?6:59#'z z z z z z x ( ( ( (X%%%X%'''X'8r?rr)" __future__r collectionsrcopyrdatetimertypingrrr r r pandasrcstreamlit.connectionsr streamlit.connections.utilr streamlit.errorsrstreamlit.runtime.cachingrsqlalchemy.enginerSQLAlchemyConnectionsqlalchemy.engine.baserrvrr*r0rrr?r=rsn#""""" ==============000000888888222222000000'DDDDDD------&&&&&&   >==SSSSSN8,SSSSSr?