o g9@srddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddl$m%Z%ddl&m'Z' ddl(m)Z)erdd lm*Z*dd!lm+Z+dd"lm,Z,dd#lm-Z-dd$l.m/Z/dd%l.m0Z0dd&l.m1Z1dd'lm2Z2dd(l3m4Z4dd)lm5Z5dd*l6m7Z7dd+l6m8Z8ed,ed-d.Z9Gd/d0d0e Z:dd5d6Z;ee>d:kZ?e>d;kZ@e>dErCz (\d+|[abc]\d)))rErF)r/Type[ClauseElement]dialects0Callable[[_CompilerProtocol], _CompilerProtocol]cGr4r5r6)r/rKr6r6r8compilesOsrM)rMFidentity&Union[Identity, schema.Sequence, None]dialect_kwargsboolDict[str, Any]csduriStr$tdsJ}|r"ttsJ|j|Si}ttjr;j |d<j dur;j |d<d}|fdd|D|S)N_as_dictalwayson_null) start incrementminvaluemaxvalue nominvalue nomaxvaluecyclecacheordercs*i|]}t|ddur|t|dqSr5)getattrrAkeyrNr6r8 zs  z._get_identity_options_dict..) identity_has_dialect_kwargshasattrrS isinstancerupdaterPrr!rTrU)rNrPas_dictattrsr6rbr8_get_identity_options_dictZs*      rj) _NoneName)symbolname_ConstraintName!TypeGuard[_ConstraintNameDefined]cCs|tup t|ttfSr5)rrfr3rkrmr6r6r8constraint_name_definedsrqTypeGuard[str]cCs t|tSr5)rfr3rpr6r6r8constraint_name_string rs Optional[str]cCst|r|SdSr5)rsrpr6r6r8constraint_name_or_nonesrvauto connectionOptional[Connection]Iterator[None]ccsz|j}Wn tyd}Ynw|durdVdS|s>|dus$J| dVWddS1s7wYdSdVdSr5)in_transactionAttributeErrorbeginrxr{r6r6r8_ensure_scope_for_ddls     " rr$r&cCs|}|r|S|Sr5)get_transactionr}rx transactionr6r6r8"_safe_begin_connection_transactionsrNonecC|}|r |dSdSr5)rcommitrr6r6r8#_safe_commit_connection_transaction rcCrr5)rrollbackrr6r6r8%_safe_rollback_connection_transactionrrcCs$z|j}W|StyYdSw)NF)r{r|r~r6r6r8_get_connection_in_transactions  ridxr"Iterable[ColumnElement[Any]]cC|jSr5) expressions)rr6r6r8_idx_table_bound_expressionssr schema_itemcKs*t|dr |jdi|S|jdi|S)N_copyr6)rercopy)rr1r6r6r8rs r connectable tablename schemanameUnion[str, None]cCs|j|||Sr5)dialect has_table)rrrr6r6r8_connectable_has_tablesrcKs8|}|||WdS1swYdSr5)_operation_contextexecute) inspector statementparamsconnr6r6r8_exec_on_inspectors  $rcCsddlm}|j|juS)Nrr)sqlalchemy.sqlr_user_defined_nullableNULL_UNSPECIFIED)metadata_columnrr6r6r8_nullability_might_be_unsets  rcGtdd|DS)Ncs|] }t|tjVqdSr5)rfrComputedrAsdr6r6r8 z._server_default_is_computed..anyserver_defaultr6r6r8_server_default_is_computedrcGr)Ncsrr5)rfrr!rr6r6r8rrz._server_default_is_identity..rrr6r6r8_server_default_is_identityrr constraintr*r#cCs&t|tr|j}|dusJ|S|jSr5)rfrparenttable)rrr6r6r8_table_for_constraints  rcCs8t|tr dd|jDSt|trt|jSt|jS)NcSsg|]}|jqSr6)r)rAfkr6r6r8rCsz+_columns_for_constraint..)rfrelementsr _find_columnssqltextlistcolumnsrr6r6r8_columns_for_constraints    rcCs&t|rt|\}}||j|S|Sr5)_type_has_variants_get_variant_mappinggetrm)type_r base_typemappingr6r6r8_resolve_for_variants r_variant_mappingcCs t|jSr5)rQrrr6r6r8rrtrcCs ||jfSr5)rrr6r6r8rrtrcCst|tjuSr5)typesqltypesVariantrr6r6r8r!scCs |j|jfSr5)implrrr6r6r8r$s rrc strjdus JjdusJtjtsJfddjD}jj}jj}jdj j j}jdj j j}ddjD}j }j }j } j} ||||||||| | f S)Ncsg|]}j|jqSr6)rrmr`rr6r8rC.s z_fk_spec..rcSsg|]}|jjqSr6)columnrm)rAr/r6r6r8rC6rD)r rrrfrr# column_keysrmrrrondeleteonupdate deferrable initially) rsource_columns source_table source_schema target_schema target_tabletarget_columnsrrrrr6rr8_fk_spec(s6 rcCsF|jd}|d}|dd|}|jdusJ||jjkS)Nr.)r _get_colspecsplitpopjoinrra)rspectokenstablekeyr6r6r8_fk_is_self_referentialIs     rcCrr5) _type_boundrr6r6r8_is_type_boundRsrcCst}t|id|ji|S)z2locate Column objects within the given expression.r)setradd)clausecolsr6r6r8rYsr collectionr'r%Union[Column[Any], ColumnClause[Any]]cCsJ|jdusJ||j}t|dst|dr|j|dS||dS)z(remove a column from a ColumnCollection.N _immutable _readonly)rare_parentremove)rr to_remover6r6r8_remove_column_from_collectionas  rrtext_*Union[str, TextClause, ColumnElement[Any]]&Union[ColumnElement[Any], Column[Any]]cCslt|trt|tj}|||St|trt||St|tr't||j St|t j r2t ||St d)zAa workaround for the Index construct's severe lack of flexibilityz#String or text() construct expected)rfr3rrNULLTYPE append_columnr_textual_index_element_textual_index_columntextrr)_copy_expression ValueError)rrcr6r6r8rss         r expressionrcsfdd}t|i|S)NcsPt|tr&|jdur&|jur&|jjvrj|jSt|}||SdSr5)rfrrrmrrr)colrrr6r8replaces     z!_copy_expression..replace)rreplacement_traverse)rrrr6rr8rs rc@s&eZdZdZdZdd d Zd 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. _textual_idx_elementrr#rrr2rcCs8||_||_|j|_t|jjtj|_||jdSr5) rrrarrrr fake_columnr)r7rrr6r6r8__init__s z_textual_index_element.__init__cKs|jgSr5)r)r7r1r6r6r8 get_childrensz#_textual_index_element.get_childrenN)rr#rrr2r)r:r;r<__doc____visit_name__rrr6r6r6r8rs   rr0r(cKs|j|jfi|Sr5)processrr/r0r1r6r6r8_render_textual_index_columnsr c@s eZdZdS)_literal_bindparamN)r:r;r<r6r6r6r8r sr cKs|j|fi|Sr5)render_literal_bindparamr r6r6r8_render_literal_bindparamsrUnion[Index, Constraint]rOptional[Dialect]cCs*|jdurdS|dus J|jj|ddSNF)_alembic_quotermidentifier_preparerformat_constraint)rrr6r6r8_get_constraint_final_names  rUnion[Constraint, Index]cCs2|jdurdS|dus J|jj|dd}|duSrr)rrrmr6r6r8_constraint_is_nameds  rindexcCs|jD] }t|r dSqdSNTF)r is_expression)rexprr6r6r8is_expression_indexs rrcCs2t|tr |j}t|tst|tr|jrdSdSr)rfrr/r is_literal)rr6r6r8rs  r)r=r3r2r>)r/rJrKr3r2rL)F)rNrOrPrQr2rR)rmrnr2ro)rmrnr2rr)rmrnr2ru)rxryr2rz)rxr$r2r&)rxr$r2r)rxryr2rQ)rr"r2r)rr,r2r,)rr$rr3rrr2rQ)r2rQ)rr*r2r#)rrr2r)rrr2rQ)rr*r2rQ)rr'rrr2r)rr#rrr2r)rr,rr#r2r,)r/rr0r(r2r3)r/r r0r(r2r3)rrrrr2ru)rrrrr2rQ)rr"r2rQ)rrr2rQ)u __future__r contextlibretypingrrrrrrr r r r r r sqlalchemyrrrrrsqlalchemy.schemarrrrrsqlalchemy.sql.basersqlalchemy.sql.elementsrrrrsqlalchemy.sql.visitorsrtyping_extensionsrsqlalchemy.sql.namingrr r!r"r#sqlalchemy.enginer$r%r&r'sqlalchemy.sql.compilerr(r)sqlalchemy.sql.schemar*r+r,r.r@tuplefindall_vers sqla_14_18 sqla_14_26sqla_2sqlalchemy_versionrMsqlalchemy.ext.compiler issubclassrdrjrksqlalchemy.utilrlr3rn_ConstraintNameDefinedrqrsrvAUTOINCREMENT_DEFAULTcontextmanagerrrrrrrrrrrrrrrrre TypeEnginerrrrrrrrrrr r rrrrrr6r6r6r8s                                              )                   !