o gu@sUddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!dd l!m"Z"er2dd!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*l/m1Z1dd+l2m3Z3dd,l2m4Z4dd-l2m5Z5dd.l2m6Z6dd/l7m8Z8dd0l9m:Z:dd1lm;Z;dd2lm?Z?dd4l>m@Z@eAeBZCGd5d6d6eDZEiZFd7eGd8<Gd9d:d:eEd;ZHGd).0kvr{rCrD s z+DefaultImpl.bulk_insert..)rrC) rlistrrrUrinsertinlinevaluesitems)rbr{rrrowrCrrD 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[^(),]+=)rR type_compilerprocessr>lowerrefindallmatchappendrsplitrkwargsargs) rbr definitiontokens term_tokens paren_termtokenrtermkeyvalrCrCrD_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|]}|qSrC)r)rtrCrCrD z=DefaultImpl._column_types_match...rC)rbatchrCrCrD sz3DefaultImpl._column_types_match.. F)token0rMjoinrissubset)rbrrsynonymsinspector_all_termsmetadata_all_termsrrCrCrD_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) lenrrr!rrNrsearchgroup)rbr'r(inspmetaregmimmrCrCrD_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&r1)rbr2r3rrrCrCrD compare_typeFs    zDefaultImpl.compare_typecCs||kSrdrC)rbr2r3rendered_metadata_defaultrendered_inspector_defaultrCrCrDcompare_server_defaultVsz"DefaultImpl.compare_server_default conn_uniquesSet[UniqueConstraint] conn_indexes Set[Index]metadata_unique_constraintsmetadata_indexescCdSrdrC)rbr8r:r<r=rCrCrDcorrect_for_autogen_constraints_sz+DefaultImpl.correct_for_autogen_constraintscCs(|jj|jurt|d||d<dSdS)Nexpr)r>_type_affinityr)rbexistingexisting_transfernew_typerCrCrDcast_for_batch_migratehs  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)rRr)r8rrR)rbr@rFrnrrCrCrDrender_ddl_sql_exprns zDefaultImpl.render_ddl_sql_expr inspectorr'rcCr~rd)autogen_column_reflect)rbrIrCrCrD_compat_autogen_column_reflect|sz*DefaultImpl._compat_autogen_column_reflectconn_fksSet[ForeignKeyConstraint] metadata_fkscCr>rdrC)rbrLrNrCrCrDcorrect_for_autogen_foreignkeyssz+DefaultImpl.correct_for_autogen_foreignkeyscCrx)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. NrC)rbrIr{ column_inforCrCrDrJr}z"DefaultImpl.autogen_column_reflectcCrx)zA hook called when :meth:`.EnvironmentContext.run_migrations` is called. Implementations can set up per-migration-run state here. NrCrrCrCrDstart_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`. BEGINNrirrrCrCrD emit_beginzDefaultImpl.emit_begincCrR)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`. COMMITNrTrrCrCrD emit_commitrVzDefaultImpl.emit_committype_objr1autogen_contextr3Union[str, Literal[False]]cCrxNFrC)rbrYrZrCrCrD render_typeszDefaultImpl.render_typecCsvt||tdhd\}}t|dd}t|dd}t|t|kr&|d||j||t|p9t|t|kfS)Nalways)skip)_compare_identity_optionsrIdentitygetattrrVadddifference_updaterQ)rbmetadata_identityinspector_identitydiffignored meta_alwaysinspector_alwaysrCrCrD_compare_identity_defaults    z%DefaultImpl._compare_identity_defaultmetadata_indexreflected_indexcCs0t|j}t|j}||krd|d|SdS)Nzunique=z to unique=)rVunique)rbrlrm conn_unique meta_uniquerCrCrD_compare_index_uniques  z!DefaultImpl._compare_index_unique constraint _autogen._Copts_constraint_sig[_autogen._C]cKtjd||fi|S)NTrfrom_constraintrbrrrtrCrCrD_create_metadata_constraint_sigz+DefaultImpl._create_metadata_constraint_sigcKrvr\rwryrCrCrD _create_reflected_constraint_sigr{z,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 )rqr rzr|r is_index_sighas_expressionslogwarningr column_namesr DifferentEqual)rbrlrmmsg unique_msgm_sigr_sigsigrCrCrDcompare_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~)rzr|unnamedrrr)rbrr metadata_tup reflected_tupmeta_sigconn_sigrCrCrDcompare_unique_constraint s z%DefaultImpl.compare_unique_constraintc Cshdd|D}t|D]&}|j|vrq t|}|r1td|jd|jdtjd||q dS)NcSsi|]}|j|qSrCrs)rrrCrCrDr(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) rrer!is_expression_indexr rr=sqlalchemy_versiondiscard)rbr=r:conn_indexes_by_nameidxiexrCrCrD_skip_functional_indexes's$    z$DefaultImpl._skip_functional_indexesreflected_objectkindcCs |diS)Ndialect_options)r_)rbrrrCrCrD adjust_reflected_dialect_options8s z,DefaultImpl.adjust_reflected_dialect_options)rRr%rSrTrUrVrWrXrYrZr[r<r\r])rRr%r\rc)rr8r\r]) rjr8rkrlrmrVrnrr\r)rwr5r\rV)rzr4r{rr\r])r\rT) rrrrrrrrr\rrd)rrrrr\r]) NFNNNNFNNNNN) rr8rr8rqrXrrrerlrrrrlrrXrrrrlrrrrrrXrrXrnrr\r])rr8rrrrr\r])rr8rrrrlr\r])rrr\r])rr,r\r])rr8rrrrr\r])r{rrnrr\r])rr.rnrr\r])r{rr\r])rrr\r])T)r{rrrrrVr\r])rrr\r)rrrrr\rV)r'rr(rr\rV)r2rr3rr\rV) r8r9r:r;r<r9r=r;r\r])F)r@r(rFrVrnrr\r8)rIr'r\r)rLrMrNrMr\r])r\r])rYr1rZr3r\r[)rlr.rmr.r\rl)rrrsrtrr\ru)rlr.rmr.r\r)rr/rr/r\r)rr<rr8r\r<)=rErFrG__doc__r=rWrrM__annotations__rNrQr? classmethodrfrirvryr|propertyrr immutabledictrrrrrrrrrrrrrrrrrr&r1r4r7r?rErHrKrOrJrQrUrXr]rkrqrzr|rrrrrCrCrCrDrHRs         "   0   t        & (  !          +rH) metaclassc@s.eZdZUded<ded<ded<ded<dS) rr8r z List[str]rrzDict[str, str]r N)rErFrGrrCrCrCrDr>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|qdSrd)set differencer_rc)rrrrattr meta_value insp_value default_valuerg ignored_attrr_rCrD check_dictsTs     z._compare_identity_options..check_dictsdialect_kwargs)rrrrrrrr)r!_get_identity_options_dictrunionidentity_has_dialect_kwargshasattrrbr)rrrr_meta_dinsp_drrCrrDr`Es(     r`)rrrrrrr_r)K __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*r+sqlalchemy.sql.schemar,r-r.r/sqlalchemy.sql.selectabler0sqlalchemy.sql.type_apir1r2autogenerate.apir3operations.batchr4r5 getLoggerrErr>r6r@rrHrr`rCrCrCrDs~                                                  q