+h^ rddlZddlmZddlmZddlmZddlmZddlmZdd lm Z dd l m Z dd l m Z dd l m Zdd lmZe j"ddiZe j"ddiZe j(e ddddggdgdGddeZGddZGddeeZdZdZee_y) N)engine)result)ReversibleProxy)StartableContext)_ensure_sync_result)util)object_session)Session)state)greenlet_spawnprebuffer_rowsTstream_resultsz:class:`_orm.Session`z:class:`_asyncio.AsyncSession`r identity_key) __contains____iter__addadd_allexpire expire_allexpunge expunge_all is_modifiedin_transactionin_nested_transaction)dirtydeletednew identity_map is_active autoflush no_autoflushinfo) classmethodsmethods attributescteZdZUdZdZdZddZeZ ee d< d dZ dZ de jdfdZde jdfd Zde jdfd Z d!d Zde jdfd Zde jdfd ZdZd"dZd#dZdZdZddZdZdZdZdZdZdZdZ e!dZ"dZ#dZ$dZ%y)$ AsyncSessionaeAsyncio version of :class:`_orm.Session`. The :class:`_asyncio.AsyncSession` is a proxy for a traditional :class:`_orm.Session` instance. .. versionadded:: 1.4 To use an :class:`_asyncio.AsyncSession` with custom :class:`_orm.Session` implementations, see the :paramref:`_asyncio.AsyncSession.sync_session_class` parameter. TNc Fd|d<|r||_tj|}|r=||_|j Dcic]\}}|tj|}}}|r||_|j |j d||d|x|_|_ycc}}w)aConstruct a new :class:`_asyncio.AsyncSession`. All parameters other than ``sync_session_class`` are passed to the ``sync_session_class`` callable directly to instantiate a new :class:`_orm.Session`. Refer to :meth:`_orm.Session.__init__` for parameter documentation. :param sync_session_class: A :class:`_orm.Session` subclass or other callable which will be used to construct the :class:`_orm.Session` which will be proxied. This parameter may be used to provide custom :class:`_orm.Session` subclasses. Defaults to the :attr:`_asyncio.AsyncSession.sync_session_class` class-level attribute. .. versionadded:: 1.4.24 Tfuture)bindbindsN) r,r_get_sync_engine_or_connectionr-itemssync_session_class_assign_proxied sync_session_proxied)selfr,r-r1kwkeybs Q/var/www/html/venv/lib/python3.12/site-packages/sqlalchemy/ext/asyncio/session.py__init__zAsyncSession.__init__Hs&8 DI88>D DJ$kkmCV::1==E &8D #,0,@,@ #D # # AU Ab A-  DMs Br3cdKt|jj|||d{S7w)aExpire and refresh the attributes on the given instance. A query will be issued to the database and all attributes will be refreshed with their current database value. This is the async version of the :meth:`_orm.Session.refresh` method. See that method for a complete description of all options. .. seealso:: :meth:`_orm.Session.refresh` - main documentation for refresh )attribute_nameswith_for_updateN)rr3refresh)r5instancer<r=s r9r>zAsyncSession.refreshs7"$    % % ++     '0.0cRKt||jg|i|d{S7w)aInvoke the given sync callable passing sync 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 AsyncSession(async_engine) as session: await session.run_sync(some_business_method) .. 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)rr3)r5fnargr6s r9run_synczAsyncSession.run_syncs+2$B(9(9FCF2FFFFs '%'c K|r)tj|jt}nt}t |j j |f|||d|d{}t||j d{S7"7w)zExecute a statement and return a buffered :class:`_engine.Result` object. .. seealso:: :meth:`_orm.Session.execute` - main documentation for execute paramsexecution_optionsbind_argumentsN)r immutabledictunion_EXECUTE_OPTIONSrr3executerr5 statementrGrHrIr6rs r9rMzAsyncSession.executes"  $ 2 23D E K K ! !1 %    % %  /)     )>>> ?s$ABA?B:B;BBclK|j|f|||d|d{}|jS7w)zExecute a statement and return a scalar result. .. seealso:: :meth:`_orm.Session.scalar` - main documentation for scalar rFN)rMscalarrNs r9rQzAsyncSession.scalarsK $t||  /)     }} 424clK|j|f|||d|d{}|jS7w)aExecute a statement and return scalar results. :return: a :class:`_result.ScalarResult` object .. versionadded:: 1.4.24 Added :meth:`_asyncio.AsyncSession.scalars` .. versionadded:: 1.4.26 Added :meth:`_asyncio.async_scoped_session.scalars` .. seealso:: :meth:`_orm.Session.scalars` - main documentation for scalars :meth:`_asyncio.AsyncSession.stream_scalars` - streaming version rFN)rMscalarsrNs r9rTzAsyncSession.scalarssL2$t||  /)     ~~ rRc jKt|jj||||||d{S7w)zReturn an instance based on the given primary key identifier, or ``None`` if not found. .. seealso:: :meth:`_orm.Session.get` - main documentation for get )optionspopulate_existingr=identity_tokenN)rr3get)r5entityidentrVrWr=rXs r9rYzAsyncSession.gets@$$    ! !  /+)    s *313cK|r)tj|jt}nt}t |j j |f|||d|d{}tj|S7w)zbExecute a statement and return a streaming :class:`_asyncio.AsyncResult` object. rFN) r rJrK_STREAM_OPTIONSrr3rM_result AsyncResultrNs r9streamzAsyncSession.stream7s  $ 2 23D E K K! !0 %    % %  /)     ""6** sAA8A6A8clK|j|f|||d|d{}|jS7w)aRExecute a statement and return a stream of scalar results. :return: an :class:`_asyncio.AsyncScalarResult` object .. versionadded:: 1.4.24 .. seealso:: :meth:`_orm.Session.scalars` - main documentation for scalars :meth:`_asyncio.AsyncSession.scalars` - non streaming version rFN)r`rTrNs r9stream_scalarszAsyncSession.stream_scalarsUsL,#t{{  /)     ~~ rRc^Kt|jj|d{S7w)aTMark an instance as deleted. The database delete operation occurs upon ``flush()``. As this operation may need to cascade along unloaded relationships, it is awaitable to allow for those queries to take place. .. seealso:: :meth:`_orm.Session.delete` - main documentation for delete N)rr3delete)r5r?s r9rdzAsyncSession.deletets'$D$5$5$<$$>???? #,*,c\Kt|jjd{S7w)z+Commit the current transaction in progress.N)rr3commitrs r9rzAsyncSession.commitGs##D$5$5$<$<====rc^Kt|jjd{y7w)aPClose out the transactional resources and ORM objects used by this :class:`_asyncio.AsyncSession`. This expunges all ORM objects associated with this :class:`_asyncio.AsyncSession`, ends any transaction in progress and :term:`releases` any :class:`_asyncio.AsyncConnection` objects which this :class:`_asyncio.AsyncSession` itself has checked out from associated :class:`_asyncio.AsyncEngine` objects. The operation then leaves the :class:`_asyncio.AsyncSession` in a state which it may be used again. .. tip:: The :meth:`_asyncio.AsyncSession.close` method **does not prevent the Session from being used again**. The :class:`_asyncio.AsyncSession` itself does not actually have a distinct "closed" state; it merely means the :class:`_asyncio.AsyncSession` will release all database connections and ORM objects. .. seealso:: :ref:`session_closing` - detail on the semantics of :meth:`_asyncio.AsyncSession.close` N)rr3closers r9rzAsyncSession.closeKs"8T..44555s #-+-c\Kt|jjd{S7w)zClose this Session, using connection invalidation. For a complete description, see :meth:`_orm.Session.invalidate`. N)rr3 invalidaters r9rzAsyncSession.invalidateis% $D$5$5$@$@AAAArc\Kt|jjd{S7w)z2Close all :class:`_asyncio.AsyncSession` sessions.N)rr3 close_allrs r9rzAsyncSession.close_allps%$D$5$5$?$?@@@@rcK|SwNr.rs r9 __aenter__zAsyncSession.__aenter__us  scKtjj|j}tj|d{y7wr)asyncioget_event_loop create_taskrshield)r5type_value tracebacktasks r9 __aexit__zAsyncSession.__aexit__xs7%%'33DJJLAnnT"""sA A A Act|Sr)_AsyncSessionContextManagerrs r9_maker_context_managerz#AsyncSession._maker_context_manager|s *400rq)NNN)NN)NFNN)TNr)&__name__ __module__ __qualname____doc__ _is_asynciodispatchr:r r1__annotations__r>rDr EMPTY_DICTrMrQrTrYr`rbrdrgrjrlrsrwryrrrrrr classmethodrrrrr.rqr9r)r)s*< KH$ L!  ?C 0G<// ?J// 8//  L >// +B//  > H  G  L \ 0-. :@>6<BAA#1rqr)ceZdZdZdZdZy)rc||_yr) async_session)r5rs r9r:z$_AsyncSessionContextManager.__init__s *rqcK|jj|_|jjd{|jS7wr)rrrprrs r9rz&_AsyncSessionContextManager.__aenter__sC''--/ jj##%%%!!! &s=AAAcKfd}tjj|}tj|d{y7w)NcKjjd{jjd{y7*7wr)rprr)r5rrrsr9goz1_AsyncSessionContextManager.__aexit__..gosK**&&ueY? ? ?$$..ueYG G G @ Gs!!AA$A A AA)rrrr)r5rrrrrs```` r9rz%_AsyncSessionContextManager.__aexit__s< H%%'33BD9nnT"""sA AAAN)rrrr:rrr.rqr9rrs+" #rqrcLeZdZdZdZd dZedZdZdZ dZ d dZ d Z y ) rmaA wrapper for the ORM :class:`_orm.SessionTransaction` object. This object is provided so that a transaction-holding object for the :meth:`_asyncio.AsyncSession.begin` may be returned. The object supports both explicit calls to :meth:`_asyncio.AsyncSessionTransaction.commit` and :meth:`_asyncio.AsyncSessionTransaction.rollback`, as well as use as an async context manager. .. versionadded:: 1.4 )sessionsync_transactionrc.||_||_d|_yr)rrr)r5rrs r9r:z AsyncSessionTransaction.__init__s  $rqc^|jduxr|jjSr)_sync_transactionr!rs r9r!z!AsyncSessionTransaction.is_actives1  " " $D 0 3&&(22 rqcR|js|j|jSr)r_raise_for_not_startedrs r9rz)AsyncSessionTransaction._sync_transactions#$$  ' ' )$$$rqcfKt|jjd{y7w)z2Roll back this :class:`_asyncio.AsyncTransaction`.N)rrrrs r9rz AsyncSessionTransaction.rollbacks#T335>>??? '1/1cfKt|jjd{y7w)z/Commit this :class:`_asyncio.AsyncTransaction`.N)rrrrs r9rzAsyncSessionTransaction.commits%T335<<===rc,K|jt|jr |jjj n|jjj d{|_|r|jj|S7+wr) r2rrrr3rrr __enter__)r5 is_ctxmanagers r9startzAsyncSessionTransaction.startsv $ 4 4 ;; ))66\\..44 !    ! ! + + -  sA$B&B',BclKt|jj|||d{y7wr)rr__exit__)r5rrrs r9rz!AsyncSessionTransaction.__aexit__s0  " " $ - -ueY   s *424N)F) rrrr __slots__r:propertyr!rrrrrr.rqr9rmrmsA :I%   % @>  rqrmc4t|}| t|Sy)aReturn the :class:`_asyncio.AsyncSession` to which the given instance belongs. This function makes use of the sync-API function :class:`_orm.object_session` to retrieve the :class:`_orm.Session` which refers to the given instance, and from there links it to the original :class:`_asyncio.AsyncSession`. If the :class:`_asyncio.AsyncSession` has been garbage collected, the return value is ``None``. This functionality is also available from the :attr:`_orm.InstanceState.async_session` accessor. :param instance: an ORM mapped instance :return: an :class:`_asyncio.AsyncSession` object, or ``None``. .. versionadded:: 1.4.18 N)r r)r?rs r9async_object_sessionrs",X&GW%%rqc0tj|dS)aReturn the :class:`_asyncio.AsyncSession` which is proxying the given :class:`_orm.Session` object, if any. :param session: a :class:`_orm.Session` instance. :return: a :class:`_asyncio.AsyncSession` instance, or ``None``. .. versionadded:: 1.4.18 F) regenerate)r)rn)rs r9rrs  2 27u 2 MMrq)rrrr^baserrrr ormr r r _instance_stateutil.concurrencyrrJrLr]create_proxy_methodsr)rrmrr_async_providerr.rqr9rs!"'!+.%4%%'7&>?$$$$&6%=> $"N3  %:I 1?I 1;:I 1X##$; o/?; |: N#0rq