+hh ddlZddlmZddlmZddlmZddlmZddlm Z ddlmZdd lm Z dd lm Z dd l m Zdd lmZdd lmZddlmZddlmZdZd%dZGddZe j2eddgggdGddeeeZe j2eddggdgdGddeeZGd d!eeZd"Ze j<ed#Ze j<ed$Z y)&N)exc)ProxyComparable)StartableContext)_ensure_sync_result) AsyncResult) inspection)util) create_engine)NestedTransaction) Connection)Engine)greenlet_spawnc|jddrtjdd|d<t|i|}t |S)aECreate a new async engine instance. Arguments passed to :func:`_asyncio.create_async_engine` are mostly identical to those passed to the :func:`_sa.create_engine` function. The specified dialect must be an asyncio-compatible dialect such as :ref:`dialect-postgresql-asyncpg`. .. versionadded:: 1.4 server_side_cursorsFz}Can't set server_side_cursors for async engine globally; use the connection.stream() method for an async streaming result setTfuture)get async_excAsyncMethodRequired_create_engine AsyncEngine)argkw sync_engines P/var/www/html/venv/lib/python3.12/site-packages/sqlalchemy/ext/asyncio/engine.pycreate_async_enginersP vv#U+++ #  BxL #,,K { ##c |jDcic]%\}}|j|r|t|d|'}}}d|d<|j||j d}t |fi|Scc}}w)aCreate a new AsyncEngine instance using a configuration dictionary. This function is analogous to the :func:`_sa.engine_from_config` function in SQLAlchemy Core, except that the requested dialect must be an asyncio-compatible dialect such as :ref:`dialect-postgresql-asyncpg`. The argument signature of the function is identical to that of :func:`_sa.engine_from_config`. .. versionadded:: 1.4.29 NT_coerce_configurl)items startswithlenupdatepopr) configurationprefixkwargskeyvalueoptionsr!s rasync_engine_from_configr-/s(--/ C >>& ! CKME!G !%G  NN6 ++e C s .g ..s*A4ceZdZdZy)AsyncConnectable)_slots_dispatch __weakref__N)__name__ __module__ __qualname__ __slots__rrr/r/Fs0Irr/z:class:`_future.Connection`z!:class:`_asyncio.AsyncConnection`)closed invalidateddialectdefault_isolation_level) classmethodsmethods attributesceZdZUdZdZd$dZeed< eed< e dZ d%dZ e d Z d Ze d Ze d Zd ZdZdZd$dZdZdZdZdZdZdZdZdZdZdZdej@fdZ!dej@fdZ"dej@fdZ#dej@fdZ$dej@fdZ%dej@fd Z&d!Z'd"Z(d#Z)y)&AsyncConnectionaAn asyncio proxy for a :class:`_engine.Connection`. :class:`_asyncio.AsyncConnection` is acquired using the :meth:`_asyncio.AsyncEngine.connect` method of :class:`_asyncio.AsyncEngine`:: from sqlalchemy.ext.asyncio import create_async_engine engine = create_async_engine("postgresql+asyncpg://user:pass@host/dbname") async with engine.connect() as conn: result = await conn.execute(select(table)) .. versionadded:: 1.4 )enginersync_connectionNc`||_|j|_|j||_yN)r@r_assign_proxiedrA)self async_enginerAs r__init__zAsyncConnection.__init__ss*" '33#33ODrrArcTttj|j|SrC)r?r_retrieve_proxy_for_targetr@clstargets r_regenerate_proxy_for_targetz,AsyncConnection._regenerate_proxy_for_targets#  2 26== A6  rcK|jrtjd|jt |j j d{|_|S7w)zzStart this :class:`_asyncio.AsyncConnection` object's context outside of using a Python ``with:`` block. zconnection is already startedN)rArInvalidRequestErrorrDrrconnectrE is_ctxmanagers rstartzAsyncConnection.startsY   ))*IJ J#33!$"2"2":":; <   =sAA#A!A#c,tjd)zeNot implemented for async; call :meth:`_asyncio.AsyncConnection.get_raw_connection`. zAsyncConnection.connection accessor is not implemented as the attribute may need to reconnect on an invalidated connection. Use the get_raw_connection() method.)rrOrEs r connectionzAsyncConnection.connections %% 3  rc`K|j}tt|dd{S7w)aReturn the pooled DBAPI-level connection in use by this :class:`_asyncio.AsyncConnection`. This is a SQLAlchemy connection-pool proxied connection which then has the attribute :attr:`_pool._ConnectionFairy.driver_connection` that refers to the actual driver connection. Its :attr:`_pool._ConnectionFairy.dbapi_connection` refers instead to an :class:`_engine.AdaptedConnection` instance that adapts the driver connection to the DBAPI protocol. rVN)_sync_connectionrgetattrrEconns rget_raw_connectionz"AsyncConnection.get_raw_connections+$$&#GT<@@@@s %.,.c|jSrC)rArUs r_proxiedzAsyncConnection._proxieds###rc.|jjS)aReturn the :attr:`_engine.Connection.info` dictionary of the underlying :class:`_engine.Connection`. This dictionary is freely writable for user-defined state to be associated with the database connection. This attribute is only available if the :class:`.AsyncConnection` is currently connected. If the :attr:`.AsyncConnection.closed` attribute is ``True``, then accessing this attribute will raise :class:`.ResourceClosedError`. .. versionadded:: 1.4.0b2 )rAinforUs rr`zAsyncConnection.infos ##(((rcR|js|j|jSrC)rA_raise_for_not_startedrUs rrXz AsyncConnection._sync_connections###  ' ' )###rc8|jt|S)z1Begin a transaction prior to autobegin occurring.rXAsyncTransactionrUs rbeginzAsyncConnection.begins %%rc<|jt|dS)z;Begin a nested transaction and return a transaction handle.T)nestedrdrUs r begin_nestedzAsyncConnection.begin_nesteds T22rclK|j}t|j|d{S7w)zInvalidate the underlying DBAPI connection associated with this :class:`_engine.Connection`. See the method :meth:`_engine.Connection.invalidate` for full detail on this method. ) exceptionN)rXr invalidate)rErkr[s rrlzAsyncConnection.invalidates-$$&#DOOyIIIIs +424chK|j}t|jd{S7wrCrXrget_isolation_levelrZs rroz#AsyncConnection.get_isolation_level+$$&#D$<$<==== )202chK|j}t|jd{S7wrCrnrZs rset_isolation_levelz#AsyncConnection.set_isolation_levelrprqcB|j}|jSzYReturn True if a transaction is in progress. .. versionadded:: 1.4.0b2 )rXin_transactionrZs rrvzAsyncConnection.in_transactions!$$&""$$rcB|j}|jSru)rXin_nested_transactionrZs rrxz%AsyncConnection.in_nested_transactions! $$&))++rcr|j}|j}|tj|Sy)aReturn an :class:`.AsyncTransaction` representing the current transaction, if any. This makes use of the underlying synchronous connection's :meth:`_engine.Connection.get_transaction` method to get the current :class:`_engine.Transaction`, which is then proxied in a new :class:`.AsyncTransaction` object. .. versionadded:: 1.4.0b2 N)rXget_transactionrerIrEr[transs rrzzAsyncConnection.get_transaction s:$$&$$&  #>>uE Ercr|j}|j}|tj|Sy)aReturn an :class:`.AsyncTransaction` representing the current nested (savepoint) transaction, if any. This makes use of the underlying synchronous connection's :meth:`_engine.Connection.get_nested_transaction` method to get the current :class:`_engine.Transaction`, which is then proxied in a new :class:`.AsyncTransaction` object. .. versionadded:: 1.4.0b2 N)rXget_nested_transactionrerIr{s rr~z&AsyncConnection.get_nested_transactions:$$&++-  #>>uE ErczK|j}t|jfi|d{}||usJ|S7 w)a'Set non-SQL options for the connection which take effect during execution. This returns this :class:`_asyncio.AsyncConnection` object with the new options added. See :meth:`_future.Connection.execution_options` for full details on this method. N)rXrexecution_options)rEoptr[c2s rrz!AsyncConnection.execution_options3sE$$&!$"8"8@C@ @Tzz As *;9 ;cjK|j}t|jd{y7w)aCommit the transaction that is currently in progress. This method commits the current transaction if one has been started. If no transaction was started, the method has no effect, assuming the connection is in a non-invalidated state. A transaction is begun on a :class:`_future.Connection` automatically whenever a statement is first executed, or when the :meth:`_future.Connection.begin` method is called. N)rXrcommitrZs rrzAsyncConnection.commitDs($$&T[[))) )313cjK|j}t|jd{y7w)a&Roll back the transaction that is currently in progress. This method rolls back the current transaction if one has been started. If no transaction was started, the method has no effect. If a transaction was started and the connection is in an invalidated state, the transaction is cleared using this method. A transaction is begun on a :class:`_future.Connection` automatically whenever a statement is first executed, or when the :meth:`_future.Connection.begin` method is called. N)rXrrollbackrZs rrzAsyncConnection.rollbackSs($$&T]]+++rcjK|j}t|jd{y7w)zClose this :class:`_asyncio.AsyncConnection`. This has the effect of also rolling back the transaction if one is in place. N)rXrcloserZs rrzAsyncConnection.closeds($$&TZZ(((rcK|j}t|j|||dd{}t||jd{S7"7w)zaExecutes a driver-level SQL string and return buffered :class:`_engine.Result`. T_require_awaitN)rXrexec_driver_sqlrrE statement parametersrr[results rrzAsyncConnection.exec_driver_sqlns^$$&%       )1E1EFFF G!.AAAAAAc K|j}t|j||tjj |ddidd{}|j jsJdt|S7,w)zXExecute a statement and return a streaming :class:`_asyncio.AsyncResult` object.stream_resultsTrNzserver side result expected) rXr _execute_20r EMPTY_DICT merge_withcontext_is_server_siderrs rstreamzAsyncConnection.streams$$&%      OO & &!$4d#;    ~~-- 77 756"" sAA?A=-A?cK|j}t|j|||dd{}t||jd{S7"7w)aExecutes a SQL statement construct and return a buffered :class:`_engine.Result`. :param object: The statement to be executed. This is always an object that is in both the :class:`_expression.ClauseElement` and :class:`_expression.Executable` hierarchies, including: * :class:`_expression.Select` * :class:`_expression.Insert`, :class:`_expression.Update`, :class:`_expression.Delete` * :class:`_expression.TextClause` and :class:`_expression.TextualSelect` * :class:`_schema.DDL` and objects which inherit from :class:`_schema.DDLElement` :param parameters: parameters which will be bound into the statement. This may be either a dictionary of parameter names to values, or a mutable sequence (e.g. a list) of dictionaries. When a list of dictionaries is passed, the underlying statement execution will make use of the DBAPI ``cursor.executemany()`` method. When a single dictionary is passed, the DBAPI ``cursor.execute()`` method will be used. :param execution_options: optional dictionary of execution options, which will be associated with the statement execution. This dictionary can provide a subset of the options that are accepted by :meth:`_future.Connection.execution_options`. :return: a :class:`_engine.Result` object. TrN)rXrrrexecuters rrzAsyncConnection.executes\J$$&%         )>>> ?rcdK|j|||d{}|jS7w)auExecutes a SQL statement construct and returns a scalar object. This method is shorthand for invoking the :meth:`_engine.Result.scalar` method after invoking the :meth:`_future.Connection.execute` method. Parameters are equivalent. :return: a scalar Python value representing the first column of the first row returned. N)rscalarrErrrrs rrzAsyncConnection.scalars0 ||Iz;LMM}}N 0.0cdK|j|||d{}|jS7w)aiExecutes a SQL statement construct and returns a scalar objects. This method is shorthand for invoking the :meth:`_engine.Result.scalars` method after invoking the :meth:`_future.Connection.execute` method. Parameters are equivalent. :return: a :class:`_engine.ScalarResult` object. .. versionadded:: 1.4.24 N)rscalarsrs rrzAsyncConnection.scalarss1"||Iz;LMM~~NrcdK|j|||d{}|jS7w)aExecutes a SQL statement and returns a streaming scalar result object. This method is shorthand for invoking the :meth:`_engine.AsyncResult.scalars` method after invoking the :meth:`_future.Connection.stream` method. Parameters are equivalent. :return: an :class:`_asyncio.AsyncScalarResult` object. .. versionadded:: 1.4.24 N)rrrs rstream_scalarszAsyncConnection.stream_scalarss1${{9j:KLL~~Mrc^K|j}t||g|i|d{S7w)aInvoke the given sync callable passing self as the first argument. This method maintains the asyncio event loop all the way through to the database connection by running the given callable in a specially instrumented greenlet. E.g.:: with async_engine.begin() as conn: await conn.run_sync(metadata.create_all) .. note:: The provided callable is invoked inline within the asyncio event loop, and will block on traditional IO calls. IO within this callable should only call into SQLAlchemy's asyncio database APIs which will be properly adapted to the greenlet context. .. seealso:: :ref:`session_run_sync` N)rXr)rEfnrrr[s rrun_synczAsyncConnection.run_sync s30$$&#B9s9b9999s $-+-c>|jjSrC)rS __await__rUs rrzAsyncConnection.__await__&szz|%%''rcKtjj|j}tj|d{y7wrC)asyncioget_event_loop create_taskrshield)rEtype_r+ tracebacktasks r __aexit__zAsyncConnection.__aexit__)s7%%'33DJJLAnnT"""sA A A ArCF)*r2r3r4__doc__r5rGr__annotations__r classmethodrMrSpropertyrVr\r^r`rXrfrirlrorsrvrxrzr~rrrrr rrrrrrrrrrr6rrr?r?JsG*I E       A"$$))"$ & 3 J>> %,((" *,")// G4// #8// .?f// ,//  .//  *:8(#rr?z:class:`_future.Engine`z:class:`_asyncio.AsyncEngine`)clear_compiled_cacheupdate_execution_optionsget_execution_options)r!poolr9r@namedriverechoczeZdZUdZdZeZeed<Gdde Z dZ e ed< e dZd Zd Zd Zd Zd Zy)raPAn asyncio proxy for a :class:`_engine.Engine`. :class:`_asyncio.AsyncEngine` is acquired using the :func:`_asyncio.create_async_engine` function:: from sqlalchemy.ext.asyncio import create_async_engine engine = create_async_engine("postgresql+asyncpg://user:pass@host/dbname") .. versionadded:: 1.4 )rr^ _option_clsc eZdZdZddZdZy)AsyncEngine._trans_ctxc||_yrC)r[rZs rrGzAsyncEngine._trans_ctx.__init__Ss DIrcK|jj|d{|jj|_|jj d{|jS7Q7w)N)rR)r[rSrf transaction __aenter__rQs rrSzAsyncEngine._trans_ctx.startVs\))// /> > >#yy0D ""--/ / /99  ? /s" A8A4AA8#A6$A86A8cKfd}tjj|}tj|d{y7w)NcKjjd{jjd{y7'7wrC)rrr[r)rErrr+srgoz,AsyncEngine._trans_ctx.__aexit__..go^sC&&00yIIIiioo'''J's!!AA !AAAA)rrrr)rErr+rrrs```` rrz AsyncEngine._trans_ctx.__aexit__]s; ())+77=D..& & &sA AAANr)r2r3r4rGrSrr6rr _trans_ctxrRs   'rrc|jjs-tjd|jjd|j |x|_|_y)NzFThe asyncio extension requires an async driver to be used. The loaded z is not async.)r9is_asyncrrOrrDrr^)rErs rrGzAsyncEngine.__init__es_""++)))1188;>K ,0+?+? +LL4=rrct|SrC)rrJs rrMz(AsyncEngine._regenerate_proxy_for_targetxs 6""rcD|j}|j|S)aReturn a context manager which when entered will deliver an :class:`_asyncio.AsyncConnection` with an :class:`_asyncio.AsyncTransaction` established. E.g.:: async with async_engine.begin() as conn: await conn.execute( text("insert into table (x, y, z) values (1, 2, 3)") ) await conn.execute(text("my_special_procedure(5)")) )rPrrZs rrfzAsyncEngine.begin|s||~t$$rc$|j|S)aReturn an :class:`_asyncio.AsyncConnection` object. The :class:`_asyncio.AsyncConnection` will procure a database connection from the underlying connection pool when it is entered as an async context manager:: async with async_engine.connect() as conn: result = await conn.execute(select(user_table)) The :class:`_asyncio.AsyncConnection` may also be started outside of a context manager by invoking its :meth:`_asyncio.AsyncConnection.start` method. )_connection_clsrUs rrPzAsyncEngine.connects ##D))rc\Kt|jjd{S7w)zReturn a "raw" DBAPI connection from the connection pool. .. seealso:: :ref:`dbapi_connections` N)rrraw_connectionrUs rrzAsyncEngine.raw_connections%$D$4$4$C$CDDDDs #,*,c Lt|jjdi|S)aReturn a new :class:`_asyncio.AsyncEngine` that will provide :class:`_asyncio.AsyncConnection` objects with the given execution options. Proxied from :meth:`_future.Engine.execution_options`. See that method for details. r6)rrr)rErs rrzAsyncEngine.execution_optionss&=4++==DDEErc^Kt|jjd{y7w)amDispose of the connection pool used by this :class:`_asyncio.AsyncEngine`. This will close all connection pool connections that are **currently checked in**. See the documentation for the underlying :meth:`_future.Engine.dispose` method for further notes. .. seealso:: :meth:`_future.Engine.dispose` N)rrdisposerUs rrzAsyncEngine.disposes"T--55666s #-+-N)r2r3r4rr5r?rtyperrrrGrrrMrfrPrrrr6rrrr.sh ",I%O'%'&M##%$*$E F7rrceZdZdZdZddZedZdZe dZ e dZ e dZ d Z d Zd Zdd Zd Zy)rez4An asyncio proxy for a :class:`_engine.Transaction`.rVsync_transactionrhc.||_d|_||_yrCr)rErVrhs rrGzAsyncTransaction.__init__s$ $ rc|j}|}t|t}tj |}|J|j |}||_|j ||_||_|SrC) rV isinstancer r?rI__new__rDrrh)rKrLrArrhasync_connectionobjs rrMz-AsyncTransaction._regenerate_proxy_for_targetsw ++!F$56*EE   +++kk#)"223CD  rcR|js|j|jSrC)rrbrUs r_sync_transactionz"AsyncTransaction._sync_transactions#$$  ' ' )$$$rc|jSrC)rrUs rr^zAsyncTransaction._proxieds$$$rc6|jjSrC)ris_validrUs rrzAsyncTransaction.is_valids%%'000rc6|jjSrC)r is_activerUs rrzAsyncTransaction.is_actives%%'111rcfKt|jjd{y7w)a;Close this :class:`.Transaction`. If this transaction is the base transaction in a begin/commit nesting, the transaction will rollback(). Otherwise, the method returns. This is used to cancel a Transaction without affecting the scope of an enclosing transaction. N)rrrrUs rrzAsyncTransaction.closes%T335;;<<< '1/1cfKt|jjd{y7w)z%Roll back this :class:`.Transaction`.N)rrrrUs rrzAsyncTransaction.rollbacks#T335>>???rcfKt|jjd{y7w)z"Commit this :class:`.Transaction`.N)rrrrUs rrzAsyncTransaction.commits%T335<<===rc<K|jt|jr$|jj j n#|jj j d{|_|r|jj|S7+w)z{Start this :class:`_asyncio.AsyncTransaction` object's context outside of using a Python ``with:`` block. N) rDrrhrVrXrirfr __enter__rQs rrSzAsyncTransaction.start s~ !% 4 4 ;;002??__557== !    ! ! + + -  sA,B.B/,BclKt|jj|||d{y7wrC)rr__exit__)rErr+rs rrzAsyncTransaction.__aexit__s0  " " $ - -ueY   s *424Nr)r2r3r4rr5rGrrMrrr^rrrrrrSrr6rrreres~> " rrect|tr |jS |jS#t$r}t j d|z|d}~wwxYw)NzAsyncEngine expected, got %r)rr?rArAttributeErrorr ArgumentError)rFes r_get_sync_engine_or_connectionr sW,0+++'''  *\ 9  s * AA  Ac0tjdd)NzInspection on an AsyncConnection is currently not supported. Please use ``run_sync`` to pass a callable where it's possible to call ``inspect`` on the passed connection.xd3scoderNoInspectionAvailablesubjects r_no_insp_for_async_conn_yetr,s # # 8  rc0tjdd)NzInspection on an AsyncEngine is currently not supported. Please obtain a connection then use ``conn.run_sync`` to pass a callable where it's possible to call ``inspect`` on the passed connection.rrrrs r_no_insp_for_async_engine_xyetr6s # #   r)z sqlalchemy.)!rrrbaserrrrrr r r@r r engine.baser rrrutil.concurrencyrrr-r/create_proxy_methodsr?rrer _inspectsrrr6rrrs*!"'5, .$./.11!'  T#o'79IT# T#n #  N J7/#3J7 J7ZV (8V r o&'k"#r