o hru@sUddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!dd l!m"Z"er,dd!lm#Z#dd"lm$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+l1m3Z3dd,l1m4Z4dd-l1m5Z5dd.l6m7Z7dd/l8m9Z9dd0lm:Z:dd1l;mZ>dd3l=m?Z?e@eAZBGd4d5d5eCZDiZEd6eFd7<Gd8d9d9eDd:ZGGd;d<dBs zImplMeta.__init__N)r6r7r8r9r:r;)__name__ __module__ __qualname__r>rBrBrBrCr5Asr5zDict[str, Type[DefaultImpl]]r?c@sZeZdZUdZdZdZdZddhfZded<d Z d ed <d Z d ed<dddZ e dd d!Z dd$d%Zdd-d.Zdd1d2Zdd6d7Zedd8d9Zd:d:efddDdEZ :dddHdIZ :  : : : :  : : : : :ddd\d]Z :dddadbZ :dddcddZddfdgZddidjZ :dddndoZddpdqZddrdsZddvdwZ ddxdyZ!ddzd{Z"dd|d}Z#dd~dZ$ ddddZ%dddZ&dddZ'dddZ(dddZ)ddZ*dddZ+ddZ, ddddZ-dddZ.dddZ/ddZ0dddZ1dddZ2dddZ3dddZ4dd„Z5dddƄZ6ddd̄Z7ddd΄Z8dddфZ9dddքZ:dd؄Z;ddd܄Zjs  zDefaultImpl.__init__Type[DefaultImpl]cCs t|jSN)r?name)r@rQrBrBrCget_by_dialects zDefaultImpl.get_by_dialectrr7cCs,|jdusJ|j|d|jdS)Nz )rXwriteflush)rarrBrBrC static_outputszDefaultImpl.static_output version_tableversion_table_schema Optional[str]version_table_pkkwrrcKs@t|ttdtddd|d}|r|td|dd|S)aGenerate a :class:`.Table` object which will be used as the structure for the Alembic version table. Third party dialects may override this hook to provide an alternate structure for this :class:`.Table`; requirements are only that it be named based on the ``version_table`` parameter and contains at least a single string-holding column named ``version_num``. .. versionadded:: 1.14 version_num F)nullabler_pkcrd)rrrrappend_constraintr)rarirjrlrmvtrBrBrCversion_table_impls zDefaultImpl.version_table_implbatch_opr4cCdS)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. FrB)rarvrBrBrCrequires_recreate_in_batchs z&DefaultImpl.requires_recreate_in_batch batch_implr3tablecCrw)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. NrB)raryrzrBrBrCprep_table_for_batchz DefaultImpl.prep_table_for_batchcC|jSrc)rRrarBrBrCbindszDefaultImpl.bindN constructUnion[Executable, str]execution_optionsOptional[Mapping[str, Any]] multiparams%Optional[Sequence[Mapping[str, Any]]]paramsMapping[str, Any]Optional[CursorResult]cCst|tr t|}|jrO|dus|rtd|jr't|tjs'tddid}ni}t r2t|t s2J|j d d|j i|}| t|dd|jdS|j}|dusXJ|rb|jd i|}|rl|durltd|rt|||S|||S) Nz&SQL parameters not allowed with as_sqlr]T)compile_kwargsrQ z z2Can't send params and multiparams at the same timerB) isinstancer7rrT TypeErrorr]r DDLElementdictrr(compilerQrhreplacestripcommand_terminatorrRrexecute)rarrrr compile_kwcompiledconnrBrBrC_execs<      zDefaultImpl._execsqlOptional[dict[str, Any]]cCs|||dSrc)r)rarrrBrBrCrszDefaultImpl.execute table_name column_namerpserver_default%Union[_ServerDefault, Literal[False]]rdtype_Optional[TypeEngine]r autoincrementcomment$Optional[Union[str, Literal[False]]]existing_comment existing_typeexisting_server_defaultOptional[_ServerDefault]existing_nullableexisting_autoincrementc Ks,|dus|durtjddd|dur#|tj||||| | | | d|durWi}t|| r3tj}nt|| rAtj }||d<ntj }|||||f|| | | | d||durk|tj ||||| | | | d| dur|tj ||| || | | | d|dur|tj ||||| | | ddSdS)NzBautoincrement and existing_autoincrement only make sense for MySQL) stacklevel)rrrrrFimpl)rrrr)r warnrrColumnNullabler!_server_default_is_computedComputedColumnDefault_server_default_is_identityIdentityColumnDefault ColumnDefault ColumnType ColumnComment ColumnName)rarrrprrdrrrrrrrrrrmcls_rBrBrC alter_columns     zDefaultImpl.alter_columncolumn Column[Any]!Optional[Union[str, quoted_name]]cC|tj|||ddSNr)rr AddColumn)rarrrrBrBrC add_columnoszDefaultImpl.add_columncKrr)rr DropColumn)rarrrrmrBrBrC drop_columnwszDefaultImpl.drop_columnconstcCs,|jdus ||r|t|dSdSrc) _create_rulerr AddConstraintrarrBrBrCadd_constraintszDefaultImpl.add_constraintr+cC|t|dSrc)rrDropConstraintrrBrBrCdrop_constraintzDefaultImpl.drop_constraintold_table_namenew_table_nameUnion[str, quoted_name]cCrr)rr RenameTable)rarrrrBrBrC rename_tableszDefaultImpl.rename_tablecKs|jj||jd|d|tj|fi||jj||jd|d|jD] }|t|q%|j j o8|j j }|j }|rE|rE| ||jD]}|j }|rV|rV||qHdSNF) checkfirst _ddl_runner)dispatch before_createrRrr CreateTable after_createindexes CreateIndexrQsupports_commentsinline_commentsrcreate_table_commentcolumnscreate_column_comment)rarzrmindex with_commentrrrBrBrC create_tables(      zDefaultImpl.create_tablecKsH|jj||jd|d|tj|fi||jj||jd|ddSr)r before_droprRrr DropTable after_drop)rarzrmrBrBrC drop_tables   zDefaultImpl.drop_tablerr-cK|tj|fi|dSrc)rrrrarrmrBrBrC create_indexzDefaultImpl.create_indexcCrrc)rrSetTableCommentrarzrBrBrCrrz DefaultImpl.create_table_commentcCrrc)rrDropTableCommentrrBrBrCdrop_table_commentrzDefaultImpl.drop_table_commentcCrrc)rrSetColumnComment)rarrBrBrCrrz!DefaultImpl.create_column_commentcKrrc)rr DropIndexrrBrBrC drop_indexrzDefaultImpl.drop_indexTUnion[TableClause, Table]rows List[dict] multiinsertc st|ts td|rt|dtstd|jr8|D]}|jdifdd| DqdS|r]|rI|j|ddS|D]}|jdi|qKdSdS)Nz List expectedrzList of dictionaries expectedcs:i|]\}}|t|tjstj||j|jdn|qS))r)rr!_literal_bindparamcr=).0kvrzrBrC s z+DefaultImpl.bulk_insert..)rrB) rlistrrrTrinsertinlinevaluesitems)rarzrrrowrBrrC bulk_inserts0    zDefaultImpl.bulk_insertrParamsc Cs|jj|j}td|}g}d}|D]}td|r!|}q||qt |d|ddgi}|r\td|D]}d|vrS| d\} } | |j | <q<|j | q<|S)Nz[\w\-_]+|\(.+?\)z^\(.*\)$rrz[^(),]+=)rQ type_compilerprocessr=lowerrefindallmatchappendrsplitrkwargsargs) rar definitiontokens term_tokens paren_termtokenrtermkeyvalrBrBrC_tokenize_column_types    z!DefaultImpl._tokenize_column_typeinspector_paramsmetadata_paramscCs||j|jkrdSdd|jD}d|jg|j}d|jg|j}|D]}||h|s8|j|jh|r;dSq&dS)NTcSsg|] }dd|DqS)cSsh|]}|qSrB)r)rtrBrBrC z=DefaultImpl._column_types_match...rB)rbatchrBrBrC sz3DefaultImpl._column_types_match.. F)token0rLjoinrissubset)rarrsynonymsinspector_all_termsmetadata_all_termsrrBrBrC_column_types_matchs&   zDefaultImpl._column_types_matchinspected_params meta_paramscCst|jt|jkr|j|jkrdSt|jt|jkr$|j|jkr$dSd|j}d|j}|jD]}t||}t||}|rV|rV|d|dkrVdSq7dS)aWe want to compare column parameters. However, we only want to compare parameters that are set. If they both have `collation`, we want to make sure they are the same. However, if only one specifies it, dont flag it for being less specific FrrT) lenrr rrrMrsearchgroup)rar%r&inspmetaregmimmrBrBrC_column_args_match$s     zDefaultImpl._column_args_matchinspector_columnmetadata_columncCs8||}||}|||sdS|||sdSdS)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)rr$r/)rar0r1rrrBrBrC compare_typeEs    zDefaultImpl.compare_typecCs||kSrcrB)rar0r1rendered_metadata_defaultrendered_inspector_defaultrBrBrCcompare_server_defaultUsz"DefaultImpl.compare_server_default conn_uniquesSet[UniqueConstraint] conn_indexes Set[Index]metadata_unique_constraintsmetadata_indexescCdSrcrB)rar6r8r:r;rBrBrCcorrect_for_autogen_constraints^sz+DefaultImpl.correct_for_autogen_constraintscCs(|jj|jurt|d||d<dSdS)Nexpr)r=_type_affinityr)raexistingexisting_transfernew_typerBrBrCcast_for_batch_migrategs  z"DefaultImpl.cast_for_batch_migrater>r(is_server_defaultcKsddd}t|j|j|dS)zdRender a SQL expression that is typically a server default, index expression, etc. TF)r] include_table)rQr)r7rrQ)rar>rDrmrrBrBrCrender_ddl_sql_exprms zDefaultImpl.render_ddl_sql_expr inspectorr'rcCr}rc)autogen_column_reflect)rarGrBrBrC_compat_autogen_column_reflect{sz*DefaultImpl._compat_autogen_column_reflectconn_fksSet[ForeignKeyConstraint] metadata_fkscCr<rcrB)rarJrLrBrBrCcorrect_for_autogen_foreignkeys~sz+DefaultImpl.correct_for_autogen_foreignkeyscCrw)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. NrB)rarGrz column_inforBrBrCrHr|z"DefaultImpl.autogen_column_reflectcCrw)zA hook called when :meth:`.EnvironmentContext.run_migrations` is called. Implementations can set up per-migration-run state here. NrBr~rBrBrCstart_migrationsr|zDefaultImpl.start_migrationscC|d|jdS)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`. BEGINNrhrr~rBrBrC emit_beginzDefaultImpl.emit_begincCrP)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`. COMMITNrRr~rBrBrC emit_commitrTzDefaultImpl.emit_committype_objr0autogen_contextr2Union[str, Literal[False]]cCrwNFrB)rarWrXrBrBrC render_typeszDefaultImpl.render_typecCsvt||tdhd\}}t|dd}t|dd}t|t|kr&|d||j||t|p9t|t|kfS)Nalways)skip)_compare_identity_optionsrIdentitygetattrrUadddifference_updaterP)rametadata_identityinspector_identitydiffignored meta_alwaysinspector_alwaysrBrBrC_compare_identity_defaults    z%DefaultImpl._compare_identity_defaultmetadata_indexreflected_indexcCs0t|j}t|j}||krd|d|SdS)Nzunique=z to unique=)rUunique)rarjrk conn_unique meta_uniquerBrBrC_compare_index_uniques  z!DefaultImpl._compare_index_unique constraint _autogen._Copts_constraint_sig[_autogen._C]cKtjd||fi|S)NTrfrom_constraintrarprrrBrBrC_create_metadata_constraint_sigz+DefaultImpl._create_metadata_constraint_sigcKrtrZrurwrBrBrC _create_reflected_constraint_sigryz,DefaultImpl._create_reflected_constraint_sigrcCsg}|||}|r||||}||}t|s Jt|s'J||fD] }|jr7td|j q+|j |j krK|d|j d|j |rRt |St S)zCompare two indexes by comparing the signature generated by ``create_index_sig``. This method returns a ``ComparisonResult``. zGenerating approximate signature for index %s. The dialect implementation should either skip expression indexes or provide a custom implementation. expression  to )ror rxrzr is_index_sighas_expressionslogwarningr column_namesr DifferentEqual)rarjrkmsg unique_msgm_sigr_sigsigrBrBrCcompare_indexess0       zDefaultImpl.compare_indexesmetadata_constraintr.reflected_constraintcCsF||}||}|j}|j}||krtd|d|StS)aCompare two unique constraints by comparing the two signatures. The arguments are two tuples that contain the unique constraint and the signatures generated by ``create_unique_constraint_sig``. This method returns a ``ComparisonResult``. r{r|)rxrzunnamedrrr)rarr metadata_tup reflected_tupmeta_sigconn_sigrBrBrCcompare_unique_constraint s z%DefaultImpl.compare_unique_constraintc Cshdd|D}t|D]&}|j|vrq t|}|r1td|jd|jdtjd||q dS)NcSsi|]}|j|qSrBrr)rrrBrBrCr'rz8DefaultImpl._skip_functional_indexes..z@autogenerate skipping metadata-specified expression-based index z ; dialect z under SQLAlchemy z6 can't reflect these indexes so they can't be compared) rrdr!is_expression_indexr rr<sqlalchemy_versiondiscard)rar;r8conn_indexes_by_nameidxiexrBrBrC_skip_functional_indexes&s$    z$DefaultImpl._skip_functional_indexesreflected_objectkindcCs |diS)Ndialect_options)r^)rarrrBrBrC adjust_reflected_dialect_options7s z,DefaultImpl.adjust_reflected_dialect_options)rQr%rRrSrTrUrVrWrXrYrZr;r[r\)rQr%r[rb)rr7r[r\) rir7rjrkrlrUrmrr[r)rvr4r[rU)ryr3rzrr[r\)r[rS) rrrrrrrrr[rrc)rrrrr[r\) NFNNNNFNNNNN) rr7rr7rprWrrrdrkrrrrkrrWrrrrkrrrrrrWrrWrmrr[r\)rr7rrrrr[r\)rr7rrrrkr[r\)rrr[r\)rr+r[r\)rr7rrrrr[r\)rzrrmrr[r\)rr-rmrr[r\)rzrr[r\)rrr[r\)T)rzrrrrrUr[r\)rrr[r)rrrrr[rU)r%rr&rr[rU)r0rr1rr[rU) r6r7r8r9r:r7r;r9r[r\)F)r>r(rDrUrmrr[r7)rGr'r[r)rJrKrLrKr[r\)r[r\)rWr0rXr2r[rY)rjr-rkr-r[rk)rprqrrrr[rs)rjr-rkr-r[r)rr.rr.r[r)rr;rr7r[r;)=rDrErF__doc__r<rVrrL__annotations__rMrPr> classmethodrerhrurxr{propertyrr immutabledictrrrrrrrrrrrrrrrrrr$r/r2r5r=rCrFrIrMrHrOrSrVr[rirorxrzrrrrrBrBrBrCrGQs         "   0   t        & (  !          +rG) metaclassc@s.eZdZUded<ded<ded<ded<dS) rr7rz List[str]rr zDict[str, str]r N)rDrErFrrBrBrBrCr=s  r metadata_io-Union[schema.Identity, schema.Sequence, None] inspector_io default_io'Union[schema.Identity, schema.Sequence]r]Set[str]cst|}t|}ttd fdd }|||t|t||tjrFt|d s2J|t|d it|d i|jt|d ifS) N meta_dictr insp_dict default_dictattrs Iterable[str]cs\t|D]$}||}||}||kr+||}||kr&|q|qdSrc)set differencer^ra)rrrrattr meta_value insp_value default_valuere ignored_attrr]rBrC check_dictsSs     z._compare_identity_options..check_dictsdialect_kwargs)rrrrrrrr)r!_get_identity_options_dictrunionidentity_has_dialect_kwargshasattrr`r)rrrr]meta_dinsp_drrBrrCr^Ds(     r^)rrrrrrr]r)J __future__rloggingrtypingrrrrrrr r r r r rrr sqlalchemyrrrrrrrrrrrrr r!r"r#sqlalchemy.enginer$r%sqlalchemy.engine.cursorr&sqlalchemy.engine.reflectionr'sqlalchemy.sqlr(r)sqlalchemy.sql.elementsr*sqlalchemy.sql.schemar+r,r-r.sqlalchemy.sql.selectabler/sqlalchemy.sql.type_apir0r1autogenerate.apir2operations.batchr3r4 getLoggerrDrr=r5r?rrGrr^rBrBrBrCs|                                                 q