Mj&dZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd lm Z dd lm Z dd lm Z dd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlmZddl!m"Z"e r0ddlm#Z#ddl$m%Z%ddlm&Z&ddlm'Z'dd l(m)Z)dd!lm*Z*dd"l+m,Z,dd#l-m.Z.e d$e%Z/Gd&d'ee/Z0Gd(d)eZ1ej23d*+Gd,d-e Z4Gd.d/ee/Z5Gd0d1e5e/ee/Z6d8d6Z7d7S)9zDynamic 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)overload)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)boundceZdZ d dd ZdS)DynamicCollectionHistoryNattrDynamicAttributeImplstateInstanceState[_T]passiverapply_to&Optional[DynamicCollectionHistory[_T]]returnNonecb|r]t||d}tj||_|j|_|j|_d|_dStj|_tj|_tj|_d|_dS)NFT) AppenderQuery autoflushrOrderedIdentitySetunchanged_items added_items deleted_items_reconcile_collection)selfr)r+r-r.colls j/lsinfo/ai/hellotax_ai/training_center/backend/venv/lib/python3.11/site-packages/sqlalchemy/orm/dynamic.py__init__z!DynamicCollectionHistory.__init__?s  / u--77>>D#'#:4#@#@D '3D !)!7D )-D & & &!%!8!:!:D #688D #'#:#<#r<r(r(>s3 <@ ///////r>r(c:eZdZUdZeeZded< dddZdS)r*TzType[AppenderMixin[Any]] query_classNclass_#Union[Type[Any], AliasedClass[Any]]keystrdispatch"_Dispatch[QueryableAttribute[Any]] target_mapper Mapper[_T]order_byr!Optional[Type[AppenderMixin[_T]]]kwrr0r1c tjj|||d|fi|||_|rt ||_|st |_dSt| vr ||_dSt||_dSr?) r AttributeImplr=rLtuplerNr3rE AppenderMixinmromixin_user_query)r:rFrHrJrLrNrErPs r<r=zDynamicAttributeImpl.__init__Xs  ) &#tX  13   +  ,!(OODM =,D    koo// / /*D   / <r?)rFrGrHrIrJrKrLrMrNrrErOrPrr0r1) r@rArB_supports_dynamic_iterationr(rcollection_history_cls__annotations__r=rCr>r<r*r*SsQ"&5c:)))):>=======r>r*dynamic)lazyceZdZeZdS) DynaLoaderN)r@rArBr* impl_classrCr>r<r]r]ps%JJJr>r]ceZdZUdZdZded<ded<d-fd Zed.dZej d/dZd0dZ e rd1dZ e d2dZe d3dZd4d Zd5d!Z d6d7d$Zd8d'Zd9d)Zd8d*Zd9d+Zd9d,ZxZS):rTzTA mixin that expects to be mixing in a Query class with AbstractAppender. NzOptional[Type[Query[_T]]]rEzTuple[ColumnElement[Any], ...]_order_by_clausesr)r*r+r,r0r1ctj||jdt||dSr?)rr=rLsuper)r:r)r+ __class__s r<r=zAppenderMixin.__init__sG        u%%%%%r>Optional[Session]ct|j}|$|jr|j|vr|t j|jsdS|Sr?)rinstancer4flushorm_util has_identityr:sesss r<sessionzAppenderMixin.sessionsVdm,,   4=D3H3H JJLLL$T]33 4Kr>rlr c||_dSr?)rk)r:rls r<rlzAppenderMixin.sessions  r>1Union[result.ScalarResult[_T], result.Result[_T]]c "|j}|tj|j}|jr)t jdtj|ztj tj |j j jgt|j tj|jt"jjdS||S)NzInstance %s is detached, dynamic relationship cannot return a correct result. This warning will become a DetachedInstanceError in a future release.T)_source_supports_scalars)rlrinstance_staterfdetachedrwarnrh state_strrIteratorResultSimpleResultMetaDatar)rFr@iter_get_collection_historyrPASSIVE_NO_INITIALIZEr7scalars _generate_iter)r:rkr+s r<r|zAppenderMixin._iters| <-dm<>$''--// /r> Iterator[_T]cdSr?rC)r:s r<__iter__zAppenderMixin.__iter__sr>indexintr%cdSr?rCr:rs r< __getitem__zAppenderMixin.__getitem__s-0Sr>sliceList[_T]cdSr?rCrs r<rzAppenderMixin.__getitem__s58Sr>Union[int, slice]Union[_T, List[_T]]c|j}|O|jtj|jt j|S| | |Sr?) rlr)rxrrqrfrryindexedr{r)r:rrks r<rzAppenderMixin.__getitem__sm| <944)$-881genn  >>$''33E:: :r>c|j}|Nt|jt j|jtjj S| | Sr?) rllenr)rxrrqrfrryr7r{countrjs r<rzAppenderMixin.countsp| < 11-dm<<5  >>$''--// /r>rk Query[_T]c|j}|Ht|}|7tjdt j|d|jjd|jr"||jj |}n| |jj }|j |_ |j |_ |j |_ |S)NzParent instance zh is not bound to a Session, and no contextual session is established; lazy load operation of attribute 'z' cannot proceed)rl)rfrorm_excDetachedInstanceErrorrh instance_strr)rHrErLquery_where_criteria _from_objr`)r:rkrfrs r<r{zAppenderMixin._generates= <!(++D|33 ,X6666 G   8$$TY%iterator Iterable[_T]c0||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 N _add_all_implr:rs r<add_allzAppenderMixin.add_alls 8$$$$$r>itemc2||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 Nrr:rs r<addzAppenderMixin.adds  D6"""""r>c0||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. Nrrs r<extendzAppenderMixin.extend s 8$$$$$r>c2||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. Nrrs r<appendzAppenderMixin.appends  D6"""""r>c0||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) _remove_implrs r<removezAppenderMixin.removes $r>)r)r*r+r,r0r1)r0rd)rlr r0r1)r0rn)r0r})rrr0r%)rrr0r)rrr0r)r0rr?)rkrdr0r)rrr0r1)rr%r0r1)r@rArB__doc__rErYr=propertyrlsetterr|r rr rrr{rrrrr __classcell__)rcs@r<rTrTus .2K11115555&&&&&&X ^^000020//// 000X0 888X8;;;; 0 0 0 0#': % % % % # # # #%%%%####        r>rTceZdZdZdS)r3zA 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)r@rArBrrCr>r<r3r3%sr>r3clsrr0type[AppenderMixin[Any]]cJd|jz}t|t|fd|iS)zAReturn a new class with AppenderQuery functionality layered over.AppenderrE)r@typerT)rnames r<rVrV0s*  $D }c*]C,@ A AAr>N)rrr0r)8r __future__rtypingrrrrrr r r r r rrrrrrrhbaserrrrlr writeonlyrrrrenginerrmapperrrr r+r!r"eventr# sql.elementsr$r%r(r*RelationshipProperty strategy_forr]rTr3rVrCr>r<rs  #""""" ######//////------''''''&&&&&&-$$$$$$666666      $$$$$$""""""!!!!!!,,,,,, WT//////3///*=====1===:#00i0@@&&&&&&&A@&m m m m m ,R0m m m `M"%uRyBBBBBBr>