o 8hv@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>rArArArBr5Asr5zDict[str, Type[DefaultImpl]]r?c @speZdZUdZdZdZdZddhfZded<d Z d ed <d Z d ed<dddZ e dd d!Z dd$d%Zdd-d.Zdd1d2Zdd6d7Zedd8d9Zd:d:efddDdEZ :dddHdIZd:dd:d:d:d:dd:d:d:d:d:dJ dd]d^Zd:d:d_ddddeZd:d:dfddhdiZddkdlZddndoZ :dddsdtZddudvZddwdxZdd{d|Z dd}d~Z!dddZ"dddZ#dddZ$ ddddZ%dddZ&dddZ'dddZ(dddZ)ddZ*dddZ+ddZ, ddddZ-dddZ.dddZ/ddZ0dddZ1dddZ2dddZ3dddńZ4ddDŽZ5ddd˄Z6dddфZ7dddӄZ8dddքZ9dddۄZ:dd݄Z;dddZjs  zDefaultImpl.__init__Type[DefaultImpl]cCs t|jSN)r?name)r@rPrArArBget_by_dialects zDefaultImpl.get_by_dialectrr7cCs,|jdusJ|j|d|jdS)Nz )rWwriteflush)r`rrArArB 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_pkcrc)rrrrappend_constraintr)r`rhrirkrlZvtrArArBversion_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. FrA)r`rtrArArBrequires_recreate_in_batchs z&DefaultImpl.requires_recreate_in_batch batch_implr3tablecCru)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. NrA)r`rwrxrArArBprep_table_for_batchz DefaultImpl.prep_table_for_batchcC|jSrb)rQr`rArArBbindszDefaultImpl.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_kwargsrP z z2Can't send params and multiparams at the same timerA) isinstancer7rrS TypeErrorr\r DDLElementdictrr(compilerPrgreplacestripcommand_terminatorrQrexecute)r`r~rrr compile_kwcompiledconnrArArB_execs<      zDefaultImpl._execsqlOptional[dict[str, Any]]cCs|||dSrb)r)r`rrrArArBrszDefaultImpl.execute) roserver_defaultrctype_r autoincrementcommentexisting_comment existing_typeexisting_server_defaultexisting_nullableexisting_autoincrement table_name column_nameror/Optional[Union[_ServerDefault, Literal[False]]]rcrOptional[TypeEngine]rrr$Optional[Union[str, Literal[False]]]rrrOptional[_ServerDefault]rrc  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)r`rrrorrcrrrrrrrrrrlcls_rArArB alter_columns     zDefaultImpl.alter_columnr if_not_existscolumn Column[Any]!Optional[Union[str, quoted_name]]rcC|tj||||ddS)Nr)rr AddColumn)r`rrrrrArArB add_columnrszDefaultImpl.add_columnr if_existsrcKr)Nr)rr DropColumn)r`rrrrrlrArArB drop_columns  zDefaultImpl.drop_columnconstcCs,|jdus ||r|t|dSdSrb) _create_rulerr AddConstraint)r`rrArArBadd_constraintszDefaultImpl.add_constraintr+cK|tj|fi|dSrb)rrDropConstraint)r`rrlrArArBdrop_constraintzDefaultImpl.drop_constraintold_table_namenew_table_nameUnion[str, quoted_name]cCs|tj|||ddS)Nr)rr RenameTable)r`rrrrArArB 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_createrQrr CreateTable after_createindexes CreateIndexrPsupports_commentsinline_commentsrcreate_table_commentcolumnscreate_column_comment)r`rxrlindexZ with_commentrrrArArB create_tables(      zDefaultImpl.create_tablecKsH|jj||jd|d|tj|fi||jj||jd|ddSr)r before_droprQrr DropTable after_drop)r`rxrlrArArB drop_tables   zDefaultImpl.drop_tablerr-cKrrb)rrrr`rrlrArArB create_indexrzDefaultImpl.create_indexcC|t|dSrb)rrSetTableCommentr`rxrArArBrz DefaultImpl.create_table_commentcCrrb)rrDropTableCommentrrArArBdrop_table_commentrzDefaultImpl.drop_table_commentcCrrb)rrSetColumnComment)r`rrArArBrrz!DefaultImpl.create_column_commentcKrrb)rr DropIndexrrArArB 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=).0kvrxrArB s z+DefaultImpl.bulk_insert..)rrA) rlistrrrSrinsertinlinevaluesitems)r`rxrrrowrArrB 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[^(),]+=)rP type_compilerprocessr=lowerrefindallmatchappendrsplitrkwargsargs) r`rZ definitiontokensZ term_tokensZ paren_termtokenrZtermkeyvalrArArB_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|]}|qSrA)r)rtrArArB &z=DefaultImpl._column_types_match...rA)rbatchrArArB &sz3DefaultImpl._column_types_match.. F)token0rKjoinr issubset)r`rrsynonymsZinspector_all_termsZmetadata_all_termsrrArArB_column_types_match s&   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) lenr r rrrLrsearchgroup)r`rrinspmetaregmimmrArArB_column_args_match6s     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)rrr()r`r)r*rrrArArB compare_typeWs    zDefaultImpl.compare_typecCs||kSrbrA)r`r)r*rendered_metadata_defaultrendered_inspector_defaultrArArBcompare_server_defaultgsz"DefaultImpl.compare_server_default conn_uniquesSet[UniqueConstraint] conn_indexes Set[Index]metadata_unique_constraintsmetadata_indexescCdSrbrA)r`r/r1r3r4rArArBcorrect_for_autogen_constraintspsz+DefaultImpl.correct_for_autogen_constraintscCs(|jj|jurt|d||d<dSdS)Nexpr)r=_type_affinityr)r`existingZexisting_transfernew_typerArArBcast_for_batch_migrateys  z"DefaultImpl.cast_for_batch_migrater7r(is_server_defaultcKsddd}t|j|j|dS)zdRender a SQL expression that is typically a server default, index expression, etc. TF)r\ include_table)rPr)r7rrP)r`r7r<rlrrArArBrender_ddl_sql_exprs zDefaultImpl.render_ddl_sql_expr inspectorr'rcCr{rb)autogen_column_reflect)r`r?rArArB_compat_autogen_column_reflectsz*DefaultImpl._compat_autogen_column_reflectconn_fksSet[ForeignKeyConstraint] metadata_fkscCr5rbrA)r`rBrDrArArBcorrect_for_autogen_foreignkeyssz+DefaultImpl.correct_for_autogen_foreignkeyscCru)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. NrA)r`r?rx column_inforArArBr@rzz"DefaultImpl.autogen_column_reflectcCru)zA hook called when :meth:`.EnvironmentContext.run_migrations` is called. Implementations can set up per-migration-run state here. NrAr|rArArBstart_migrationsrzzDefaultImpl.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`. BEGINNrgrr|rArArB emit_beginzDefaultImpl.emit_begincCrH)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`. COMMITNrJr|rArArB emit_commitrLzDefaultImpl.emit_committype_objr0autogen_contextr2Union[str, Literal[False]]cCruNFrA)r`rOrPrArArB render_typeszDefaultImpl.render_typecCsvt||tdhd\}}t|dd}t|dd}t|t|kr&|d||j||t|p9t|t|kfS)Nalways)skip)_compare_identity_optionsrIdentitygetattrrTadddifference_updaterO)r`metadata_identityinspector_identitydiffignoredZ meta_alwaysZinspector_alwaysrArArB_compare_identity_defaults    z%DefaultImpl._compare_identity_defaultmetadata_indexreflected_indexcCs0t|j}t|j}||krd|d|SdS)Nzunique=z to unique=)rTunique)r`r`raZ conn_uniqueZ meta_uniquerArArB_compare_index_uniques  z!DefaultImpl._compare_index_unique constraint _autogen._Copts_constraint_sig[_autogen._C]cKtjd||fi|S)NTrZfrom_constraintr`rdrfrArArB_create_metadata_constraint_sigz+DefaultImpl._create_metadata_constraint_sigcKrhrRrirjrArArB _create_reflected_constraint_sigrlz,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 )rcrrkrmrZ is_index_sigZhas_expressionslogwarningr column_namesr DifferentEqual)r`r`ramsgZ unique_msgZm_sigZr_sigsigrArArBcompare_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``. rnro)rkrmZunnamedrrsrt)r`rxryZ metadata_tupZ reflected_tupZmeta_sigZconn_sigrArArBcompare_unique_constraints z%DefaultImpl.compare_unique_constraintc Cshdd|D}t|D]&}|j|vrq t|}|r1td|jd|jdtjd||q dS)NcSsi|]}|j|qSrArq)rrrArArBr9rz8DefaultImpl._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) rrcr!is_expression_indexr rr<sqlalchemy_versiondiscard)r`r4r1Zconn_indexes_by_nameidxZiexrArArB_skip_functional_indexes8s$    z$DefaultImpl._skip_functional_indexesreflected_objectkindcCs |diS)Ndialect_options)r])r`rrrArArB adjust_reflected_dialect_optionsIs z,DefaultImpl.adjust_reflected_dialect_options)rPr%rQrRrSrTrUrVrWrXrYr;rZr[)rPr%rZra)rr7rZr[) rhr7rirjrkrTrlrrZr)rtr4rZrT)rwr3rxrrZr[)rZrR) r~rrrrrrrrZrrb)rrrrrZr[) rr7rr7rorVrrrcrjrrrrjrrVrrrrjrrrrrrVrrVrlrrZr[) rr7rrrrrrVrZr[) rr7rrrrjrrVrZr[)rrrZr[)rr+rlrrZr[)rr7rrrrrZr[)rxrrlrrZr[)rr-rlrrZr[)rxrrZr[)rrrZr[)T)rxrrrrrTrZr[)rrrZr)rrrrrZrT)rrrrrZrT)r)rr*rrZrT) r/r0r1r2r3r0r4r2rZr[)F)r7r(r<rTrlrrZr7)r?r'rZr)rBrCrDrCrZr[)rZr[)rOr0rPr2rZrQ)r`r-rar-rZrj)rdrerfrrZrg)r`r-rar-rZr)rxr.ryr.rZr)rr;rr7rZr;)=rCrDrE__doc__r<rUrrK__annotations__rLrOr> classmethodrdrgrsrvrypropertyr}r immutabledictrrrrrrrrrrrrrrrrrrr(r+r.r6r;r>rArEr@rGrKrNrSr_rcrkrmrwrzrrrArArArBrFQs         "   0  x        & (  !          +rF) metaclassc@s.eZdZUded<ded<ded<ded<dS) rr7rz List[str]r r zDict[str, str]r N)rCrDrErrArArArBrOs  r metadata_io-Union[schema.Identity, schema.Sequence, None] inspector_io default_io'Union[schema.Identity, schema.Sequence]rUSet[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|qdSrb)set differencer]rY)rrrrattrZ meta_valueZ insp_value default_valuer]Z ignored_attrrUrArB check_dictses     z._compare_identity_options..check_dictsdialect_kwargs)rrrrrrrr)r!_get_identity_options_dictrunionidentity_has_dialect_kwargshasattrrXr)rrrrUZmeta_dZinsp_drrArrBrVVs(     rV)rrrrrrrUr)J __future__rloggingrtypingrrrrrrr r r r r rrr sqlalchemyrrrrrrrrrrrrr r!r"r#sqlalchemy.enginer$r%sqlalchemy.engine.cursorr&Zsqlalchemy.engine.reflectionr'sqlalchemy.sqlr(r)sqlalchemy.sql.elementsr*sqlalchemy.sql.schemar+r,r-r.sqlalchemy.sql.selectabler/sqlalchemy.sql.type_apir0r1autogenerate.apir2Zoperations.batchr3r4 getLoggerrCrpr=r5r?rrFrrVrArArArBs~