hZ'fdZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd lm Z dd lm Z d d l mZd dl mZd dl mZd dl mZd dl mZd dl mZd dlmZd dlmZd dlmZd dlmZd dlmZddl mZddl mZddlmZddlmZ ddl!m"Z"ddl!m#Z#ddl!m$Z$ddl!m%Z%dd l!m&Z&dd!l!m'Z'dd"l(m)Z)dd#l(m*Z*dd$l(m+Z+dd%l,m-Z-dd&l,m.Z.dd'l,m/Z/dd(lm0Z0dd)l1m2Z2e rHd d*l3m4Z4d d+l3m5Z5d d,l3m6Z6d d-l7m8Z8d d.l9m:Z:d d/l9m;Z;d d0l9mm?Z?dd3lm@Z@dd4lmAZAdd5lBmCZCe d6eD7ZEe dN dOd8ZFe dN dPd9ZF dQ dRd;ZFe dS dTd<ZGe dS dUd=ZG dV dWd>ZGd?ZHGd@dAeZIGdBdCeIZJe*jdDdEGdFdGeIe.ZLe*jdDdHGdIdJeJe/ZMe*jdDdKGdLdMeJe-ZNy:)Xzoadditional ORM persistence classes related to "bulk" operations, specifically outside of the flush() process. ) annotations)Any)cast)Dict)Iterable)Optional)overload) TYPE_CHECKING)TypeVar)Union) attributes)context) evaluator)exc)loading) persistence)NO_VALUE)AbstractORMCompileState) FromStatement)ORMFromStatementCompileState) QueryContext)util)Dialect)result) coercions)dml) expression)roles)select)sqltypes)_entity_namespace_key) CompileState)Options)DeleteDMLState)InsertDMLState)UpdateDMLState) EMPTY_DICT)Literal)DMLStrategyArgument)OrmExecuteOptionsParameter)SynchronizeSessionArgument)Mapper)_BindArguments)ORMExecuteState)Session)SessionTransaction) InstanceState) Connection)cursor)_CoreAnyExecuteParams_O)boundcyNmappermappingssession_transactionisstatesreturn_defaults render_nullsuse_orm_insert_stmtexecution_optionss n/var/www/html/tulostaulu.pesapallolive.fi/venv/lib/python3.12/site-packages/sqlalchemy/orm/bulk_persistence.py _bulk_insertrFJs cyr:r;r<s rErFrFWs #rGNc j}|jjr td|rL|r0|D cgc]} | | jf} } | D cgc]\} } |  }} } n>|D cgc]} | j}} n$|D cgc] } t | }} t ||j |d} |jjDcgc]\}}|jvr||f}}}|rd}nt|dkDrd}nd}|D]\}}|9|jDcgc]}|j|dvr |j!c}nd}fdtj|fd|Dd||| D}tj|d|||||| }||j r| |} |j"s|sJ| j%|} |r^|r\j&}j(Dcgc]}|j}} D](\} } |t+|Dcgc]}| | c}f| _ *|| J| Sycc} wcc} } wcc} wcc} wcc}}wcc}wcc}wcc}w) NzJconnection_callable / per-instance sharding not supported in bulk_insert()Tr Frr;c 3DK|]\}}}}}}}}d|| |||fywr:r;) .0state state_dictparamsmpconn value_params has_all_pkshas_all_defaults connectionr=s rE z_bulk_insert..sM   #   s c3*K|] }d|f ywr:r;rKmappingrTr=s rErUz_bulk_insert..sM$4Ms)bulkrArBinclude_bulk_keys) bookkeepingrCrD) base_mappersessionconnection_callableNotImplementedErrordict_expand_compositesrT_sorted_tablesitems _pks_by_tablelen_get_embedded_bindparamskeyr_collect_insert_commands_emit_insert_statements _returning returns_rowssplice_horizontally_identity_class_identity_key_propstuple)r=r>r?r@rArBrCrDr\rLstatesdict_m return_resulttablerOmappers_to_runr[ super_mapperbextra_bp_namesrecordsr identity_clspidentity_propsrgrTs` @rErFrFds$$K""66! -   7?@euejj)@F@4:;.5%;H;089u 9H9%-.DG..68,$// r?r@update_changed_onlyuse_orm_update_stmtenable_check_rowcounts rE _bulk_updaters rGcyr:r;r~s rErrsrGc j}jjr&jjhj fd}|r3|r|D cgc] } ||  }} n>|D cgc]} | j }} n$|D cgc] } t | }} t ||jjr td|j||9|jD cgc]} | j|dvr | j!c} nd} |jjD]g\} }j|r| jvr&t!j"d| fd|Dd|| }t!j$|d|| |d|| i|t'j(Sycc} wcc} wcc} wcc} w) Nc|jjDcic]\}}||jvs|vr||c}}Scc}}wr:)r`rccommitted_state)r=rLkv search_keyss rE _changed_dictz#_bulk_update.._changed_dict sO ((* 1E)))Q+-= qD   sAzJconnection_callable / per-instance sharding not supported in bulk_update()rr;c3vK|]0}d|jr|jjndf2ywr:)_version_id_proprgrWs rErUz_bulk_update..6sM "22  7 7 ; ;<!  s69T)rYrrZF)r[rr)r\_primary_key_propkeysrrgunionr`rar]r^r_rTrfrbrcisardr_collect_update_commands_emit_update_statements_result null_result)r=r>r?r@rrrr\rrLrrrwrxrtrvryrTrs` @@rErrs$$K..K ..22399+F   BJK fe4KHK089u 9H9%-.DG..68,""66! -  %// composite_attrscomposite_keysrg populatorsrXs rEraraWs''O --/0N"  _S ! K K MMJ%!..w7 %C JsOG $ %% sA3ceZdZUdZdZded<edZedZedZ edZ ed Z dd d Z ed Z y) ORMDMLStateTNz&Optional[ORMFromStatementCompileState]from_statement_ctxc#Ktj}|D]j\}}tjtj |}t |trt||t}|turdtjtj ||r9tjtj|tjdn|f|||j||Ed{d|jvrA|j} t| d| d} ||| j||Ed{*||s|n8tjtj|tjdfmy77Mw)N)defaultT)type_is_crudentity_namespace proxy_key)r(_get_crud_kv_pairsrexpectr DMLColumnRole isinstancestrr#rExpressionElementRoler"NullType_bulk_update_tuples _annotations) clsr= statement kv_iteratorneeds_to_be_cacheablecore_get_crud_kv_pairsrrdesck_annoattrs rE_get_orm_crud_kv_pairsz"ORMDMLState._get_orm_crud_kv_pairsjs"0!B!B0 DAq  !4!4a8A!S!,VQI8#!(()<) = >#334DE^%:%:!44;(=   & &%%%    rGc |jjd}|j}|jr |j}n|j j }||j |j|j|jdS)N parententity)nametypeexprentityrt) rtrr=is_aliased_classrclass___name__r local_table)rrext_infor= _label_names rEget_entity_descriptionz"ORMDMLState.get_entity_descriptionsm??//?  $ $"--K --00K MMOOoo''   rGc d}d}|jDcgc] }|||fc}Dcgc]<\}}|j|j||||j|jd>c}}Scc}wcc}}w)Nc:|jjddS)Nr)rget)cs rE _ent_for_colzCORMDMLState.get_returning_column_descriptions.._ent_for_cols>>%%nd; ;rGcx||S|jjdd}|s|St|j||S)Nr)rrgetattrr)rentrs rE _attr_for_colzDORMDMLState.get_returning_column_descriptions.._attr_for_cols={**;=Iszz9a88rG)rrraliasedr)_all_selected_columnsrgrrr)rrrrrrs rE!get_returning_column_descriptionsz-ORMDMLState.get_returning_column_descriptionss < 9$/8.M.M)*LO$  3 %a-//**      s A1AA6)use_supplemental_colsc,|jrt|j|d}|jdi|j}|j|j }||_tj||x|_ }|j||j}d|_|jDcgc]}|| } }| s| j|j|r|j|jd| i}|S|j | }|Scc}w)alestablish ORM column handlers for an INSERT, UPDATE, or DELETE which uses explicit returning(). called within compilation level create_for_statement. The _return_orm_returning() method then receives the Result after the statement was executed, and applies ORM loading to the state that we first established here. F)_adapt_on_namesr;supplemental_cols)rjrrD_execution_optionsoptions _with_optionsselect_statementrcreate_for_statementr!setup_dml_returning_compile_state _generateprimary_columnsextend primary_keyrA returning) selfcompilerorm_level_statementdml_level_statement dml_mapperrfsfscrcols_to_returns rE_setup_orm_returningz ORMDMLState._setup_orm_returnings9(  ) )#..# %B &%%O(;(N(NOB0>>?B$&D !,AA"hO D #c  1 1* ="5"?"?"A -/  *),)<)<NA aNNN"%%j&<&<=$&I&9&I&I ++''5'#(#" 'D&9&C&C#'##"9Os 5D=Dc X|j}|jj}|jr{|jjj s[|j dtj} t|j|j||| ||} tj|| S|S)N_sa_orm_load_options) rcompiled compile_statercompile_options_is_starrrdefault_load_optionsrr instances) rr]rrNrDbind_argumentsrexecution_contextr load_options querycontexts rE_return_orm_returningz!ORMDMLState._return_orm_returning9s#NN)22@@   , ,!44DDMM,00& (I(IL(00..!L$$V\: :MrG)r __module__ __qualname__is_dml_returningr__annotations__ classmethodrrrrrrrr;rGrErrfsAE>E55n  "  *    B#A#FrGrc eZdZGddeZeddddd ddZedZedZedZ ed Z ed Z ed Z ed Z ed ZedZedZedZy)BulkUDCompileStateceZdZUdZded<dZded<dZded<dZded <dZded <d Z ded <d Z ded<e Z d Z d Zd Zy ))BulkUDCompileState.default_update_optionsautor+ _dml_strategyr-_synchronize_sessionFbool_can_use_returning_is_delete_using_is_update_fromT _autoflushNOptional[Mapper[Any]]_subject_mapper)rrrrrrr r r r rr)_resolved_values_eval_condition _matched_rows_identity_tokenr;rGrEdefault_update_optionsr]sc-3 *3;A8A#(D(!&$& %% D15.5% rGrF is_multitableis_update_fromis_delete_usingis_executemanyctr:)r_)rdialectr=rrrrs rEcan_use_returningz$BulkUDCompileState.can_use_returningjs "##rGc tjjdhd||j\}}||d< |jd}|r!|j |d<|d|j iz } d|jjvr|d d iz }nat|ts;|jd k(r|d d iz }n:|jd k(r+tjd|jd k(r|d d iz }|j} | B| dvrtjd|jd k(r| dk(rtjd|s |j r|j!|jd k(rr|jd k(r|j#||||||}nq|jdk(r|j%||||||}nK|jdk(r<|j'||||||}n%|jd k(r|jd k(r|ddiz }|j)|j|j*|j,|j|j.d}|t1j2|j5d|ifS#t $rJdwxYw)N_sa_orm_update_options> autoflush dml_strategyidentity_tokenrrsynchronize_sessionclauserr=r0statement had 'orm' plugin but no plugin_subjectrr core_onlyrormrYHCan't use "bulk" ORM insert strategy without passing separate parameters)revaluatefetchFzSValid strategies for session synchronization are 'auto', 'evaluate', 'fetch', Falser)zkThe 'fetch' synchronization strategy is not available for 'bulk' ORM updates (i.e. multiple parameter sets)r(r)r"rrr r)rrfrom_execution_optionsrrr=KeyErrorrtrrrrsa_excInvalidRequestErrorr ArgumentErrorr _do_pre_synchronize_auto_do_pre_synchronize_evaluate_do_pre_synchronize_fetch _annotater r r r immutabledictr) rr]rrNrDr is_pre_eventupdate_optionsrsyncs rEorm_pre_session_execz'BulkUDCompileState.orm_pre_session_execws  5 5 L L $    ( (   $-x  M&778HIN+9+@+@x(#4n6K6K"LL !=!= =  < :   i M LL L Ms I11JcN|d}|jdk(rH|jdk(r|j||||nX|jdk(rI|j||||n4|jdk(r%|jdk(r|j |||||S|j ||||||S)Nrr&r(r)rY)rr_do_post_synchronize_evaluate_do_post_synchronize_fetch"_do_post_synchronize_bulk_evaluater)rr]rrNrDrrr5s rEorm_setup_cursor_resultz*BulkUDCompileState.orm_setup_cursor_results$++CD  ' '5 022j@11Y 44?..Y ) )V 322j@66VV^M((         rGc:d}jr jnddjf|vr&|tfd|djfDz }jj|jjfz }rtfd|D}|S)aApply extra criteria filtering. For all distinct single-table-inheritance mappers represented in the table being updated or deleted, produce additional WHERE criteria such that only the appropriate subtypes are selected from the total results. Additionally, add WHERE criteria originating from LoaderCriteriaOptions collected from the statement. r;Nadditional_entity_criteriac3tK|]/}|js|jur|j1ywr:)include_aliasesr_resolve_where_criteria)rKaers rErUz@BulkUDCompileState._adjust_for_extra_criteria..0s:!%%h)> **84!s58c3@K|]}j|ywr:)traverse)rKcritadapters rErUz@BulkUDCompileState._adjust_for_extra_criteria..<sO4 0 0 6Os)r_adapterr=ro_single_table_criterion)rglobal_attributesr return_critrFs ` @rE_adjust_for_extra_criteriaz-BulkUDCompileState._adjust_for_extra_criterias '/'@'@(##d ) OO   5!+18??C! K ?? 2 2 > HOOCCE EK O;OOKrGc|j|jjur|S|jjDcic]}||}}|jD]}|jn|j|jjur4t |j |jj}||Dcic]\}}|| } }}|jD] \}} | | ||<t|jjD cic] \} } || |  } } } |jjDcgc]}| | }}|Dcgc]tfd|Dc}Scc}wcc}}wcc} } wcc}wcc}w)atranslate from local inherited table columns to base mapper primary key columns. Joined inheritance mappers always establish the primary key in terms of the base table. When we UPDATE a sub-table, we can only get RETURNING for the sub-table's columns. Here, we create a lookup from the local sub table's primary key columns to the base table PK columns so that we can get identity key values from RETURNING that's against the joined inheritance sub-table. the complexity here is to support more than one level deep of inheritance, where we have to link columns to each other across the inheritance hierarchy. c3(K|] }| ywr:r;)rKidxrows rErUz?BulkUDCompileState._interpret_returning_rows..os>3c#h>) rr\riterate_to_rootinheritsr`_table_to_equatedrc enumeratero)rr=rowspklocal_pk_to_base_pkrOt_to_esuper_pksub_pk col_to_colsuper_rNlpklookupbpkprimary_key_convertrOs `rE_interpret_returning_rowsz,BulkUDCompileState._interpret_returning_rows@s(   V%7%7%C%C CK 170B0B0N0NO"r2vOO((* =B{{"2;;#:#::"..r{{/F/FGHFCI":N/?x&(*NJN1779 = F*4V*<#B' = =&f&8&8&D&DE S  $c )  $*#5#5#A#A F3K  KOO3>*=>>OO)PO  Ps E2 E7E= FFc|j}|j}|Dcgc]F}|jj|r)|js|j ||j fH}}|j}|)|D cgc]\}}} |j|k(r||| f}}}} g} |D]J\}}} ||} | dus| tjus&| j||| | tjufL| Scc}wcc} }}wNT) rrr=rexpiredobjr`rr!r_EXPIRED_OBJECTappend) rr5rpr=eval_conditionrLraw_datar!rerqrevaled_conditions rE _get_matched_objects_on_criteriaz3BulkUDCompileState._get_matched_objects_on_criteriaqs,//'77  ||' YY[% ,  (77  %*2%C''>9eU#H !)  C-c2  !D(#y'@'@@ (I,E,EE  " A s A C4>C9cZ|j}|j}tj|}d}|jr||jz }i}|j D] }|j s|j|"|r||j||z }|r|j|} | Sd} | } | S)Nr;cyrcr;)res rErzJBulkUDCompileState._eval_condition_from_statement.._eval_conditionsrG) rrr_EvaluatorCompiler_where_criteriar_is_criteria_optionget_global_criteriarKprocess) rr5rr= target_clsevaluator_compilerrErIoptrhrs rE_eval_condition_from_statementz1BulkUDCompileState._eval_condition_from_statements//]] &99*E  $ $ I-- -D** ;C&&''(9: ;  C223DfM MD 7/77>N  -NrGc |j||}||ddzS#tj$rYnwxYw|ddiz }|j||||||S)asetup auto sync strategy "auto" checks if we can use "evaluate" first, then falls back to "fetch" evaluate is vastly more efficient for the common case where session is empty, only has a few objects, and the UPDATE statement can potentially match thousands/millions of rows. OTOH more complex criteria that fails to work with "evaluate" we would hope usually correlates with fewer net rows. r()rrrr))rvrUnevaluatableErrorr1)rr]rrNrDrr5rhs rEr/z+BulkUDCompileState._do_pre_synchronize_autos2  ?? N"#1(2% ++    17;;,,         s 22c |j||}|d|izS#tj$r}tjd|z|d}~wwxYw)Nz{Could not evaluate current criteria in Python: "%s". Specify 'fetch' or False for the synchronize_session execution option.r)rvrrxr,r-) rr]rrNrDrr5rherrs rEr0z/BulkUDCompileState._do_pre_synchronize_evaluatesr  ?? N ~!   ++ ,,8:=>   sA AA c|jrgS|jrt|jS|jr#t|jj SgSr:) _multi_values_ordered_valuesr_valuesrc)rr=rs rE_get_resolved_valuesz'BulkUDCompileState._get_resolved_valuessR  " "I  & & 112 2    ))//12 2IrGcg}|D]f\}}|rHt|tjr. |j|}|j |j |fPtjd|z|S#t j$rYwxYw)NzCAttribute name not found, can't be synchronized back to objects: %r) rr ColumnElement_columntopropertyrgrgorm_excUnmappedColumnErrorr,r-)rr=resolved_valuesvaluesrrrs rE_resolved_keys_as_propnamesz.BulkUDCompileState._resolved_keys_as_propnames s# DAq*Q (@(@A1!33A6DMM488Q-00079:;  22sA00BBcB j t j jfzj j |j }|j|_d d fd }|j|||||} | j} | dzS)Nc||jjdi|j}j|jj j |j}|k7r:tjd|jr|stjd||rtjSy)N)rrrzjFor synchronize_session='fetch', can't mix multiple backends where some support RETURNING and others don'tzFor synchronize_session='fetch', can't use multiple parameter sets in ORM mode, which this backend does not support with RETURNINGr;) r]get_bindrrrr r rr,r-rr) orm_contextbindper_bind_resultrrr=r5s rEskip_for_returningzHBulkUDCompileState._do_pre_synchronize_fetch..skip_for_returning>s/;&&//M+2L2LMD"33 -== . ? ?*99 4O!,$7 44  ++O00- %4!**,,rG)rDr _add_event)rr )rr0returnr) rr!rselect_identity_token select_fromrrroexecutefetchall) rr]rrNrDrr5 select_stmtrr matched_rowsrr=s ` ` @@rEr1z,BulkUDCompileState._do_pre_synchronize_fetchs // FV''6+G+G*II K [ Wi-- /  '0&?&? #!  B  /)) ! ( )"3!   rGNrrr= Mapper[Any]rr rr rr rr rr )rrrr%rrrr7r<rKrarkrvr/r0rrr1r;rGrErr\s\   $$ %$ $ $ $  $  $ $ $  $ $u u n) ) V""H.P.P`$$L8- - ^  2"N N rGrr&insertceZdZUGddeZdZded<edZe d dZ ed fd Z ed Z d Z d Z xZS) BulkORMInsertcbeZdZUdZded<dZded<dZded<dZd ed <d Zded <dZ ded <y)$BulkORMInsert.default_insert_optionsrr+rFr  _render_nulls_return_defaultsNrrTr _populate_existing) rrrrrrrrr rr;rGrEdefault_insert_optionsrpsB-3 *3# t#!&$&15.5 D#(D(rGrNzOptional[FromStatement]rctjjdhd||j\}}||d< |jd}|r!|j |d<|d|j iz } |s;|jdk(r|d d iz }n:|jd k(r+tjd |jdk(r|d d iz }|jd k7r2|stj}n|jtj}|s|jr|j|jd|ji}|tj |jd|ifS#t $rJdwxYw)N_sa_orm_insert_options>rr rBpopulate_existingr#rr=rr$rrr&rYr'rawr )rrr*rrr=r+rr,r-r_orm_load_exec_optionsrr r2rr3) rr]rrNrDrr4insert_optionsrs rEr7z"BulkORMInsert.orm_pre_session_execzs  0 0 G G $ N   ( (    $-x  M&778HIN+9+@+@x(#4n6K6K"LL++v5?E"::--7002 ++v5?F";;  ' '5 0 %$+$B$B!$5$;$;22%! 9 9    '' ^99 :    0 1 7 7)>:   O M LL L Ms E!!E2c t|jd|j}|jdvrtjd|jdk(r|j ||xsi|}|S|jdk(r|j } |j$| jrtjd| | J|jJt| tdt|tr|gn||jd |j|j || }n2|jd k(r|j ||xsi|}n t#t%|j&s|S|j(r:|jd t*j,} | d diz } |j/d | i}|j1||||||S)Nr)rrYr&rzHValid strategies for ORM insert strategy are 'raw', 'orm', 'bulk', 'autor)rDrYz`bulk INSERT with a 'post values' clause (typically upsert) not supported for multi-table mapper Iterable[Dict[str, Any]]F)r@rArBrCrDr&rrT)rrrr,r.rr_post_values_clause_multiple_persistence_tablesr- _transactionrFrrr`rrAssertionErrorr rjrrrrr) rr]rrNrDrrPrrr=rs rEorm_execute_statementz#BulkORMInsert.orm_execute_statements+.. $c&@&@   ' '0  &&2   ' '5 0\\6>)HHK++,<= --11.%H 6 !  ' ' 1 U "  & &x 8 rGcfd|jDDcic]\}}}| |jn||c}}}Scc}}}w)Nc3`K|]%\}}jj|||f'ywr:)rr)rKrrr=s rErUz;BulkORMInsert._resolved_keys_as_col_keys..2s-,0Aqa!Q's+.)rcrg)rr=resolved_value_dictcolrrs ` rE_resolved_keys_as_col_keysz(BulkORMInsert._resolved_keys_as_col_keys.sP4G4M4M4O  QCGGAq 0   sActtj|jx}}|j ||||d}||_y)NFrr)rrInsertrr)rrr=rrs rErz#BulkORMInsert._setup_for_orm_insert7sI*.szz4>>*JJ '--   "' . #rGc ttj|jx}}|j}|d|d}}|j }||_|jrA|jjDcic]\}}|j |ur||c}}|_|j||||d}|j5|jjjrtjd||_ycc}}w)zestablish an INSERT statement within the context of bulk insert. This method will be within the "conn.execute()" call that is invoked by persistence._emit_insert_statement(). _emit_insert_table_emit_insert_mapperTrNzCan't use RETURNING * with bulk ORM INSERT. Please use a different INSERT form, such as INSERT..VALUES or INSERT with a Core Connection)rrrrr_clonert_dict_parametersrcrrrrr, CompileError) rrrranemit_insert_tableemit_insert_mapperrvals rErz$BulkORMInsert._setup_for_bulk_insertCs+/szz4>>*JJ '  # # # $ $ %. $$& +   !% 5 5 ; ; =%C99 11S%D ! --   )"& .   # # /''77@@%%3  #1%s=D)r]r1rz dml.InsertrNr6rDr,rr/rPr4r_result.Result)rr)rrrr%rrrrr7rrrrr __classcell__rs@rErrns))15-4A A FV V V & V 6 V ' V V  V V p,   #,#rGrupdateceZdZedZdZdZe d fd Zeddddd d dZedZ ed Z ed Z ed Z xZ S) BulkORMUpdatec X|j|}|jjdd}|j }|r|dk(r|j |||S|dk(s|dk(r2d|j jvrt j|||fi||S|r|dvr|j|||S)Nr  unspecifiedrYr%r)r&r) __new__rrr_setup_for_bulk_updatertr(__init___setup_for_orm_update)rrrrrr rs rErz"BulkORMUpdate.create_for_statementts{{3 --11 M  ~~%  .  ' ' 8 <  K '},ioo&B&BB  # #D)X D D \-CC  & &y( ; rGc |}|j }|jjd}|jx|_}|j |||_|j |||||jrt|j |_|j}|jr|j |_ n|jr|j |_|j|j|} | r|j| }tj|||fi|d} |sd} n|jj!dd} |jj!dd} | dur/| dk(xr(|j#|j$||j&} | dk(r'| r%d} |j(|jj*}|r|j-||||| }||_y) Nrrprocess_criteria_for_toplevelFr"rr))rTr)rrtrr=rr_init_global_attributesr~r`rr}rKrIwherer(rrrrrrArrr) rrrrrrrr=new_stmtnew_critrr"rs rErz#BulkORMUpdate._setup_for_orm_updates'~~%??//?'. f $ 9 9&) L $$  *2 %    $()>)>$?D !##%  $ $'+'<'|jd|j}|jdvrtjd|jdk(r|j }|j dk7sJ|j r$|j dk(rtjd|j} | J|jJt| tdt|tr|gn||jd d || } |j|||||| St |A||||||S) Nr)r&rrYr%zOValid strategies for ORM UPDATE strategy are 'orm', 'auto', 'bulk', 'core_only'rYr)r(zbulk synchronize of persistent objects not supported when using bulk update with additional WHERE criteria right now. add synchronize_session=None execution option to bypass synchronize of persistent objects.rF)r@rrr)rrrr,r.rorr-rrrrrr`r<rr) rr]rrNrDrrPr5rr=rrs rErz#BulkORMUpdate.orm_execute_statementsa+.. $c&@&@   ' '0  &&9   ' '6 1(1(A(A$A !!66'A A))"77:E00$33F% %''3 3!. *64 8VHf$$$)$-&; F..!  70!  rGFrc|jxr|jj}|sy|r |jS|r |jS|r/|js#t j d|jdy)NF Dialect "z" does not support RETURNING with UPDATE..FROM; for synchronize_session='fetch', please add the additional execution option 'is_update_from=True' to the statement to indicate that a separate SELECT should be used for this backend.T)update_returningrimplicit_returningupdate_executemany_returningupdate_returning_multifromr,rrrrr=rrrr normal_answers rErzBulkORMUpdate.can_use_returningZs  $ $ N););)N)N  77 7 55 5 7#E#E%%GLL>*EE rGc|sy|j}|jDcgc]}|j}}|j}|D]|j fd|D|j } |j | } | s=tj|} | j} | jj| } | D]}|| vs|| |<| jjj| d| | j| t!| | j| j| }|s| j#| | ycc}w)Nc3(K|] }| ywr:r;)rKrgparams rErUzCBulkORMUpdate._do_post_synchronize_bulk_evaluate..s/s/rP)rrnrg identity_mapidentity_key_from_primary_keyrfast_get_stater differencer` unmodifiedrmanagerdispatchrefresh_commitr_expire_attributes)rr]rNrr5r=proppk_keysr identity_keyrLevaluated_keysrq to_evaluaterg to_expirers @rEr;z0BulkORMUpdate._do_post_synchronize_bulk_evaluatesH //(.(B(BC488CC++  ;E!??/w/..L!// =E Z227;NJJE**77GK" ,%>   + + -  //   =L M M%9S%c5% M   NsAc |j}|j}|r9|j||}|Dcgc]}t||jfz} }n |j } | Dcgc] }|dd|dfc}D cgc]<\} } |j| |jk(r|j t| | >c} } D cgc]} | |jvr|j| !} } | sy|j|||| Dcgc]-}|tj|tj|f/c}ycc}wcc}wcc} } wcc} wcc}wNr)r!) rreturned_defaults_rowsrarorrrrrrrinstance_state instance_dict)rr]rrr5 target_mapperrpk_rowsrOrrr!robjsres rEr:z(BulkORMUpdate._do_post_synchronize_fetchs'66 !'!>!> !335G #c n<<>>L *77L5A4-0S2YB(4 ! 0K"119!^%C%CC;;%#1< ! w333   .    //       --c2,,S1  54 ! . s D7$D<;AE$E=2E c|j}|j}tj|}|j ||}|j ||} i} | D]>\} } |j tjtj| } | | | <@t| j}| Dchc]\}}| }}}t}|D]\}}}|jj!|}|D]} | |vs| | ||| <|j"j$j'|d||j)|t||j!|j+|}|r|j-|||j/||j1|y#tj$rYgwxYwcc}}w)zeapply values to objects derived from an update statement, e.g. UPDATE..SET N)rrrrnrrrrrrr rrxrrrrrrrrrrradd_register_altered)rr]r5rrr=rsrtrresolved_keys_as_propnamesvalue_evaluatorsrgvalue _evaluatorrrrattribrprerLrqrrs rErz1BulkORMUpdate._apply_update_set_values_to_objectss //]] &99*E2269E%(%D%D O& "4 3JC 3/77$$U%@%@%H )3 % 3.3356 :;1!;;!0  C**77GK" <%=F>)r]r1rz dml.UpdaterNr6rDr,rr/rPr4rrr)rrrrrrrrrr;r9r:rrrs@rErrrs=,]"~"#HIII& I 6 I ' II IIV $$ %$"""  "  """ ""H(;(;T    1 1 f1*1*rGrdeleteceZdZedZe dfd Zeddddd d dZedZedZxZ S) BulkORMDeletec |j|}|jjdd}|dk(s|dk(r2d|jjvrt j |||fi||S|j }|}|jjd}|jx|_} |j|||||j} |j|j| } | r| j| } t j || |fi|d} |sd} n|jjdd} |jjdd}|durJ| d k(xrC|j|j| |j|jjd d }|r%d } | j | jj"} |r|j%||| | | } | |_|S)Nr rr%rrFr"rr)r)rrTr)rrrrtr&rrr=rrrKrIrrrrrArrr)rrrrrr rrrr=rrrr"rs rErz"BulkORMDelete.create_for_statement%s{{3 --11 M K '},ioo&B&BB  # #D)X D DK~~%'??//?'. f $$  *2 % ##%11  " "F  %x~~x0H h?B? %"& "*"7"7";";%t# %1155   E )$w.**$$"&"4"4$,$9$9$=$=)5% +  $( !/x//1K1KLH 00#!&; 1H" rGc|jd|j}|jdk(rtjd|jdvrtj dt |||||||S)NrrYzBulk ORM DELETE not supported right now. Statement may be invoked at the Core level using session.connection().execute(stmt, parameters))r&rr%zGValid strategies for ORM DELETE strategy are 'orm', 'auto', 'core_only')rrrr,r-r.rr) rr]rrNrDrrPr5rs rErz#BulkORMDelete.orm_execute_statement~s+.. $c&@&@   ' '6 1,,A   ' '/K K&&  w, Y(9>4  rGFrc|jxr|jj}|sy|r |jS|r/|js#t j d|j dy)NFrz" does not support RETURNING with DELETE..USING; for synchronize_session='fetch', please add the additional execution option 'is_delete_using=True' to the statement to indicate that a separate SELECT should be used for this backend.T)delete_returningrrdelete_returning_multifromr,rrrs rErzBulkORMDelete.can_use_returningsz  $ $ N););)N)N  55 5 7#E#E %%GLL>*EE rGc|j||jj}g}|D]A\}}} } | r'|j| |jj1|j |C|r|j |yyr:)rkrr_expire _modifiedrg_remove_newly_deleted) rr]rrr5r to_deleterrLrqis_partially_expireds rEr9z+BulkORMDelete._do_post_synchronize_evaluates>>   + + -   5D ( 1Aue1# eW%9%9%C%CD  '  (   ) )) 4 rGc|j}|j}|r9|j||}|Dcgc]}t||jfz} }n |j } | D]i}|dd} |d} |j t| | } | |jvs8|jtj|j| gkycc}wr) rrrarorrrrrrrr) rr]rrr5rrr rOrrr!rs rEr:z(BulkORMDelete._do_post_synchronize_fetchs'66 !'!>!> !335G #c n<<>>L *77L Ca)K WN)FF[!-GLw333--"11#00> s C)r]r1rz dml.DeleterNr6rDr,rr/rPr4rrr) rrrrrrrr9r:rrs@rErr#sVVp   &  6  '      > $$ %$&&&  &  &&& &&P55&$$rGr)..)r= Mapper[_O]r>r#r?r2r@r rAr rBr rCOptional[dml.Insert]rDr%rzcursor.CursorResult[Any])NN)r=r"r>r#r?r2r@r rAr rBr rCr'rDr%rz"Optional[cursor.CursorResult[Any]]).T)r=rr>r#r?r2r@r rr rr$rr rr&)r=rr>r#r?r2r@r rr rOptional[dml.Update]rr rz_result.Result[Any]rc)r=rr>r#r?r2r@r rr rr(rr rzOptional[_result.Result[Any]])O__doc__ __future__rtypingrrrrrr r r r rrrrrrrbaserrrrrr,renginerrrsqlrrrr r!r"sql.baser#r$r%sql.dmlr&r'r(r) util.typingr*_typingr+r,r-r=r.r]r/r0r1r2rLr3r4r5engine.interfacesr6objectr7rFrrarr plugin_forrrrr;rGrEr7s # ,"1!&,#$$$!,33'( +$#9 T  *->A   J ,      ' <      14>A #  #J #, # #  #  #. #< # #  #&15>B} }J},} }  }  }.}<}(}@ *-"& J,    '     14"& J,    .   "15"&X% X%JX%,X% X%  X% . X% X%#X%v %s)slO O d)@#K@#*@#F)m*&m**m*` )\&\*\rG