U [hX&@sdZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd lm Z dd lm Z dd lm Z d dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZddlmZddl m!Z!e rd dlm"Z"d dl#m$Z$d dlm%Z%d dlm&Z&d dl'm(Z(d d lm)Z)dd!l*m+Z+dd"l,m-Z-e d#ed$Z.Gd%d&d&ee.Z/Gd'd(d(eZ0ej1j2d)d*Gd+d,d,eZ3Gd-d.d.ee.Z4Gd/d0d0e4e.ee.Z5d1d2d3d4d5Z6d6S)7zDynamic collection API. Dynamic collections act like Query() objects for read operations and support basic add/delete mutation. .. legacy:: the "dynamic" loader is a legacy feature, superseded by the "write_only" loader. ) annotations)Any)Iterable)Iterator)List)Optional)Tuple)Type) TYPE_CHECKING)TypeVar)Union) attributes)exc) relationships)util) PassiveFlag)Query)object_session)AbstractCollectionWriter)WriteOnlyAttributeImpl)WriteOnlyHistory)WriteOnlyLoader)result)QueryableAttribute)Mapper)_RelationshipOrderByArg)Session) InstanceState) AliasedClass) _Dispatch) ColumnElement_T)Zboundc@s$eZdZd dddddddd ZdS) DynamicCollectionHistoryNDynamicAttributeImplInstanceState[_T]rz&Optional[DynamicCollectionHistory[_T]]None)attrstatepassiveapply_toreturncCs`|r8t||d}t||_|j|_|j|_d|_n$t|_t|_t|_d|_dS)NFT) AppenderQuery autoflushrZOrderedIdentitySetZunchanged_items added_itemsZ deleted_itemsZ_reconcile_collection)selfr(r)r*r+Zcollr1<./venv/lib/python3.8/site-packages/sqlalchemy/orm/dynamic.py__init__>s    z!DynamicCollectionHistory.__init__)N)__name__ __module__ __qualname__r3r1r1r1r2r$=sr$c @s@eZdZUdZeeZded<dddddd d d d d ddZdS)r%TzType[AppenderMixin[Any]] query_classNz#Union[Type[Any], AliasedClass[Any]]strz"_Dispatch[QueryableAttribute[Any]]z Mapper[_T]rz!Optional[Type[AppenderMixin[_T]]]rr')class_keydispatch target_mapperorder_byr7kwr,cKsZtjj|||d|f|||_|r,t||_|s8t|_nt| krL||_n t ||_dSN) rZ AttributeImplr3r<tupler=r-r7 AppenderMixinmromixin_user_query)r0r9r:r;r<r=r7r>r1r1r2r3Ws"   zDynamicAttributeImpl.__init__)N) r4r5r6Z_supports_dynamic_iterationr$rZcollection_history_cls__annotations__r3r1r1r1r2r%Rs  r%Zdynamic)Zlazyc@seZdZeZdS) DynaLoaderN)r4r5r6r%Z impl_classr1r1r1r2rEosrEcseZdZUdZdZded<ded<ddd d fd d Zed dddZej dd dddZddddZ e rddddZ dddddZ ddd d!Zd4d d"d#d$d%Zd&d d'd(d)Zd*d d+d,d-Zd&d d'd.d/Zd*d d+d0d1Zd*d d+d2d3ZZS)5rAzTA mixin that expects to be mixing in a Query class with AbstractAppender. NzOptional[Type[Query[_T]]]r7zTuple[ColumnElement[Any], ...]_order_by_clausesr%r&r')r(r)r,cs"t||jdt||dSr?)rr3r<super)r0r(r) __class__r1r2r3~s zAppenderMixin.__init__zOptional[Session])r,cCsBt|j}|dk r*|jr*|j|kr*|t|js:dS|SdSr?)rinstancer.flushorm_utilZ has_identityr0sessr1r1r2sessions   zAppenderMixin.sessionr)rOr,cCs ||_dSr?)rN)r0rOr1r1r2rOsz1Union[result.ScalarResult[_T], result.Result[_T]]cCs|j}|dkrpt|j}|jr4tdt|t j t |j j jgt|j t|jtjjddS||SdS)NzInstance %s is detached, dynamic relationship cannot return a correct result. This warning will become a DetachedInstanceError in a future release.T)Z_source_supports_scalars)rOrinstance_staterJZdetachedrwarnrLZ state_strrZIteratorResultZSimpleResultMetaDatar(r9r4iter_get_collection_historyrPASSIVE_NO_INITIALIZEr/Zscalars _generate_iter)r0rNr)r1r1r2rVs(   zAppenderMixin._iterz Iterator[_T]cCsdSr?r1)r0r1r1r2__iter__zAppenderMixin.__iter__rzUnion[_T, List[_T]])indexr,cCs@|j}|dkr,|jt|jtj|S| | |SdSr?) rOr(rSrrPrJrrTZindexedrU __getitem__)r0rYrNr1r1r2rZs zAppenderMixin.__getitem__intcCs>|j}|dkr,t|jt|jtjj S| | SdSr?) rOlenr(rSrrPrJrrTr/rUcountrMr1r1r2r]s zAppenderMixin.countz Query[_T])rNr,cCs~|j}|dkr:t|}|dkr:tdt||jjf|jrT|j|jj |d}n| |jj }|j |_ |j |_ |j |_ |S)NzParent instance %s is not bound to a Session, and no contextual session is established; lazy load operation of attribute '%s' cannot proceed)rO)rJrorm_excZDetachedInstanceErrorrLZ instance_strr(r:r7r<queryZ_where_criteriaZ _from_objrF)r0rNrJr_r1r1r2rUs zAppenderMixin._generatez Iterable[_T])iteratorr,cCs||dS)a~Add an iterable of items to this :class:`_orm.AppenderQuery`. The given items will be persisted to the database in terms of the parent instance's collection on the next flush. This method is provided to assist in delivering forwards-compatibility with the :class:`_orm.WriteOnlyCollection` collection class. .. versionadded:: 2.0 NZ _add_all_implr0r`r1r1r2add_alls zAppenderMixin.add_allr#)itemr,cCs||gdS)apAdd an item to this :class:`_orm.AppenderQuery`. The given item will be persisted to the database in terms of the parent instance's collection on the next flush. This method is provided to assist in delivering forwards-compatibility with the :class:`_orm.WriteOnlyCollection` collection class. .. versionadded:: 2.0 Nrar0rdr1r1r2adds zAppenderMixin.addcCs||dS)zAdd an iterable of items to this :class:`_orm.AppenderQuery`. The given items will be persisted to the database in terms of the parent instance's collection on the next flush. Nrarbr1r1r2extendszAppenderMixin.extendcCs||gdS)zAppend an item to this :class:`_orm.AppenderQuery`. The given item will be persisted to the database in terms of the parent instance's collection on the next flush. Nrarer1r1r2append szAppenderMixin.appendcCs||dS)zRemove an item from this :class:`_orm.AppenderQuery`. The given item will be removed from the parent instance's collection on the next flush. N)Z _remove_implrer1r1r2removeszAppenderMixin.remove)N)r4r5r6__doc__r7rDr3propertyrOsetterrVr rWrZr]rUrcrfrgrhri __classcell__r1r1rHr2rAts(       rAc@seZdZdZdS)r-zA dynamic query that supports basic collection storage operations. Methods on :class:`.AppenderQuery` include all methods of :class:`_orm.Query`, plus additional methods used for collection persistence. N)r4r5r6rjr1r1r1r2r-sr-rztype[AppenderMixin[Any]])clsr,cCsd|j}t|t|fd|iS)zAReturn a new class with AppenderQuery functionality layered over.ZAppenderr7)r4typerA)rnnamer1r1r2rC)s rCN)7rjZ __future__rtypingrrrrrrr r r r rrr^rrrLbaserr_rrOrZ writeonlyrrrrZenginerrZmapperrrrr)rr Zeventr!Z sql.elementsr"r#r$r%ZRelationshipPropertyZ strategy_forrErAr-rCr1r1r1r2 sT                                  +