hjpUddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZerddlmZddlmZddl m!Z!ddl m"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-m.Z.dd'l-m/Z/dd(l-m0Z0dd)l-m1Z1dd*l-m2Z2dd+l-m3Z3dd,l4m5Z5dd-l6m7Z7dd.lm8Z8dd/l9m:Z:dd0l;mDict[str, Any])__name__ __module__ __qualname__r9rAr@r5r5?s#  (  rAr5zDict[str, Type[DefaultImpl]]r:ceZdZUdZdZdZdZddhfZded<d Z d ed <d Z d ed< d=dZ e d>dZ d?dZ d@dZ dAdZedBdZdd ej(f dCdZ dD dEdZ dF dGdZ dD dHdZ dD dIdZdJdZdKdZ dD dLdZdMdZdMdZdNd ZdMd!Z dMd"Z!dOd#Z"dNd$Z# dP dQd%Z$dRd&Z% dSd'Z& dTd(Z' dUd)Z(d*Z) dVd+Z*d,Z+ dW dXd-Z,dYd.Z- dZd/Z.d0Z/d[d1Z0d[d2Z1d[d3Z2 d\d4Z3d5Z4 d]d6Z5 d^d7Z6 d^d8Z7 d_d9Z8 d`d:Z9d;Z: dad<Z;y)b DefaultImplaProvide the entrypoint for major migration operations, including database-specific behavioral variances. While individual SQL/DDL constructs already provide for database-specific implementations, variances here allow for entirely different sequences of operations to take place for a particular migration, such as SQL Server's special 'IDENTITY INSERT' step for bulk inserts. defaultF;NUMERICDECIMALzTuple[Set[str], ...] type_synonymsrGz Sequence[str]type_arg_extract)orderon_nullzTuple[str, ...]identity_attrs_ignorec||_||_||_|jdd|_||_i|_||_|||_|jr"|jstjdyy)N literal_bindsFz3Can't use literal_binds setting without as_sql mode) dialect connectionas_sqlgetrT output_buffermemo context_optstransactional_ddlr CommandError)selfrUrVrWr\rYr[s r@r9zDefaultImpl.__init__is $ )--ouE* (  (%6D "   ;;''I rAc(t|jSN)r:name)r;rUs r@get_by_dialectzDefaultImpl.get_by_dialectsgll##rAc|jJ|jj|dz|jjy)Nz )rYwriteflush)r^rs r@ static_outputzDefaultImpl.static_outputs>!!---   /   "rAcy)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. FrG)r^batch_ops r@requires_recreate_in_batchz&DefaultImpl.requires_recreate_in_batchsrAcy)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. NrG)r^ batch_impltables r@prep_table_for_batchz DefaultImpl.prep_table_for_batchrAc|jSr`)rVr^s r@bindzDefaultImpl.binds rANctt|tr t|}|jr|s|r t d|j r)t|t jstddi}ni}trt|tsJ|jdd|ji|}|jt|jddj|j zy|j"}|J|r|j$di|}|rt|t&sJ||fz }|j)||S) Nz+Execution arguments not allowed with as_sqlrTT)compile_kwargsrU z rG) isinstancerBrrW ExceptionrTr DDLElementdictrr$compilerUrfreplacestripcommand_terminatorrVexecution_optionstupleexecute)r^ constructr} multiparamsparams compile_kwcompiledconns r@_execzDefaultImpl._execs/ i %YI ;;f MNN!!*6,,+"/41HI  !)];;;(y((LLLH   H %%dF399;))* ??D# ## -t--B0AB!+u555y( << ;7 7rAc(|j||yr`)r)r^sqlr}s r@rzDefaultImpl.executes 3)*rAc ||tjdd|,|jtj||||| | | | |duri}t j || rtj}n(J II,    JJ##!"/,C&7%5    &B66 71188 711!6 )) JJ" ""/,C&7%5      JJ!"/,C&7%5   %  JJ""!"/,C&7%5     JJ!"/,C&7  rAcR|jtj|||yNr)rr AddColumn)r^rcolumnrs r@ add_columnzDefaultImpl.add_columnHs 4>>*fVDErAc R|jtj|||yr)rr DropColumn)r^rrrrs r@ drop_columnzDefaultImpl.drop_columnPs 4??:vfEFrAc|j|j|r%|jtj|yyr`) _create_rulerr AddConstraintr^consts r@add_constraintzDefaultImpl.add_constraintYs8    %););D)A JJv++E2 3*BrAcL|jtj|yr`)rrDropConstraintrs r@drop_constraintzDefaultImpl.drop_constraint]s 6((/0rAcR|jtj|||yr)rr RenameTable)r^old_table_namenew_table_namers r@ rename_tablezDefaultImpl.rename_table`s"   ^^F K rAcb|jj||jd||jt j ||jj ||jd||jD]&}|jt j|(|jjxr|jj }|j}|r|r|j||jD]%}|j}|s|s|j|'yNF) checkfirst _ddl_runner)dispatch before_createrVrr CreateTable after_createindexes CreateIndexrUsupports_commentsinline_commentsrcreate_table_commentcolumnscreate_column_comment)r^rlindex with_commentrrs r@ create_tablezDefaultImpl.create_tablejs $$ 4??u$ %  6%%e,- ## 4??u$ $ ]]E JJv))%0 1# LL * * O4<<3O3O/O -- |  % %e ,mmFnnG<**62$rAc|jj||jd||jt j ||jj ||jd|yr)r before_droprVrr DropTable after_dropr^rls r@ drop_tablezDefaultImpl.drop_tablesc "" 4??u$ #  6##E*+ !! 4??u$ " rAc N|jtj|fi|yr`)rrrr^rrs r@ create_indexzDefaultImpl.create_indexs 6%%e2r23rAcL|jtj|yr`)rrSetTableCommentrs r@rz DefaultImpl.create_table_comments 6))%01rAcL|jtj|yr`)rrDropTableCommentrs r@drop_table_commentzDefaultImpl.drop_table_comments 6**512rAcL|jtj|yr`)rrSetColumnComment)r^rs r@rz!DefaultImpl.create_column_comments 6**623rAc N|jtj|fi|yr`)rr DropIndexrs r@ drop_indexzDefaultImpl.drop_indexs 6##E0R01rAct|ts td|rt|dts td|jr|D]}|j t j|jdi|jDcic]Q\}}|t|t js/t j|||j|jn|Sc}}y|re|r'|j t j||y|D]6}|j t j|jdi|8yycc}}w)Nz List expectedrzList of dictionaries expected)r)rrG) rulist TypeErrorrxrWrr_insert_inlinevaluesitems_literal_bindparamcr8)r^rlrows multiinsertrowkvs r@ bulk_insertzDefaultImpl.bulk_insertsH $%O, , *T!Wd3;< < ;; }||hj |s(|j|jhj |s>yycc}wcc}}w)NT F)token0rNrjoinrissubset)r^inspector_paramsmetadata_paramsbatchtsynonymsinspector_all_termsmetadata_all_termss r@_column_types_matchzDefaultImpl._column_types_matchs  " "o&<&< <<@)>%? ?  $4$9$99xx(//0668xx **+113((C3%B3%BbRXXa[BHHQK7 )rAc|j|}|j|}|j||sy|j||syy)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^inspector_columnmetadata_columnr r s r@ compare_typezDefaultImpl.compare_typesQ 556FG44_E''(8/J&&'7IrAc ||k7Sr`rG)r^rr rendered_metadata_defaultrendered_inspector_defaults r@compare_server_defaultz"DefaultImpl.compare_server_default.s*-FFFrAcyr`rG)r^ conn_uniques conn_indexesmetadata_unique_constraintsmetadata_indexess r@correct_for_autogen_constraintsz+DefaultImpl.correct_for_autogen_constraints7s rAcn|jj|jurt|d||d<yy)Nexpr)r8_type_affinityr)r^existingexisting_transfernew_types r@cast_for_batch_migratez"DefaultImpl.cast_for_batch_migrate@s: == ' 'x/F/F F(,!&)8) f % GrAc Xddd}t|j|j|S)zdRender a SQL expression that is typically a server default, index expression, etc. TF)rT include_table)rUrs)rBryrU)r^r-is_server_defaultrrs r@render_ddl_sql_exprzDefaultImpl.render_ddl_sql_exprFs/(,eD  LLjL I  rAc|jSr`)autogen_column_reflect)r^ inspectors r@_compat_autogen_column_reflectz*DefaultImpl._compat_autogen_column_reflectTs***rAcyr`rG)r^conn_fks metadata_fkss r@correct_for_autogen_foreignkeysz+DefaultImpl.correct_for_autogen_foreignkeysWs rAcy)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. NrG)r^r9rl column_infos r@r8z"DefaultImpl.autogen_column_reflect^rnrAcy)zA hook called when :meth:`.EnvironmentContext.run_migrations` is called. Implementations can set up per-migration-run state here. NrGrps r@start_migrationszDefaultImpl.start_migrationsgrnrAc@|jd|jzy)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`. BEGINNrfr|rps r@ emit_beginzDefaultImpl.emit_beginos 7T%<%<<=rAc@|jd|jzy)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`. COMMITNrErps r@ emit_commitzDefaultImpl.emit_commitys 8d&=&==>rAcyNFrG)r^type_objautogen_contexts r@ render_typezDefaultImpl.render_typesrAcTt||tjdh\}}t|dd}t|dd}t |t |k7r|j d|j |j||t |xst |t |k7fS)Nalways)skip)_compare_identity_optionsrIdentitygetattrbooladddifference_updaterR)r^metadata_identityinspector_identitydiffignored meta_alwaysinspector_alwayss r@_compare_identity_defaultz%DefaultImpl._compare_identity_defaults2    "  g/4@ "#5xF  %5 6 6 HHX  t99:   J M$01T:L5MM   rAcrt|j}t|j}||k7rd|d|Sy)Nzunique=z to unique=)rUunique)r^metadata_indexreflected_index conn_unique meta_uniques r@_compare_index_uniquez!DefaultImpl._compare_index_uniquesB?112 >001 + %[MB BrAc 2tjd||fi|S)NTrfrom_constraintr^ constraintoptss r@_create_metadata_constraint_sigz+DefaultImpl._create_metadata_constraint_sigs..tT:NNNrAc 2tjd||fi|SrKrgris r@ _create_reflected_constraint_sigz,DefaultImpl._create_reflected_constraint_sigs..udJO$OOrAc:g}|j||}|r|j||j|}|j|}t j |sJt j |sJ||fD]/}|j stjd|j1|j|jk7r+|jd|jd|j|rtj|StjS)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 )rerrlrnr is_index_sighas_expressionslogwarningr column_namesr DifferentEqual)r^rarbmsg unique_msgm_sigr_sigsigs r@compare_indexeszDefaultImpl.compare_indexess// O   JJz "44^D55oF$$U+++$$U+++%r8rBrFrIrNr^rerlrnr~rrrrGrAr@rIrIOs K,5y+A*CM'C&(m( .B?A)  *  ( % 4$$# +     (  16     7;&(!3!3!3!5 '8)'84'8$ '8  '8 '8X7;+ #+4+  +$(@E"&* $(,8=*..2<@,015nnn! n > n  n$nn&n6n(n,n":n*n!/n !n" #nh59 FFF2 F  F!% GGG G  G4159   0 2    3. 42342! $($$ $  $L&P &9? , &5; B +>D  G + ! &;  %    >C  !  6:  JM    + + 0      >?"5C #  8#6; O%O/2O %O P%P/2P %P ),),),  ),V,-,/,  ,8."; .;69; ;rArI) metaclassc6eZdZUded<ded<ded<ded<y) rrBrz List[str]rrzDict[str, str]rN)rDrErFrrGrAr@rrs K  O rArc tj|}tj|}tt dfd }|||tj|t|j|tjr6|t |dit |di|j t |difS)Nct|j D]c}|j|}|j|}||k7s+|j|}||k(r j|Sj|eyr`)set differencerXrV) meta_dict insp_dict default_dictattrsattr meta_value insp_value default_valuerZ ignored_attrrQs r@ check_dictsz._compare_identity_options..check_dicts,ss J))$/D"t,J"t,JZ' , 0 0 6 . $$T*HHTN0rAdialect_kwargs)rMapping[str, Any]rrrrrz Iterable[str])r_get_identity_options_dictrunionidentity_has_dialect_kwargsrTr) metadata_io inspector_io default_iorQmeta_dinsp_drrZrs ` @@r@rRrRs 3 3K @F  3 3L AF 5D5L#$#$#(# # ..z: F &!  ..  K!12 6 L"2B 7  % % L"2B 7    rA)r-Union[schema.Identity, schema.Sequence, None]rrrz'Union[schema.Identity, schema.Sequence]rQzSet[str])H __future__rloggingrtypingrrrrrr r r r r rrrr sqlalchemyrrrrrrrrrrrsqlalchemy.enginer r!sqlalchemy.engine.cursorr"sqlalchemy.engine.reflectionr#sqlalchemy.sqlr$r%sqlalchemy.sql.elementsr&r'sqlalchemy.sql.schemar(r)r*r+r,r-sqlalchemy.sql.selectabler.sqlalchemy.sql.type_apir/r0autogenerate.apir1operations.batchr2r3 getLoggerrDrtr8r5r:rrIrrRrGrAr@rs #  8:,)56,)53,0:++652$116g! t (*$)D ;HD ;NZ/>/?/8/  /rA