a g g#E@shddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd l mZdd l mZdd l mZddl mZddl mZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddl$m%Z%ddl&m'Z'e rddl m(Z(ddl m)Z)ddlm*Z*dd lm+Z+dd!lm,Z,dd"l-m.Z.dd#l/m0Z0dd$l1m2Z2dd%l3m4Z4dd&lm5Z5dd'l6m7Z7dd(l6m8Z8dd)l9m:Z:dd*l9m;Z;e d+e d,d-Zd3d4e?d5eDZ@e@d6kZAe@d7kZBe@d8kZCe@d9kZDeZEzdd:lFmGZGWn eHydd:lmGZGYn0eBrddlm*ZIeJeId; ZKndZNd>ZOYn0dZQYn0dAZReRdBZSd<r8z (\d+|[abc]\d)))r:)r:r<)) _NONE_NAMEcommitT)ComputedF)r:r;)Identity) start incrementZminvalueZmaxvalueZ nominvalueZ nomaxvaluecyclecacheorder)Zon_null) _NoneName)symbol_ConstraintNamez!TypeGuard[_ConstraintNameDefined])namer0cCs|tupt|ttfSr1)r? isinstancer.rIrLr3r3r4constraint_name_defined}srOzTypeGuard[str]cCs t|tSr1)rMr.rNr3r3r4constraint_name_stringsrPz Optional[str]cCst|r |SdSr1)rPrNr3r3r4constraint_name_or_nonesrQautozOptional[Connection]zIterator[None]) connectionr0ccs~z |j}Wnty d}Yn0|dur2dVnH|st|dusDJ|dVWdqz1sh0YndVdSr1)in_transactionAttributeErrorbeginrSrTr3r3r4_ensure_scope_for_ddls    &rXcCs tr|j|dS|j|dSdS)N) hide_password)sqla_14Zrender_as_stringZ __to_string__)rrYr3r3r4url_render_as_strings r[r r"cCst|}|r|S|SdSr1)_get_connection_transactionrVrSZ transactionr3r3r4"_safe_begin_connection_transactionsr^NonecCst|}|r|dSr1)r\r@r]r3r3r4#_safe_commit_connection_transactionsr`cCst|}|r|dSr1)r\rollbackr]r3r3r4%_safe_rollback_connection_transactionsrbboolcCs*z |j}WntyYdS0|SdS)NF)rTrUrWr3r3r4_get_connection_in_transactions   rdrzIterable[ColumnElement[Any]])idxr0cCs|jSr1) expressionsrer3r3r4_idx_table_bound_expressionssrh) schema_itemr0cKs.t|dr|jfi|S|jfi|SdS)N_copy)hasattrrjcopy)rikwr3r3r4rjs rjzOptional[Transaction]cCstr |S|j}|jSdSr1)rZZget_transaction_rootZ_Connection__transaction)rSrr3r3r4r\sr\zurl.URL)r0cOs2ttjdrtjj|i|Stj|i|SdS)Ncreate)rkrURLrp)argrmr3r3r4 _create_urls rszUnion[str, None]) connectable tablename schemanamer0cCs(trt|||S|j|||SdSr1)rZr Z has_tabledialect)rtrurvr3r3r4_connectable_has_tables rxcKsLtr:|}|||WdS1s.0Yn|j||SdSr1)rZZ_operation_contextexecutebind) inspectorZ statementparamsconnr3r3r4_exec_on_inspectors ,r~cCs&ts |jSddlm}|j|juSdS)Nrr )rZZnullablesqlalchemy.sqlrZ_user_defined_nullableZNULL_UNSPECIFIED)Zmetadata_columnrr3r3r4_nullability_might_be_unsets   rcGstsdStdd|DSdS)NFcss|]}t|tVqdSr1)rMrAr6sdr3r3r4 r9z._server_default_is_computed..) has_computedanyZserver_defaultr3r3r4_server_default_is_computedsrcGstsdStdd|DSdS)NFcss|]}t|tVqdSr1)rMrCrr3r3r4rr9z._server_default_is_identity..)rZrrr3r3r4_server_default_is_identitysrr(r) constraintr0cCs*t|tr |j}|dusJ|S|jSdSr1)rMrparenttable)rrr3r3r4_table_for_constraints   rcCs<t|trdd|jDSt|tr.t|jSt|jSdS)NcSsg|] }|jqSr3)r)r6Zfkr3r3r4r8'r9z+_columns_for_constraint..)rMrelementsr _find_columnsZsqltextlistcolumnsrr3r3r4_columns_for_constraint%s    rr#)r{r include_colsr0cCs tr||dS||dSdSr1)rZZ reflect_tableZ reflecttable)r{rrr3r3r4_reflect_table.s  rcCs*t|r"t|\}}||j|S|SdSr1)_type_has_variants_get_variant_mappinggetrL)type_rwZ base_typemappingr3r3r4_resolve_for_variant9s r_variant_mappingcCs t|jSr1)rcrrr3r3r4rCsrcCs ||jfSr1)rrr3r3r4rFsrcCst|tjuSr1)typesqltypesZVariantrr3r3r4rKscCs |j|jfSr1)implrrr3r3r4rNsc sfddjD}jj}jj}jdjjj}jdjjj}ddjD}j}j}j } j } ||||||||| | f S)Ncsg|]}j|jqSr3)rrL)r6keyrr3r4r8Ssz_fk_spec..rcSsg|] }|jjqSr3)columnrL)r6elementr3r3r4r8[r9) Z column_keysrrLrrrrondeleteonupdate deferrable initially) rZsource_columnsZ source_tableZ source_schemaZ target_schema target_tableZtarget_columnsrrrrr3rr4_fk_specRs. rrcCsF|jd}|d}|dd|}|jdus:J||jjkS)Nr.)rZ _get_colspecsplitpopjoinrr)rspectokensZtablekeyr3r3r4_fk_is_self_referentialns    rcCs|jSr1)Z _type_boundrr3r3r4_is_type_boundwsrcCst}t|id|ji|S)z2locate Column objects within the given expression.r)setradd)Zclausecolsr3r3r4r~srr$zUnion[Column, ColumnClause]) collectionrr0cCsH|jdusJ||j}t|ds,t|dr:|j|n ||dS)z(remove a column from a ColumnCollection.NZ _immutableZ _readonly)rrk_parentremove)rrZ to_remover3r3r4_remove_column_from_collections  rz%Union[str, TextClause, ColumnElement]zUnion[ColumnElement, Column])rtext_r0cCspt|tr$t|tj}|||St|tr8t||St|trNt||j St|t j rdt ||St ddS)zAa workaround for the Index construct's severe lack of flexibilityz#String or text() construct expectedN)rMr.rrNULLTYPE append_columnr_textual_index_element_textual_index_columntextrr'_copy_expression ValueError)rrcr3r3r4rs         r) expressionrr0csfdd}t|i|S)NcsVt|trN|jdurN|jurN|jjvr6j|jSt|}||SndSdSr1)rMrrrLrrjr)colrrr3r4replaces   z!_copy_expression..replace)rZreplacement_traverse)rrrr3rr4rs rc@s.eZdZdZdZddddddZd d Zd S) 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. Z_textual_idx_elementrrr_)rrr0cCs8||_||_|j|_t|jjtj|_||jdSr1) rrrrrrr fake_columnr)selfrrr3r3r4__init__s z_textual_index_element.__init__cCs|jgSr1)r)rr3r3r4 get_childrensz#_textual_index_element.get_childrenN)__name__ __module__ __qualname____doc__Z__visit_name__rrr3r3r3r4rsrr%)rcompilerr0cKs|j|jfi|Sr1)processrrrrmr3r3r4_render_textual_index_columnsrc@s eZdZdS)_literal_bindparamN)rrrr3r3r3r4rsrcKs|j|fi|Sr1)Zrender_literal_bindparamrr3r3r4_render_literal_bindparamsrcCs t|jSr1)rrfrgr3r3r4_get_index_expressionssrcCsddt|DS)NcSsg|]}t|ddqS)rLN)getattr)r6expr3r3r4r8r9z+_get_index_column_names..)rrgr3r3r4_get_index_column_namessrrr)rr0cCstr |jSiSdSr1)sqla_13kwargs)rr3r3r4_column_kwargssrzUnion[Index, Constraint]zOptional[Dialect])rrwr0cCs|jdurdS|dusJtr.|jj|ddSt|jdrFt|j}nt}|t|jdd}|j|d}t |t j r| |d}| |S|j|SdS)NFZ_alembic_quotequote)rrN)rLrZidentifier_preparerformat_constraintrkrrr. __class__rMrrZ ddl_compilerZ_prepared_index_name)rrwZquoted_name_clsnew_namedr3r3r4_get_constraint_final_names$       rzUnion[Constraint, Index]cCsDtr6|jdurdS|dusJ|jj|dd}|duS|jduSdS)NFr)rZrLrr)rrwrLr3r3r4_constraint_is_named%s  rr!) mysql_dialectr0cCstr |jSt|jo|jSdSr1)rZZ is_mariadbrcZserver_version_info _is_mariadbrr3r3r4r4s rcCs|jSr1) _mariadb_normalized_version_inforr3r3r4r>srzUnion[TableClause, Table]r&)rr0cCs tr|S|jddSdS)NT)inline)rZinsertr)rr3r3r4_insert_inlineBs r)create_mock_engine)select create_enginecKstdd|dS)Nz postgresql://Zmock)Zstrategyexecutorr)rrrmr3r3r4rOsrr*cOstjt|fi|Sr1)rrr)rrmr3r3r4_selectTsr)indexr0cCs8|jD],}t|tr|j}q t|tr,|jrdSqdS)NTF)rfrMrrrZ is_literal)rexprr3r3r4is_expression_indexXs   r)T) __future__r contextlibretypingrrrrrrr r Z sqlalchemyr r rrrrZsqlalchemy.enginerZsqlalchemy.ext.compilerrZsqlalchemy.schemarrrrrZsqlalchemy.sql.elementsrrrrrZsqlalchemy.sql.visitorsrZtyping_extensionsrrrr r!r"Zsqlalchemy.engine.reflectionr#Zsqlalchemy.sql.baser$Zsqlalchemy.sql.compilerr%Zsqlalchemy.sql.dmlr&r'Zsqlalchemy.sql.schemar(r)Zsqlalchemy.sql.selectabler*r+r,r5tuplefindallZ_versrrZZ sqla_14_26Zsqla_2Zsqlalchemy_versionZsqlalchemy.sql.namingr? ImportErrorZlegacy_connectionrkZsqla_1xrArrZhas_computed_reflectionrCZ has_identityZ_identity_options_attrsZ_identity_attrsrIZsqlalchemy.utilrJr.rKZ_ConstraintNameDefinedrOrPrQZAUTOINCREMENT_DEFAULTcontextmanagerrXr[r^r`rbrdrhrjr\rsrxr~rrrrrrrZ TypeEnginerrrrrrrrrrrrrrrrrrrrrrrrrrr3r3r3r4s                                                          (