o @ju@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) rlistrrrUrr!_insert_inlinevaluesitems)rbr{rrrowrCrrD bulk_inserts2     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) lenrr r 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%r0)rbr1r2rrrCrCrD compare_typeHs    zDefaultImpl.compare_typecCs||kSrdrC)rbr1r2rendered_metadata_defaultrendered_inspector_defaultrCrCrDcompare_server_defaultXsz"DefaultImpl.compare_server_default conn_uniquesSet[UniqueConstraint] conn_indexes Set[Index]metadata_unique_constraintsmetadata_indexescCdSrdrC)rbr7r9r;r<rCrCrDcorrect_for_autogen_constraintsasz+DefaultImpl.correct_for_autogen_constraintscCs(|jj|jurt|d||d<dSdS)Nexpr)r>_type_affinityr)rbexistingexisting_transfernew_typerCrCrDcast_for_batch_migratejs  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?rErnrrCrCrDrender_ddl_sql_exprps zDefaultImpl.render_ddl_sql_expr inspectorr'rcCr~rd)autogen_column_reflect)rbrHrCrCrD_compat_autogen_column_reflect~sz*DefaultImpl._compat_autogen_column_reflectconn_fksSet[ForeignKeyConstraint] metadata_fkscCr=rdrC)rbrKrMrCrCrDcorrect_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)rbrHr{ column_inforCrCrDrIr}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_begincCrQ)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`. COMMITNrSrrCrCrD emit_commitrUzDefaultImpl.emit_committype_objr1autogen_contextr3Union[str, Literal[False]]cCrxNFrC)rbrXrYrCrCrD render_typeszDefaultImpl.render_typecCsvt||tdhd\}}t|dd}t|dd}t|t|kr&|d||j||t|p9t|t|kfS)Nalways)skip)_compare_identity_optionsr!IdentitygetattrrVadddifference_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)rbrkrl conn_unique meta_uniquerCrCrD_compare_index_uniques  z!DefaultImpl._compare_index_unique constraint _autogen._Copts_constraint_sig[_autogen._C]cKtjd||fi|S)NTrfrom_constraintrbrqrsrCrCrD_create_metadata_constraint_sigz+DefaultImpl._create_metadata_constraint_sigcKrur[rvrxrCrCrD _create_reflected_constraint_sigrzz,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 )rpr ryr{r is_index_sighas_expressionslogwarningr column_namesr DifferentEqual)rbrkrlmsg 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})ryr{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<r9conn_indexes_by_nameidxiexrCrCrD_skip_functional_indexes)s$    z$DefaultImpl._skip_functional_indexesreflected_objectkindcCs |diS)Ndialect_options)r_)rbrrrCrCrD adjust_reflected_dialect_options:s 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)r1rr2rr\rV) r7r8r9r:r;r8r<r:r\r])F)r?r(rErVrnrr\r8)rHr'r\r)rKrLrMrLr\r])r\r])rXr1rYr3r\rZ)rkr.rlr.r\rl)rqrrrsrr\rt)rkr.rlr.r\r)rr/rr/r\r)rr<rr8r\r<)=rErFrG__doc__r=rWrrM__annotations__rNrQr? classmethodrfrirvryr|propertyrr immutabledictrrrrrrrrrrrrrrrrrr%r0r3r6r>rDrGrJrNrIrPrTrWr\rjrpryr{rrrrrCrCrCrDrHRs         "   0   t        ( (  !          +rH) metaclassc@s.eZdZUded<ded<ded<ded<dS) rr8rz List[str]rr zDict[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||tjr?|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_rb)rrrrattr meta_value insp_value default_valuerf ignored_attrr^rCrD check_dictsVs     z._compare_identity_options..check_dictsdialect_kwargs)rrrrrrrr)r!_get_identity_options_dictrunionidentity_has_dialect_kwargsrar)rrrr^meta_dinsp_drrCrrDr_Gs&     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~                                                  s