o DfPL@sdddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl#m%Z%ddl#m&Z&ddl#m'Z'ddl#m(Z(dd l)m*Z*dd!l+m,Z,er%dd"lm-Z-dd#lm.Z.dd$lm/Z/dd%lm0Z0dd&lm1Z1dd'lm2Z2dd(l3m4Z4dd)l!m5Z5dd*l6m7Z7dd+l8m9Z9dd,l#m:Z:dd-l;mm?Z?dd0l>m@Z@ed1ed2d3ZAGd4d5d5e ZBdd:d;ZCeDdeDZFeFd?kZGeFd@kZHeFdAkZIeFdBkZJeFdCkZKeZLzddDlMmNZNWneOyuddDl#mNZNYnwGdEdFdFZPerddKdLZQnddMlRmQZQzddNlmSZSWneOyesGdOdPdPePZSdQZTdQZUYnwdRZTeFdSkZUzddTlmVZVWneOyesGdUdVdVePZVdQZWYnweXeVe"ZY Qddd\d]ZZdRZWeKrdd^l!m[Z[ndd_l\m]Z[ede^e[fZ_ee^e[fZ`ddcddZaddfdgZbddidjZcdkZdejeddodpZfddqdrZgddudvZhddxdyZiddzd{Zjdd|d}ZkdddZldddZmdddZndddZodddZpddZqddZrdddZsdddZtd ddZuddZvd ddZwddZxeyejzdrddZ{ddZ|nddZ{ddZ|d ddZ}d ddZ~d ddZddZdddZdddÄZdddDŽZGddɄdej:ZeQeddd̈́ZGddτde$ZeQedddфZdddքZdddۄZdddބZdddZddZdddZeHrddlmZddlmZeZnddlmZddZdddZdddZdddZdS() annotationsN)Any)Callable)Dict)Iterable)Iterator)Mapping)Optional)Protocol)Set)Type) TYPE_CHECKING)TypeVar)Union) __version__)inspectschema)sql)types)url)CheckConstraint)Column)ForeignKeyConstraint)visitors) DialectKWArgs) BindParameter) ColumnClause) quoted_name) TextClause)UnaryExpression)traverse) TypeGuard) ClauseElement)Index)Table) Connection)Dialect) Transaction) Inspector)ColumnCollection) SQLCompiler)Insert) ColumnElement) Constraint) SchemaItem)Select) TableClause_CE)zColumnElement[Any]r/)boundc@seZdZd ddZd S) _CompilerProtocolelementrcompilerkwreturnstrcKdSN)selfr5r6r7r<r<Q/home/ubuntu/webapp/venv/lib/python3.10/site-packages/alembic/util/sqla_compat.py__call__>sz_CompilerProtocol.__call__N)r5rr6rr7rr8r9)__name__ __module__ __qualname__r?r<r<r<r>r4=sr4valuer9r8Union[int, str]cCszt|WS|YSr;)int)rCr<r<r> _safe_intBs rFcCsg|]}t|qSr<)rF).0xr<r<r> JrIz (\d+|[abc]\d)))rK)rKrM)rKrM)) _NONE_NAMEc@seZdZdZdS) _Unsupportedz.Placeholder for unsupported SQLAlchemy classesN)r@rArB__doc__r<r<r<r>rRZsrRr5Type[ClauseElement]dialects0Callable[[_CompilerProtocol], _CompilerProtocol]cGr:r;r<)r5rUr<r<r>compiles`srW)rW)Computedc@ eZdZdS)rXNr@rArBr<r<r<r>rXmrXFT)rKrL)Identityc@rY)r]NrZr<r<r<r>r]{r[r]identity&Union[Identity, schema.Sequence, None]dialect_kwargsboolDict[str, Any]csduriStr}|rttsJ|j|Si}ttr3j|d<jdur3j|d<d}|fdd|D|S)Nalwayson_null) start incrementminvaluemaxvalue nominvalue nomaxvaluecyclecacheordercs*i|]}t|ddur|t|dqSr;)getattrrGkeyr^r<r> s  z._get_identity_options_dict..) identity_has_dialect_kwargs_as_dict isinstancerupdater`r]rcrd)r^r`as_dictattrsr<rqr>_get_identity_options_dicts(      ry) _NoneName)symbolname_ConstraintName!TypeGuard[_ConstraintNameDefined]cCs|tup t|ttfSr;)rQrur9rzr|r<r<r>constraint_name_definedsrTypeGuard[str]cCs t|tSr;)rur9rr<r<r>constraint_name_strings r Optional[str]cCst|r|SdSr;)rrr<r<r>constraint_name_or_nonesrauto connectionOptional[Connection]Iterator[None]ccsz|j}Wn tyd}Ynw|durdVdS|s>|dus$J| dVWddS1s7wYdSdVdSr;)in_transactionAttributeErrorbeginrrr<r<r>_ensure_scope_for_ddls     " rcCstr|j|dS|j|dS)N) hide_password)sqla_14render_as_string __to_string__)rrr<r<r>url_render_as_string  rr&r(cCst|}|r|S|Sr;)_get_connection_transactionrr transactionr<r<r>"_safe_begin_connection_transactionsrNonecCt|}|r |dSdSr;)rcommitrr<r<r>#_safe_commit_connection_transaction rcCrr;)rrollbackrr<r<r>%_safe_rollback_connection_transactionrrcCs$z|j}W|StyYdSw)NF)rrrr<r<r>_get_connection_in_transactions  ridxr$Iterable[ColumnElement[Any]]cC|jSr;) expressions)rr<r<r>_idx_table_bound_expressions r schema_itemcKs*t|dr |jdi|S|jdi|S)N_copyr<)hasattrrcopy)rr7r<r<r>rs rOptional[Transaction]cCstr|S|j}|jSr;)rget_transaction_root_Connection__transaction)rrr<r<r>rsrurl.URLcOs.ttjdrtjj|i|Stj|i|S)Ncreate)rrURLr)argr7r<r<r> _create_url"s r connectable tablename schemanameUnion[str, None]cCs$tr t|||S|j|||Sr;)rr has_tabledialect)rrrr<r<r>_connectable_has_table)s rcKsJtr|}|||WdS1swYdS|j||Sr;)r_operation_contextexecutebind) inspector statementparamsconnr<r<r>_exec_on_inspector4s   $rcCs"ts|jSddlm}|j|juS)Nrr)rnullablesqlalchemy.sqlr_user_defined_nullableNULL_UNSPECIFIED)metadata_columnrr<r<r>_nullability_might_be_unset<s   rcGtsdStdd|DS)NFcs|]}t|tVqdSr;)rurXrGsdr<r<r> Kz._server_default_is_computed..) has_computedanyserver_defaultr<r<r>_server_default_is_computedGrcGr)NFcsrr;)rur]rr<r<r>rRrz._server_default_is_identity..)rrrr<r<r>_server_default_is_identityNrr constraintr.r%cCs&t|tr|j}|dusJ|S|jSr;)rurparenttable)rrr<r<r>_table_for_constraintUs  rcCs8t|tr dd|jDSt|trt|jSt|jS)NcSsg|]}|jqSr<)r)rGfkr<r<r>rI`sz+_columns_for_constraint..)rurelementsr _find_columnssqltextlistcolumnsrr<r<r>_columns_for_constraint^s    rrr)rcCstr||dS||dSr;)r reflect_table reflecttable)rrr<r<r>_reflect_tablegs  rcCs&t|rt|\}}||j|S|Sr;)_type_has_variants_get_variant_mappinggetr|)type_r base_typemappingr<r<r>_resolve_for_variantps r_variant_mappingcCs t|jSr;)rarrr<r<r>rz rcCs ||jfSr;)rrr<r<r>r}rrcCst|tjuSr;)typesqltypesVariantrr<r<r>rscCs |j|jfSr;)implrrr<r<r>rs rrc strjdus JjdusJtjtsJfddjD}jj}jj}jdj j j}jdj j j}ddjD}j }j }j } j} ||||||||| | f S)Ncsg|]}j|jqSr<)rr|rorr<r>rIs z_fk_spec..rcSsg|]}|jjqSr<)columnr|)rGr5r<r<r>rIrJ)r rrrurr% column_keysr|rrrondeleteonupdate deferrable initially) rsource_columns source_table source_schema target_schema target_tabletarget_columnsrrrrr<rr>_fk_specs6 rcCsF|jd}|d}|dd|}|jdusJ||jjkS)Nr.)r _get_colspecsplitpopjoinrrp)rspectokenstablekeyr<r<r>_fk_is_self_referentials     r cCrr;) _type_boundrr<r<r>_is_type_boundsr cCst}t|id|ji|S)z2locate Column objects within the given expression.r)setr!add)clausecolsr<r<r>rsr 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)rpr_parentremove)rr to_remover<r<r>_remove_column_from_collections  rtext_*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)rur9rrNULLTYPE append_columnr_textual_index_element_textual_index_columntextrr-_copy_expression ValueError)rrcr<r<r>rs         r expressionrcsfdd}t|i|S)NcsPt|tr&|jdur&|jur&|jjvrj|jSt|}||SdSr;)rurrr|r"rr)colr"rr<r>replaces     z!_copy_expression..replace)rreplacement_traverse)r#rr&r<r%r>r s r c@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%rrr8rcCs8||_||_|j|_t|jjtj|_||jdSr;) rrrprrrr fake_columnr)r=rrr<r<r>__init__ s z_textual_index_element.__init__cCs|jgSr;)r))r=r<r<r> get_childrensz#_textual_index_element.get_childrenN)rr%rrr8r)r@rArBrS__visit_name__r*r+r<r<r<r>rs   rr6r+cKs|j|jfi|Sr;)processrr5r6r7r<r<r>_render_textual_index_columnsr/c@rY)_literal_bindparamNrZr<r<r<r>r0r[r0cKs|j|fi|Sr;)render_literal_bindparamr.r<r<r>_render_literal_bindparam#sr2r$rrcCstr|jSiSr;)sqla_13kwargs)r$r<r<r>_column_kwargs*sr5Union[Index, Constraint]rOptional[Dialect]cCs|jdurdS|dus Jtr|jj|ddSt|jdr#t|j}nt}|t|jdd}|j|d}t |t j rE| |d}| |S|j|S)NF_alembic_quotequote)r:r)r|ridentifier_preparerformat_constraintrrrr9 __class__rurr$ ddl_compiler_prepared_index_name)rrquoted_name_clsnew_namedr<r<r>_get_constraint_final_name1s         rCUnion[Constraint, Index]cCs@tr|jdur dS|dusJ|jj|dd}|duS|jduS)NFr8)rr|r;r<)rrr|r<r<r>_constraint_is_namedVs   rE mysql_dialectr'cCstr|jSt|jo |jSr;)r is_mariadbraserver_version_info _is_mariadbrFr<r<r>rIes rIcCrr;) _mariadb_normalized_version_inforJr<r<r>rKorrKUnion[TableClause, Table]r,cCstr|S|jddS)NT)inline)rinsertrM)rr<r<r>_insert_inlinesrrO)create_mock_engine)select create_enginecKstdd|dS)Nz postgresql://mock)strategyexecutorrR)rrVr7r<r<r>rPsrPr0cOstjt|fi|Sr;)rrQr)rr7r<r<r>_selectsrWindexcCs|jD] }t|r dSqdSNTF)r is_expression)rXexprr<r<r>is_expression_indexs r\r[cCs2t|tr |j}t|tst|tr|jrdSdSrY)rur r5r is_literal)r[r<r<r>rZs  rZ)rCr9r8rD)r5rTrUr9r8rV)F)r^r_r`rar8rb)r|r}r8r~)r|r}r8r)r|r}r8r)rrr8r)T)rr&r8r()rr&r8r)rrr8ra)rr$r8r)rr2r8r2)rr&r8r)r8r)rr&rr9rrr8ra)r8ra)rr.r8r%)rr)rr%r8r)rrr8r)rrr8ra)rr.r8ra)rr*rrr8r)rr%rrr8r)r#r2rr%r8r2)r5rr6r+r8r9)r5r0r6r+r8r9)r$rr8r)rr6rr7r8r)rrDrr7r8ra)rFr'r8ra)rrLr8r,)r8r0)rXr$r8ra)r[rr8ra) __future__r contextlibretypingrrrrrrr r r r r rr sqlalchemyrrrrrrsqlalchemy.enginersqlalchemy.schemarrrrrsqlalchemy.sql.basersqlalchemy.sql.elementsrrrrr sqlalchemy.sql.visitorsr!typing_extensionsr"r#r$r%r&r'r(sqlalchemy.engine.reflectionr)r*sqlalchemy.sql.compilerr+sqlalchemy.sql.dmlr,r-sqlalchemy.sql.schemar.r/sqlalchemy.sql.selectabler0r1r2r4rFtuplefindall_versr3r sqla_14_18 sqla_14_26sqla_2sqlalchemy_versionsqlalchemy.sql.namingrQ ImportErrorrRrWsqlalchemy.ext.compilerrXrhas_computed_reflectionr] has_identity issubclassrsryrzsqlalchemy.utilr{r9r}_ConstraintNameDefinedrrrAUTOINCREMENT_DEFAULTcontextmanagerrrrrrrrrrrrrrrrrrrrr TypeEnginerrrr r rrrr rr/r0r2r5rCrErIrKrOrPrQ _sa_selectrWrSr\rZr<r<r<r>s4                                                     '                     !       %