ZeIivUddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!dd l!m"Z"errdd!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?ej@eAZBGd4d5eCZDiZEd6eFd7<Gd8d9eD:ZGGd;d(>(>     rFbatch_opr5cdS)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. FrJ)rgrs rErequires_recreate_in_batchz&DefaultImpl.requires_recreate_in_batchs urF batch_implr4tablecdS)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. NrJ)rgrrs rEprep_table_for_batchz DefaultImpl.prep_table_for_batchrFc|jSrj)rWrgs rEbindzDefaultImpl.binds rFN constructUnion[Executable, str]execution_optionsOptional[Mapping[str, Any]] multiparams%Optional[Sequence[Mapping[str, Any]]]paramsMapping[str, Any]Optional[CursorResult]ct|trt|}|jr||rt d|jr-t|t jstddi}ni}trt|tsJ|j d d|j i|}| t|dd|jzdS|j}|J|r |jd i|}|r|t d|r|||S|||S) Nz&SQL parameters not allowed with as_sqlrcT)compile_kwargsrV z z2Can't send params and multiparams at the same timerJ) isinstancer9rrY TypeErrorrcr DDLElementdictrr)compilerVrpreplacestripcommand_terminatorrWrexecute)rgrrrr compile_kwcompiledconns rE_execzDefaultImpl._execs i % % (YI ;" 7&&& HIII! *6,++ "/41HIII   <!)];;;;;(y(LLLLLH   H %%dF3399;;)*   4?D###  C-t-BB0ABB +1H 7||I{;;;||Iv666rFsqlOptional[dict[str, Any]]c2|||dSrj)r)rgrrs rErzDefaultImpl.executes 3)*****rF) ryserver_defaultrktype_r autoincrementcommentexisting_comment existing_typeexisting_server_defaultexisting_nullableexisting_autoincrement table_name column_nameryr/Optional[Union[_ServerDefault, Literal[False]]]rkrOptional[TypeEngine]rrr$Optional[Union[str, Literal[False]]]rrrOptional[_ServerDefault]rrc ||tjdd|/|tj||||| | | | |dur{i}t j|| r tj}n3t j|| rtj }||d<n tj }|||||f|| | | | d||/|tj ||||| | | | | dur/|tj ||| || | | | |0|tj ||||| | | dSdS)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)rgrrryrrkrrrrrrrrrrucls_s rE alter_columnzDefaultImpl.alter_columns5*  $(>(J I,       JJ#!"/,C&7%5        & &B6 7 *18 7 *1!6 ) JJ" ""/,C&7%5          JJ!"/,C&7%5       %   JJ"!"/,C&7%5         JJ!"/,C&7       rFr if_not_existscolumn Column[Any]!Optional[Union[str, quoted_name]]rc\|tj||||dS)Nr)rr AddColumn)rgrrrrs rE add_columnzDefaultImpl.add_columnrsD N+         rFr if_existsrc \|tj||||dS)Nr)rr DropColumn)rgrrrrrus rE drop_columnzDefaultImpl.drop_columnsA OF6Y        rFconstc|j||r)|tj|dSdSrj) _create_rulerr AddConstraint)rgrs rEadd_constraintzDefaultImpl.add_constraintsI   %););D)A)A % JJv+E22 3 3 3 3 3 & %rFr,c P|tj|fi|dSrj)rrDropConstraint)rgrrus rEdrop_constraintzDefaultImpl.drop_constraints- 6(55"5566666rFold_table_namenew_table_nameUnion[str, quoted_name]cZ|tj|||dS)Nr)rr RenameTable)rgrrrs rE rename_tablezDefaultImpl.rename_tables9  ^^F K K K     rFc |j||jd||t j|fi||j||jd||jD])}|t j|*|j j o |j j }|j }|r|r| ||jD]"}|j }|r|r||#dSNF) checkfirst _ddl_runner)dispatch before_createrWrr CreateTable after_createindexes CreateIndexrVsupports_commentsinline_commentsrcreate_table_commentcolumnscreate_column_comment)rgrruindex with_commentrrs rE create_tablezDefaultImpl.create_tables> $$ 4?u$ %    6%e22r22333 ## 4?u$ $   ] 2 2E JJv)%00 1 1 1 1 L * O4<3O/O -  -| -  % %e , , ,m 3 3FnG 3< 3**6222 3 3rFc |j||jd||t j|fi||j||jd|dSr)r before_droprWrr DropTable after_drop)rgrrus rE drop_tablezDefaultImpl.drop_tables "" 4?u$ #    6#E00R00111 !! 4?u$ "     rFrr.c P|tj|fi|dSrj)rrrrgrrus rE create_indexzDefaultImpl.create_indexs- 6%e22r2233333rFcT|tj|dSrj)rrSetTableCommentrgrs rErz DefaultImpl.create_table_comments% 6)%0011111rFcT|tj|dSrj)rrDropTableCommentrs rEdrop_table_commentzDefaultImpl.drop_table_comments% 6*51122222rFcT|tj|dSrj)rrSetColumnComment)rgrs rErz!DefaultImpl.create_column_comments% 6*62233333rFc P|tj|fi|dSrj)rr DropIndexrs rE drop_indexzDefaultImpl.drop_indexs- 6#E00R0011111rFTUnion[TableClause, Table]rows List[dict] multiinsertc t|tstd|r*t|dtstd|jri|D]d}|jdifd| DedS|r|r=||dS|D]H}|jdi|GdSdS)Nz List expectedrzList of dictionaries expectedc i|]I\}}|t|tjs'tj||j|jn|JS))r)rr"_literal_bindparamcr@).0kvrs rE z+DefaultImpl.bulk_insert..sx   !%1(2$%{'E("("!' >$%q !"!"!"!" &'   rF)rrJ) rlistrrrYrinsertinlinevaluesitems)rgrrrrows ` rE bulk_insertzDefaultImpl.bulk_inserts $%% =O,, ,  =*T!Wd33 =;<< < ; J   ELLNNVXX    ),      ( JJJJu||~~4466DJIIIII#JJ #A5<<>>#8#8#:#:#A#H#HC#H#HIIII  J JJJrFrParamscj|jj|j}t jd|}g}d}|D]/}t jd|r|}||0t|d|ddgi}|rt jd|D]y}d|vrG| d\} } | |j | <M|j | z|S)Nz[\w\-_]+|\(.+?\)z^\(.*\)$rrz[^(),]+=)rV type_compilerprocessr@lowerrefindallmatchappendrsplitrkwargsargs) rgr definitiontokens term_tokens paren_termtokenrtermkeyvals rE_tokenize_column_typez!DefaultImpl._tokenize_column_types*\/77 DDJJLL J':JGG!#   * *Ex U++ *" ""5)))) A ABBR@@  5 9j99 5 5$;;#zz#HC14FM#))++..K&&tzz||4444 rFinspector_paramsmetadata_paramscX|j|jkrdSd|jD}d|jg|jz}d|jg|jz}|D]=}||h|s!|j|jh|rdS>dS)NTc&g|]}d|DS)c6h|]}|SrJ)r)r ts rE z=DefaultImpl._column_types_match...&s ...1QWWYY...rFrJ)r batchs rE z3DefaultImpl._column_types_match..&s'OOO5.....OOOrF F)token0rQjoinr&issubset)rgr.r/synonymsinspector_all_termsmetadata_all_termsr5s rE_column_types_matchzDefaultImpl._column_types_match s  "o&< < <4OOD%?%? ? ? $4$9995xx(/006688xx *++1133(  C3%%B3%%B b RXXa[[BHHQKK77uutrFinspector_columnmetadata_columnc||}||}|||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)r-r>rJ)rgrKrLr.r/s rE compare_typezDefaultImpl.compare_typeWsm 556FGG44_EE''(8/JJ 4&&'7II 4urFc||kSrjrJ)rgrKrLrendered_metadata_defaultrendered_inspector_defaults rEcompare_server_defaultz"DefaultImpl.compare_server_defaultgs*-FFFrF conn_uniquesSet[UniqueConstraint] conn_indexes Set[Index]metadata_unique_constraintsmetadata_indexescdSrjrJ)rgrSrUrWrXs rEcorrect_for_autogen_constraintsz+DefaultImpl.correct_for_autogen_constraintsps  rFcb|jj|jurt|d||d<dSdS)Nexpr)r@_type_affinityr)rgexistingexisting_transfernew_types rEcast_for_batch_migratez"DefaultImpl.cast_for_batch_migrateysC = 'x/F F F(,!&)8)) f % % % G FrFr\r)is_server_defaultc ^ddd}t||j|S)zdRender a SQL expression that is typically a server default, index expression, etc. TF)rc include_table)rVr)r9rrV)rgr\rbrurs rErender_ddl_sql_exprzDefaultImpl.render_ddl_sql_exprs8(,eDD  LLjL I I   rF inspectorr(rc|jSrj)autogen_column_reflect)rgrfs rE_compat_autogen_column_reflectz*DefaultImpl._compat_autogen_column_reflects **rFconn_fksSet[ForeignKeyConstraint] metadata_fkscdSrjrJ)rgrjrls rEcorrect_for_autogen_foreignkeysz+DefaultImpl.correct_for_autogen_foreignkeyss rFcdS)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. NrJ)rgrfr column_infos rErhz"DefaultImpl.autogen_column_reflectrrFcdS)zA hook called when :meth:`.EnvironmentContext.run_migrations` is called. Implementations can set up per-migration-run state here. NrJrs rEstart_migrationszDefaultImpl.start_migrationsrrFc@|d|jzdS)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`. BEGINNrprrs rE emit_beginzDefaultImpl.emit_begins& 7T%<<=====rFc@|d|jzdS)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`. COMMITNrurs rE emit_commitzDefaultImpl.emit_commits& 8d&==>>>>>rFtype_objr1autogen_contextr3Union[str, Literal[False]]cdSNFrJ)rgrzr{s rE render_typezDefaultImpl.render_types urFct||tjdh\}}t|dd}t|dd}t |t |kr|d||j||t |pt |t |kfS)Nalways)skip)_compare_identity_optionsrIdentitygetattrrZadddifference_updaterU)rgmetadata_identityinspector_identitydiffignored meta_alwaysinspector_alwayss rE_compare_identity_defaultz%DefaultImpl._compare_identity_defaults2   O      g/4@@ "#5xFF   %5 6 6 6 6 HHX    t9:::   JJ M$011T:L5M5MM   rFmetadata_indexreflected_indexcrt|j}t|j}||krd|d|SdS)Nzunique=z to unique=)rZunique)rgrr conn_unique meta_uniques rE_compare_index_uniquez!DefaultImpl._compare_index_uniquesJ?122 >011 + % %B[BB[BB B4rF constraint _autogen._Copts_constraint_sig[_autogen._C]c *tjd||fi|S)NTrfrom_constraintrgrrs rE_create_metadata_constraint_sigz+DefaultImpl._create_metadata_constraint_sigs!.tT:NNNNNrFc *tjd||fi|Sr~rrs rE _create_reflected_constraint_sigz,DefaultImpl._create_reflected_constraint_sigs!.udJOO$OOOrFrc$g}|||}|r||||}||}t j|sJt j|sJ||fD])}|jr td|j *|j |j kr%|d|j d|j |rtj |Stj 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 )rr!rrr is_index_sighas_expressionslogwarningr column_namesr DifferentEqual)rgrrmsg unique_msgm_sigr_sigsigs rEcompare_indexeszDefaultImpl.compare_indexessB// O    # JJz " " "44^DD55oFF$U+++++$U+++++%<  C"  :I   !3 3 3 JJJe0JJe6HJJ     ,#-c22 2#)++ +rFmetadata_constraintr/reflected_constraintc||}||}|j}|j}||krtjd|d|StjS)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``. rr)rrunnamedrrr)rgrr metadata_tup reflected_tupmeta_sigconn_sigs rEcompare_unique_constraintz%DefaultImpl.compare_unique_constraints;;   ==    ' ( x  #-6h66H66 $)++ +rFc d|D}t|D]i}|j|vr tj|}|rGt jd|jd|jdtjd||jdS)Nci|] }|j| SrJr{)r r s rErz8DefaultImpl._skip_functional_indexes..9s@@@a@@@rFz@autogenerate skipping metadata-specified expression-based index z ; dialect z under SQLAlchemy z6 can't reflect these indexes so they can't be compared) rrkr"is_expression_indexr!rr?sqlalchemy_versiondiscard)rgrXrUconn_indexes_by_nameidxiexs rE_skip_functional_indexesz$DefaultImpl._skip_functional_indexes8s@@<@@@()) . .Cx///1#66C . FxFF-1-=FF#."@FFF!((--- . .rFreflected_objectkindc.|diS)Ndialect_options)rd)rgrrs rE adjust_reflected_dialect_optionsz,DefaultImpl.adjust_reflected_dialect_optionsIs ##$5r:::rF)rVr&rWrXrYrZr[r\r]r^r_r=r`ra)rVr&r`rh)rr9r`ra) rqr9rrrsrtrZrurr`r)rr5r`rZ)rr4rrr`ra)r`rX) rrrrrrrrr`rrj)rrrrr`ra) rr9rr9ryr\rrrkrsrrrrsrr\rrrrsrrrrrr\rr\rurr`ra) rr9rrrrrr\r`ra) rr9rrrrsrr\r`ra)rrr`ra)rr,rurr`ra)rr9rrrrr`ra)rrrurr`ra)rr.rurr`ra)rrr`ra)rrr`ra)T)rrrrrrZr`ra)rrr`r)r.rr/rr`rZ)r?rr@rr`rZ)rKrrLrr`rZ) rSrTrUrVrWrTrXrVr`ra)F)r\r)rbrZrurr`r9)rfr(r`r)rjrkrlrkr`ra)r`ra)rzr1r{r3r`r|)rr.rr.r`rs)rrrrr`r)rr.rr.r`r)rr/rr/r`r)rr=rr9r`r=)=rGrHrI__doc__r?r[rrQ__annotations__rRrUrA classmethodrlrpr~rrpropertyrr! immutabledictrrrrrrrrrrrrrrrrr-r>rJrNrRrZrarerirnrhrrrvryrrrrrrrrrrJrFrErLrLQs  K,5y+A*CMCCCC&((((( .BAAAA4$$$[$####    D         X :>=A$6D$6$8$8 +7+7+7+7+7`7;+++++$( "&* $(,8=*..2<@,015%qqqqqqp59(,       ,!%$(       4444777759      3333.    44442222333344442222! $J$J$J$J$JL&&&&P,B GGG    >C      ++++           >>>>????    8OOOO PPPP ),),),),V,,,,8...";;;;;;rFrL) metaclassc8eZdZUded<ded<ded<ded<dS) rr9r8z List[str]r&r$zDict[str, str]r#N)rGrHrIrrJrFrErrOs<KKKOOOrFr metadata_io-Union[schema.Identity, schema.Sequence, None] inspector_io default_io'Union[schema.Identity, schema.Sequence]rSet[str]c tj|}tj|}ttd fd }|||tj|t||tjrRt |dsJ|t |dit |di|jt |difS) N meta_dictr insp_dict default_dictattrs Iterable[str]c>t| D]x}||}||}||krF||}||kr |c|ydSrj)set differencerdr) rrrrattr meta_value insp_value default_valuer ignored_attrrs rE check_dictsz._compare_identity_options..check_dictses JJ))$// # #D"t,,J"t,,JZ'' , 0 0 6 6 .. $$T****HHTNNN # #rFdialect_kwargs)rrrrrrrr)r"_get_identity_options_dictrunionidentity_has_dialect_kwargshasattrrr) rrrrmeta_dinsp_drrrs ` @@rErrVs 3K @ @F  3L A AF 55D55L######## K.z:: F &!!  .  z#344444  K!12 6 6 L"2B 7 7  % L"2B 7 7      rF)rrrrrrrr)J __future__rloggingrtypingrrrrrr r r r r rrrr 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.selectabler0sqlalchemy.sql.type_apir1r2autogenerate.apir3operations.batchr4r5 getLoggerrGrr@r7rBrrLrrrJrFrErsT#""""""  ++++++888888::::::7,,,,,,))))))555555666666,,,,,,))))))333333000000::::::++++++666666555555222222$$$$$$111111111111666666g!!     t   (*)))){ ;{ ;{ ;{ ;{ ;H{ ;{ ;{ ;{ ;|Z000000rF