PqhcUddlmZddlmZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZerddlmZddlmZddlmZddlmZddl 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(l*m0Z0dd)l1m2Z2dd*l3m4Z4dd+lm5Z5dd,l6m7Z7dd-l8m9Z9dd.l8m:Z:Gd/d0e;Zd2<ed3gd4Z?Gd5d6e<7Z@ d9d8ZAy):) annotations) namedtupleN)Any)Callable)Dict)Iterable)List)Mapping)Optional)Sequence)Set)Tuple)Type) TYPE_CHECKING)Union)castschema)text)base)util) sqla_compat)Literal)TextIO) Connection)Dialect) CursorResult) Inspector) ClauseElement) Executable) ColumnElement) quoted_name)Column) Constraint)ForeignKeyConstraint)Index)Table)UniqueConstraint) TableClause) TypeEngine)_ServerDefault)AutogenContext)ApplyBatchImpl)BatchOperationsImplc eZdZ ddZy)ImplMetacVtj||||}d|vr |t|d<|S)N __dialect__)type__init___impls)cls classnamebasesdict_newtypes Q/srv/www/ollama/chat-ollama/venv/lib/python3.12/site-packages/alembic/ddl/impl.pyr6zImplMeta.__init__7s4 --Yu= E !+.F5' (N)r9strr:zTuple[Type[DefaultImpl]]r;Dict[str, Any])__name__ __module__ __qualname__r6r>r=r2r26s#  (  r>r2zDict[str, Type[DefaultImpl]]r7Params)token0tokensargskwargsc4eZdZUdZdZdZdZddhfZded<d Z d ed <d Z d ed< d:dZ e d;dZ ddZed?dZdd ej(f d@dZ dA dBdZ dC dDdZ dA dEdZ dA dFdZdGdZdHdZ dA dIdZdJdZdJdZdKd ZdJd!Z dJd"Z!dLd#Z"dKd$Z# dM dNd%Z$dOd&Z% dPd'Z& dQd(Z' dRd)Z(d*Z) dSd+Z*d,Z+ dT dUd-Z,dVd.Z- dWd/Z.d0Z/dXd1Z0dXd2Z1dXd3Z2 dYd4Z3d5Z4dZd6Z5 d[d7Z6d8Z7 d\d9Z8y)] DefaultImplaProvide the entrypoint for major migration operations, including database-specific behavioral variances. While individual SQL/DDL constructs already provide for database-specific implementations, variances here allow for entirely different sequences of operations to take place for a particular migration, such as SQL Server's special 'IDENTITY INSERT' step for bulk inserts. defaultF;NUMERICDECIMALzTuple[Set[str], ...] type_synonymsrDz Sequence[str]type_arg_extract)orderon_nullzTuple[str, ...]identity_attrs_ignorec||_||_||_|jdd|_||_i|_||_|||_|jr"|jstjdyy)N literal_bindsFz3Can't use literal_binds setting without as_sql mode) dialect connectionas_sqlgetrV output_buffermemo context_optstransactional_ddlr CommandError)selfrWrXrYr^r[r]s r=r6zDefaultImpl.__init__bs $ )--ouE* (  (%6D "   ;;''I r>c(t|jSN)r7name)r8rWs r=get_by_dialectzDefaultImpl.get_by_dialect|sgll##r>c|jJ|jj|dz|jjy)Nz )r[writeflush)r`rs r= static_outputzDefaultImpl.static_outputs>!!---   /   "r>cy)zReturn True if the given :class:`.BatchOperationsImpl` would need the table to be recreated and copied in order to proceed. Normally, only returns True on SQLite when operations other than add_column are present. FrD)r`batch_ops r=requires_recreate_in_batchz&DefaultImpl.requires_recreate_in_batchsr>cy)zperform any operations needed on a table before a new one is created to replace it in batch mode. the PG dialect uses this to drop constraints on the table before the new one uses those same names. NrD)r` batch_impltables r=prep_table_for_batchz DefaultImpl.prep_table_for_batchr>c|jSrb)rXr`s r=bindzDefaultImpl.binds r>Nctt|tr t|}|jr|s|r t d|j r)t|t jstddi}ni}trt|tsJ|jdd|ji|}|jt|jddj|j zy|j"}|J|r|j$di|}|rt|t&sJ||fz }|j)||S) Nz+Execution arguments not allowed with as_sqlrVT)compile_kwargsrW z rD) isinstancer?rrY ExceptionrVr DDLElementdictrr!compilerWrhreplacestripcommand_terminatorrXexecution_optionstupleexecute)r` constructr multiparamsparams compile_kwcompiledconns r=_execzDefaultImpl._execs/ i %YI ;;f MNN!!*6,,+"/41HI  !)];;;(y((LLLH   H %%dF399;))* ??D# ## -t--B0AB!+u555y( << ;7 7r>c(|j||yrb)r)r`sqlrs r=rzDefaultImpl.executes 3)*r>c ||tjdd|,|jtj||||| | | | |duri}t j || rtj}n(J II,    JJ##!"/,C&7%5    &B66 71188 711!6 )) JJ" ""/,C&7%5      JJ!"/,C&7%5   %  JJ""!"/,C&7%5     JJ!"/,C&7  r>cR|jtj|||yNr)rr AddColumn)r`rcolumnrs r= add_columnzDefaultImpl.add_columnAs 4>>*fVDEr>c R|jtj|||yr)rr DropColumn)r`rrrrs r= drop_columnzDefaultImpl.drop_columnIs 4??:vfEFr>c|j|j|r%|jtj|yyrb) _create_rulerr AddConstraintr`consts r=add_constraintzDefaultImpl.add_constraintRs8    %););D)A JJv++E2 3*Br>cL|jtj|yrb)rrDropConstraintrs r=drop_constraintzDefaultImpl.drop_constraintVs 6((/0r>cR|jtj|||yr)rr RenameTable)r`old_table_namenew_table_namers r= rename_tablezDefaultImpl.rename_tableYs"   ^^F K r>cb|jj||jd||jt j ||jj ||jd||jD]&}|jt j|(|jjxr|jj }|j}|r|r|j||jD]%}|j}|s|s|j|'yNF) checkfirst _ddl_runner)dispatch before_createrXrr CreateTable after_createindexes CreateIndexrWsupports_commentsinline_commentsrcreate_table_commentcolumnscreate_column_comment)r`rnindex with_commentrrs r= create_tablezDefaultImpl.create_tablecs $$ 4??u$ %  6%%e,- ## 4??u$ $ ]] 2E JJv))%0 1 2 LL * * O4<<3O3O/O -- |  % %e ,mm 3FnnG<**62 3r>c|jj||jd||jt j ||jj ||jd|yr)r before_droprXrr DropTable after_dropr`rns r= drop_tablezDefaultImpl.drop_tablezsc "" 4??u$ #  6##E*+ !! 4??u$ " r>c N|jtj|fi|yrb)rrrr`rrs r= create_indexzDefaultImpl.create_indexs 6%%e2r23r>cL|jtj|yrb)rrSetTableCommentrs r=rz DefaultImpl.create_table_comments 6))%01r>cL|jtj|yrb)rrDropTableCommentrs r=drop_table_commentzDefaultImpl.drop_table_comments 6**512r>cL|jtj|yrb)rrSetColumnComment)r`rs r=rz!DefaultImpl.create_column_comments 6**623r>c N|jtj|fi|yrb)rr DropIndexrs r= drop_indexzDefaultImpl.drop_indexs 6##E0R01r>ct|ts td|rt|dts td|jr|D]}|j t j|jdi|jDcic]Q\}}|t|t js/t j|||j|jn|Sc}}y|re|r'|j t j||y|D]6}|j t j|jdi|8yycc}}w)Nz List expectedrzList of dictionaries expected)r)rrD) rwlist TypeErrorrzrYrr_insert_inlinevaluesitems_literal_bindparamcr5)r`rnrows multiinsertrowkvs r= bulk_insertzDefaultImpl.bulk_insertsK $%O, , *T!Wd3;< < ;;  c |j|jk(ry|jDcgc] }|Dchc]}|jc}"}}}dj|jg|jz}dj|jg|jz}|D]>}||hj |s(|j|jhj |s>yycc}wcc}}w)NT F)rFrPrjoinrGissubset)r`inspector_paramsmetadata_paramsbatchtsynonymsinspector_all_termsmetadata_all_termss r=_column_types_matchzDefaultImpl._column_types_matchs  " "o&<&< <<@)>%? ?  $4$9$99xx(//0668xx **+113(( C3%B3%BbRXXa[BHHQK7  r>c|j|}|j|}|j||sy|j||syy)zReturns True if there ARE differences between the types of the two columns. Takes impl.type_synonyms into account between retrospected and metadata types TF)rrr)r`inspector_columnmetadata_columnrrs r= compare_typezDefaultImpl.compare_typesQ 556FG44_E''(8/J&&'7Ir>c ||k7SrbrD)r`rrrendered_metadata_defaultrendered_inspector_defaults r=compare_server_defaultz"DefaultImpl.compare_server_default%s*-FFFr>cyrbrD)r` conn_uniques conn_indexesmetadata_unique_constraintsmetadata_indexess r=correct_for_autogen_constraintsz+DefaultImpl.correct_for_autogen_constraints.s r>cn|jj|jurt|d||d<yy)Nexpr)r5_type_affinityr)r`existingexisting_transfernew_types r=cast_for_batch_migratez"DefaultImpl.cast_for_batch_migrate7s: == ' 'x/F/F F(,!&)8) f % Gr>c Xddd}t|j|j|S)zdRender a SQL expression that is typically a server default, index expression, etc. TF)rV include_table)rWru)r?r{rW)r`r*is_server_defaultrrs r=render_ddl_sql_exprzDefaultImpl.render_ddl_sql_expr=s/(,eD  LLjL I  r>c|jSrb)autogen_column_reflect)r` inspectors r=_compat_autogen_column_reflectz*DefaultImpl._compat_autogen_column_reflectKs***r>cyrbrD)r`conn_fks metadata_fkss r=correct_for_autogen_foreignkeysz+DefaultImpl.correct_for_autogen_foreignkeysNs r>cy)zA hook that is attached to the 'column_reflect' event for when a Table is reflected from the database during the autogenerate process. Dialects can elect to modify the information gathered here. NrD)r`r6rn column_infos r=r5z"DefaultImpl.autogen_column_reflectUrpr>cy)zA hook called when :meth:`.EnvironmentContext.run_migrations` is called. Implementations can set up per-migration-run state here. NrDrrs r=start_migrationszDefaultImpl.start_migrations^rpr>c@|jd|jzy)zEmit the string ``BEGIN``, or the backend-specific equivalent, on the current connection context. This is used in offline mode and typically via :meth:`.EnvironmentContext.begin_transaction`. BEGINNrhr~rrs r= emit_beginzDefaultImpl.emit_beginfs 7T%<%<<=r>c@|jd|jzy)zEmit the string ``COMMIT``, or the backend-specific equivalent, on the current connection context. This is used in offline mode and typically via :meth:`.EnvironmentContext.begin_transaction`. COMMITNrBrrs r= emit_commitzDefaultImpl.emit_commitps 8d&=&==>r>cy)NFrD)r`type_objautogen_contexts r= render_typezDefaultImpl.render_typezsr>cTt||tjdh\}}t|dd}t|dd}t |t |k7r|j d|j |j||t |xst |t |k7fS)Nalways)skip)_compare_identity_optionsrIdentitygetattrbooladddifference_updaterT)r`metadata_identityinspector_identitydiffignored meta_alwaysinspector_alwayss r=_compare_identity_defaultz%DefaultImpl._compare_identity_defaults2    "  g/4@ "#5xF  %5 6 6 HHX  t99:   J M$01T:L5MM   r>c:td|jDS)Nc34K|]}|jywrb)rc).0cols r= z/DefaultImpl.create_index_sig..s7#SXX7s)rr)r`rs r=create_index_sigzDefaultImpl.create_index_sigs7777r>cxtt|jDcgc]}|jc}Scc}wrb)rsortedrrc)r`rr^s r=create_unique_constraint_sigz(DefaultImpl.create_unique_constraint_sigs*V?#SXX?@AA?s7 c T|Dcic]}|j|}}t|D]{}|j|vrtj|}|s*t j d|jd|j dtjd|j|}ycc}w)Nz@autogenerate skipping metadata-specified expression-based index z ; dialect z under SQLAlchemy z6 can't reflect these indexes so they can't be compared) rcrris_expression_indexrrr4sqlalchemy_versiondiscard)r`r'r%rconn_indexes_by_nameidxiexs r=_skip_functional_indexesz$DefaultImpl._skip_functional_indexess3?@a @@() .Cxx//11#6C .xxl*T-=-=,@A""-"@"@!ABFF!((- . AsB%c&|jdiS)Ndialect_options)rZ)r`reflected_objectkinds r= adjust_reflected_dialect_optionsz,DefaultImpl.adjust_reflected_dialect_optionss ##$5r::r>)rWrrXOptional[Connection]rYrQr^Optional[bool]r[zOptional[TextIO]r]r@returnNone)rWrrszType[DefaultImpl])rr?rsrt)rjr0rsrQ)rmr/rnr)rsrt)rsrq) rUnion[Executable, str]rOptional[dict[str, Any]]rzSequence[dict]rr@rszOptional[CursorResult]rb)rrurrvrsrt) NFNNNNFNNNNN) rr?rr?rrrrz%Union[_ServerDefault, Literal[False]]rc Optional[str]rOptional[TypeEngine]rrwrrrrz$Optional[Union[str, Literal[False]]]rrwrrxrzOptional[_ServerDefault]rrrrrrrrrsrt)rr?r Column[Any]r!Optional[Union[str, quoted_name]]rsrt)rr?rryrrwrsrt)rrrsrt)rr&rsrt)rr?rzUnion[str, quoted_name]rrzrsrt)rnr)rsrt)rr(rrrsrt)rzColumnElement[Any]rsrt)T)rnzUnion[TableClause, Table]rz List[dict]rrQrsrt)rr%rsrE)rrErrErsrQ)rrErrErsrQ)rryrr%rsrQ) r$Set[UniqueConstraint]r% Set[Index]r&r{r'r|rsrt)F)r*r!r2rQrrrsr?)r6r rsr)r9Set[ForeignKeyConstraint]r:r}rsrt)rsrt)rHr,rIr.rszUnion[str, Literal[False]])rr(rsTuple[Any, ...])rr*rsr~)rnr@ror?rsr@)9rArBrC__doc__r4r^r~rP__annotations__rQrTr6 classmethodrdrhrkropropertyrsr immutabledictrrrrrrrrrrrrrrrrrrrrr"r(r/r3r7r;r5r?rCrFrJrZr`rcrkrprDr>r=rKrKHs K,5y+A*CM'C&(m( .B?A)  *  ( % 4$$# +     (  16     7;&(!3!3!3!5 '8)'84'8$ '8  '8 '8X7;+ #+4+  +$(@E"&* $(,8=*..2<@,015nnn! n > n  n$nn&n6n(n,n":n*n!/n !n" #nh59 FFF2 F  F!% GGG G  G4159   0 2    3. 42342! $($$ $  $L$L &9? , &5; B +>D  G + ! &;  %    >C  !  6:  JM    + + 0      >?"5C #  88B%B B ."; .;69; ;r>rK) metaclassc tj|}tj|}tt dfd }|||tj|t|j|tjr6|t |dit |di|j t |difS)Nct|j D]c}|j|}|j|}||k7s+|j|}||k(r j|Sj|eyrb)set differencerZrR) meta_dict insp_dict default_dictattrsattr meta_value insp_value default_valuerV ignored_attrrMs r= check_dictsz._compare_identity_options..check_dictssx J))$/ #D"t,J"t,JZ' , 0 0 6 . $$T*HHTN #r>dialect_kwargs)rMapping[str, Any]rrrrrz Iterable[str])r_get_identity_options_dictrunionidentity_has_dialect_kwargsrPr) metadata_io inspector_io default_iorMmeta_dinsp_drrVrs ` @@r=rNrNs 3 3K @F  3 3L AF 5D5L#$#$#(# # ..z: F &!  ..  K!12 6 L"2B 7  % % L"2B 7    r>)r-Union[schema.Identity, schema.Sequence, None]rrrz'Union[schema.Identity, schema.Sequence]rMzSet[str])B __future__r collectionsrrtypingrrrrr r r r r rrrr sqlalchemyrrrrrrrrsqlalchemy.enginerrsqlalchemy.engine.cursorrsqlalchemy.engine.reflectionr sqlalchemy.sqlr!r"sqlalchemy.sql.elementsr#r$sqlalchemy.sql.schemar%r&r'r(r)r*sqlalchemy.sql.selectabler+sqlalchemy.sql.type_apir,r-autogenerate.apir.operations.batchr/r0r5r2r7rrErKrNrDr>r=rs""  ,)56,)53,0:++652$116 t (*$) HD Eq ;Hq ;h/>/?/8/  /r>