PqhIddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl!m#Z#ddl!m$Z$ddl!m%Z%ddl!m&Z&ddl'm(Z(ddl)m*Z*e rTddlm+Z+dd lm,Z,dd!lm-Z-dd"lm.Z.dd#lm/Z/dd$l0m1Z1dd%lm2Z2dd&l3m4Z4dd'l5m6Z6dd(l!m7Z7dd)l8m9Z9dd*l8m:Z:dd+l;md}d0Z?e@ejd1eDcgc] }e?| c}ZBeBd2k\ZCeBd3k\ZDeBd4k\ZEeBd5k\ZFeBd6k\ZGeZH dd7lImJZJGd8d9ZL dd:lmMZMd;ZNeBdeLZMd?ZNd?ZOYwxYw#eK$re s GdBdCeLZPd?ZTYwxYw)) annotationsN)Any)Dict)Iterable)Iterator)Mapping)Optional) TYPE_CHECKING)TypeVar)Union) __version__)inspectschema)sql)types)url)compiles)CheckConstraint)Column)ForeignKeyConstraint)visitors) DialectKWArgs) BindParameter) ColumnClause) quoted_name) TextClause)UnaryExpression)traverse) TypeGuard)Index)Table) Connection)Dialect) Transaction) Inspector)ColumnCollection) SQLCompiler)Insert) ColumnElement) Constraint) SchemaItem)Select) TableClause_CE)zColumnElement[Any]r,)boundc, t|S#|cYSxYwN)int)values Y/srv/www/ollama/chat-ollama/venv/lib/python3.12/site-packages/alembic/util/sqla_compat.py _safe_intr66s5z s z (\d+|[abc]\d)))r7)r7r9)r7r9)) _NONE_NAMEceZdZdZy) _Unsupportedz.Placeholder for unsupported SQLAlchemy classesN)__name__ __module__ __qualname____doc__r5r?r?Ns4rEr?)ComputedT)r7r8c eZdZy)rFNr@rArBrDrEr5rFrFW rErFF)Identityc |iStrA|j}|r-t|tsJ|j |j |Si}t|t r*|j|d<|j|j|d<d}|j |Dcic]}t||d|t||dc}|Scc}w)Nalwayson_null) start incrementminvaluemaxvalue nominvalue nomaxvaluecyclecacheorder) identity_has_dialect_kwargs_as_dict isinstancerupdatedialect_kwargsrKrMrNgetattr)identityr\as_dictattrskeys r5_get_identity_options_dictrbls  I ('')G!(M:::x66763G(H-$,OO!##/)1)9)9GI& E NN %xd3?355  s"Cc eZdZy)rKNrIrDrEr5rKrKerJrErK) _NoneName)symbolcB|tuxst|ttfSr2)r=rZstrrdnames r5constraint_name_definedrjs  :  CD3 2B!CCrEc"t|tSr2)rZrgrhs r5constraint_name_stringrls dC  rEc t|r|SdSr2)rlrhs r5constraint_name_or_nonerns*$/49T9rEautoc#K |j}|dy|s"|J|j5ddddydy#t$rd}YBwxYw#1swYyxYwwr2)in_transactionAttributeErrorbegin connectionrqs r5_ensure_scope_for_ddlrvsy #22 ) ))!!#          s=A" A"A"AA" AA"AA"AA"cZtr|j|S|j|S)N) hide_password)sqla_14render_as_string __to_string__)rrxs r5url_render_as_stringr|s+#s##-@@ s  }==rEc@t|}|r|S|jSr2)_get_connection_transactionrsru transactions r5"_safe_begin_connection_transactionrs&.j9K!!rEc@t|}|r|jyyr2)r~commitrs r5#_safe_commit_connection_transactionrs".j9KrEc@t|}|r|jyyr2)r~rollbackrs r5%_safe_rollback_connection_transactionrs".j9KrEcH |j}|S#t$rYywxYw)NF)rqrrrts r5_get_connection_in_transactionrs1 #22  s  !!c|jSr2) expressionsidxs r5_idx_table_bound_expressionsrs ??rEc bt|dr|jdi|S|jdi|S)N_copyrD)hasattrrcopy) schema_itemkws r5rrs8{G$ {  &2&&{%"%%rEc^tr|jS|j}|jSr2)ryget_transaction_root_Connection__transaction)rurs r5r~r~s-))++   )))rEcttjdrtjj|i|Stj|i|S)Ncreate)rrURLr)argrs r5 _create_urlr s<sww!ww~~s)b))ww"r""rEc~trt|j||S|jj|||Sr2)ryr has_tabledialect) connectable tablename schemanames r5_connectable_has_tablers?{#--iDD"",, J  rEc tr,|j5}|j||cdddS|jj||S#1swYyxYwr2)ry_operation_contextexecutebind) inspector statementparamsconns r5_exec_on_inspectorrsT  ) ) + 3t<< 62 3 3~~%%i88 3 3s AAcbts |jSddlm}|j|j uS)Nrr)rynullablesqlalchemy.sqlr_user_defined_nullableNULL_UNSPECIFIED)metadata_columnrs r5_nullability_might_be_unsetr&s/ ''')  2 2f6M6M M rEc4tsytd|DS)NFc3<K|]}t|tywr2)rZrF.0sds r5 z._server_default_is_computed..5E:b(+E) has_computedanyserver_defaults r5_server_default_is_computedr1s EnEEErEc4tsytd|DS)NFc3<K|]}t|tywr2)rZrKrs r5rz._server_default_is_identity..<rr)ryrrs r5_server_default_is_identityr8s EnEEErEc^t|tr|j}|J|S|jSr2)rZrparenttable) constraintrs r5_table_for_constraintr?s5*23!!    rEct|tr$|jDcgc]}|jc}St|trt |j St|jScc}wr2) rZrelementsrr _find_columnssqltextlistcolumns)rfks r5_columns_for_constraintrHsY*23$.$7$78b 88 J 0Z//00J&&'' 9sA/cVtr|j|dS|j|dSr2)ry reflect_table reflecttable)rrs r5_reflect_tablerQs0&&ud33%% 4  rEcpt|r*t|\}}|j|j|S|Sr2)_type_has_variants_get_variant_mappinggetri)type_r base_typemappings r5_resolve_for_variantrZs3% 1%8 7{{7<<33 rE_variant_mappingc,t|jSr2)boolrrs r5rrdsE**++rEc||jfSr2)rrs r5rrgse,,,,rEc8t|tjuSr2)typesqltypesVariantrs r5rrlsE{h....rEc2|j|jfSr2)implrrs r5rroszz5==((rEc T|jDcgc]}|j|j}}|jj}|jj}|j dj jj}|j dj jj}|j Dcgc]}|j j}}|j} |j} |j} |j} ||||||| | | | f Scc}wcc}w)Nr) column_keysrrirrrcolumnrondeleteonupdate deferrable initially) rrasource_columns source_table source_schema target_schema target_tableelementtarget_columnsrrrrs r5_fk_specrss0:0F0F), 3$$N$$))L%%,,M''*1177>>M&&q)0066;;L9C9L9LMggnn))MNM""H""H&&J$$I  Ns D D%c|jdj}|jd}|jddj |}|j J||j j k(S)Nr.)r _get_colspecsplitpopjoinrra)rspectokenstablekeys r5_fk_is_self_referentialrsj   q ! . . 0D ZZ_F JJrNxxH    (( ( z((,, ,,rEc|jSr2) _type_bound)rs r5_is_type_boundrs  ! !!rEcLt}t|id|ji|S)z2locate Column objects within the given expression.r)setradd)clausecolss r5rrs% 5D VR(DHH-. KrEc|jJ||j}t|ds t|dr|jj|y|j|y)z(remove a column from a ColumnCollection.N _immutable _readonly)rar_parentremove) collectionr to_removes r5_remove_column_from_collectionrsY :: !! !6::&Iz<(GJ ,L!!),)$rEcbt|tr-t|tj}|j ||St|t r t||St|trt||jSt|tjr t||Std)zAa workaround for the Index construct's severe lack of flexibilityz#String or text() construct expected)rZrgrrNULLTYPE append_columnr_textual_index_element_textual_index_columntextrr*_copy_expression ValueError)rtext_cs r5rrs% 5(++ , A E: &%eU33 E1 2$UEJJ77 E3,, -u-->??rEc<fd}tj|i|S)Nct|tri|j]|jurO|jjvrj|jSt |}j ||Syr2)rZrrrirrr)colrrs r5replacez!_copy_expression..replacese sF # % -xx<>>)#~~chh//#J**1-rE)rreplacement_traverse) expressionrr!s ` r5rrs$   ( (B rEc"eZdZdZdZddZdZy)raQWrap around a sqlalchemy text() construct in such a way that we appear like a column-oriented SQL expression to an Index construct. The issue here is that currently the Postgresql dialect, the biggest recipient of functional indexes, keys all the index expressions to the corresponding column expressions when rendering CREATE INDEX, so the Index we create here needs to have a .columns collection that is the same length as the .expressions collection. Ultimately SQLAlchemy should support text() expressions in indexes. See SQLAlchemy issue 3174. _textual_idx_elementc||_||_|j|_tj|jjt j |_|j|jyr2) rrrarrrr fake_columnr)selfrrs r5__init__z_textual_index_element.__init__sP  99!==9J9JK D,,-rEc|jgSr2)r')r(s r5 get_childrenz#_textual_index_element.get_childrens  !!rEN)rr"rrreturnNone)r@rArBrC__visit_name__r)r+rDrEr5rrs ,N."rErc <|j|jfi|Sr2)processrrcompilerrs r5_render_textual_index_columnr3s  8  GLL /B //rEc eZdZy)_literal_bindparamNrIrDrEr5r5r5srEr5c (|j|fi|Sr2)render_literal_bindparamr1s r5_render_literal_bindparamr8s -8 , ,W ; ;;rEc,t|jSr2)rrrs r5_get_index_expressionsr:s   rEcTt|Dcgc]}t|ddc}Scc}w)Nri)r:r])rexps r5_get_index_column_namesr=s$2H2M N3GC & NN Ns%c*tr |jSiSr2)sqla_13kwargs)r s r5_column_kwargsrAszz rEc|jy|Jtr|jj|dSt |jdrt |j}nt }|t|jd}|j|}t|tjr#|j|d}|j|S|jj|S)NF_alembic_quotequote)rErh)riryidentifier_preparerformat_constraintrrrrg __class__rZrr! ddl_compiler_prepared_index_name)rrquoted_name_clsnew_nameds r5_get_constraint_final_namerNs    **<< u=  :??G ,%)$9O)O"3z#7uE))x)8 j&,, /$$Wd3A))  ..@@L LrEctr2|jy|J|jj|d}|duS|jduS)NFrC)ryrirFrG)rrris r5_constraint_is_namedrPEs\ ?? """"**<< u= 4d**rEcltr |jSt|jxr |jSr2)ry is_mariadbrserver_version_info _is_mariadb mysql_dialects r5rTrTTs5'''  - - *))  rEc|jSr2) _mariadb_normalized_version_inforUs r5rXrX^s  9 99rEcntr|jjS|jdS)NT)inline)ryinsertrZ)rs r5_insert_inliner\bs+||~$$&&||4|((rE)create_mock_engine)select create_enginec tdd|S)Nz postgresql://mock)strategyexecutorr_)rrdrs r5r]r]os fx  rEc@tjt|fi|Sr2)rr^r)rrs r5_selectrftszz$w-.2..rEc|jD]M}t|tr|j}t|trt|tr |j sMyy)NTF)rrZrrr is_literal)indexexprs r5is_expression_indexrkxsN!!/<rs"  "(!,-$2#-10/.3,'  ,)-643)50005 e5!CDE %2::&6 DEaYq\E  6/ 6/ j j $ A> 55 2#L#z13##-X}"E %%8%% %NL -3c9,-sI~.D D&D ! !! : ::  $,>"""   &***#  (+ 9I   9 FF (  8   23,- /)8-"% %*O% %$@ @C@+@$*"S..": !0 #0/:00"0    < <+6<<< !O$M($M3D$M$MN +( +3D +  + :) -,( /uFA@@A$  | L#$  | L s<5L''L,8L>M, L;:L;>MMM32M3