jxzUddlmZddlmZddlmZddlmZddlZddlZddlm Z ddlm Z ddlm Z dd lm Z dd l mZdd l mZdd l mZdd l mZddl mZddl mZddl mZddl mZddlmZddlmZddlmZddlmZddlmZddlmZe rddlmZddlmZddlm Z ddlm!Z!ddl"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/m0Z0dd(l1m2Z2dd)l1m3Z3dd*l1m4Z4dd+l1m5Z5dd,l1m6Z6dd-l7m8Z8dd.l9m:Z:dd/lm;Z;dd0lm?Z?dd2l>m@Z@e%e&ze(ze'ze)zZAejBeCZDGd3d4eEZFiZGd5eHd6<Gd7d8eF9ZIGd:d;e ZJdDdCZKdS)E) annotations)Iterable)Mapping)SequenceN)Any)Callable) NamedTuple) TYPE_CHECKING)cast)Column)MetaData)PrimaryKeyConstraintschema)String)Table)text)_autogen)base)_constraint_sig)ComparisonResult)util) sqla_compat)Literal)TextIO) Connection)Dialect) CursorResult)ReflectedCheckConstraint)ReflectedForeignKeyConstraint)ReflectedIndex)ReflectedPrimaryKeyConstraint)ReflectedUniqueConstraint) Inspector) ClauseElement) Executable) quoted_name)CheckConstraint) Constraint)ForeignKeyConstraint)Index)UniqueConstraint) TableClause) TypeEngine)_ServerDefaultArgument)AutogenContext)ApplyBatchImpl)BatchOperationsImplceZdZd dZdS) ImplMeta classnamestrbasestuple[type[DefaultImpl]]dict_dict[str, Any]cht||||}d|vr|t|d<|S)N __dialect__)type__init___impls)clsr7r9r;newtypes s/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/alembic/ddl/impl.pyr@zImplMeta.__init__Hs: --Yu== E ! !+.F5' (N)r7r8r9r:r;r<)__name__ __module__ __qualname__r@rErDr6r6Gs(      rEr6zdict[str, type[DefaultImpl]]rAc @eZdZUdZdZdZdZddhfZded<d Z d ed <d Z d ed<ddZ e ddZ dd"Zdd*Zdd-Zdd1Zedd2Zd3d3ejfdd=Z ddd@Zd3dd3d3d3d3dd3d3d3d3d3dA ddSZd3d3d3d3dTdd[Zd3d3d\dd^Zdd`ZddbZ dddfZddgZddhZddkZ ddlZ!ddmZ"ddnZ#ddoZ$ ddduZ%ddxZ&dd{Z'dd~Z(ddZ)dZ*ddZ+dZ, dddZ-ddZ.ddZ/dZ0ddZ1ddZ2ddZ3ddZ4dZ5ddZ6ddZ7ddZ8ddZ9ddZ:ddZ;dZ(>(>     rEbatch_opr4cdS)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. FrI)rfr~s rDrequires_recreate_in_batchz&DefaultImpl.requires_recreate_in_batchs urE batch_implr3tablecdS)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. NrI)rfrrs rDprep_table_for_batchz DefaultImpl.prep_table_for_batchrEc|jSri)rVrfs rDbindzDefaultImpl.binds rEN constructExecutable | strexecution_optionsMapping[str, Any] | None multiparams"Sequence[Mapping[str, Any]] | NoneparamsMapping[str, Any]CursorResult | Nonect|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_sqlrbT)compile_kwargsrU z z2Can't send params and multiparams at the same timerI) isinstancer8rrX TypeErrorrbr DDLElementdictr r'compilerUroreplacestripcommand_terminatorrVrexecute)rfrrrr compile_kwcompiledconns rD_execzDefaultImpl._execs i % % (YI ;" 7&&& HIII! *6,++ "/41HIII   <!)];;;;;(y(LLLLLH   H %%dF3399;;)*   4?D###  C-t-BB0ABB +1H 7||I{;;;||Iv666rEsqldict[str, Any] | Nonec2|||dSri)r)rfrrs rDrzDefaultImpl.executes 3)*****rE) rxserver_defaultrjtype_r autoincrementcommentexisting_comment existing_typeexisting_server_defaultexisting_nullableexisting_autoincrement table_name column_namerxrr1rjrTypeEngine | Nonerrrstr | Literal[False] | Nonerrrrrc ||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)rwarnrrColumnNullabler_server_default_is_computedComputedColumnDefault_server_default_is_identityIdentityColumnDefault ColumnDefault ColumnType ColumnComment ColumnName)rfrrrxrrjrrrrrrrrrrtcls_s rD 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       rEr if_not_existsinline_referencesinline_primary_keycolumn Column[Any]str | quoted_name | Nonerrrc `|tj||||||dS)Nr)rr AddColumn)rfrrrrrrs rD add_columnzDefaultImpl.add_columnvsJ N+"3#5     rEr if_existsrc \|tj||||dS)Nr)rr DropColumn)rfrrrrrts rD drop_columnzDefaultImpl.drop_columnsA OF6Y        rEconstc |j||rItjr|dd|t j|fi|dSdS)Nisolate_from_tableT) _create_rulersqla_2_1 setdefaultrr AddConstraintrfrrts rDadd_constraintzDefaultImpl.add_constraintsq   %););D)A)A %# : 2D999 JJv+E88R88 9 9 9 9 9 & %rEr+c P|tj|fi|dSri)rrDropConstraintrs rDdrop_constraintzDefaultImpl.drop_constraints- 6(55"5566666rEold_table_namenew_table_namestr | quoted_namecZ|tj|||dS)Nr)rr RenameTable)rfrrrs rD rename_tablezDefaultImpl.rename_tables9  ^^F K K K     rEc |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_createrVrr CreateTable after_createindexes CreateIndexrUsupports_commentsinline_commentsrcreate_table_commentcolumnscreate_column_comment)rfrrtindex with_commentrrs rD 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 3rEc |j||jd||t j|fi||j||jd|dSr)r before_droprVrr DropTable after_drop)rfrrts rD drop_tablezDefaultImpl.drop_tables "" 4?u$ #    6#E00R00111 !! 4?u$ "     rErr-c P|tj|fi|dSri)rrrrfrrts rD create_indexzDefaultImpl.create_indexs- 6%e22r2233333rEcT|tj|dSri)rrSetTableCommentrfrs rDrz DefaultImpl.create_table_comments% 6)%0011111rEcT|tj|dSri)rrDropTableCommentrs rDdrop_table_commentzDefaultImpl.drop_table_comments% 6*51122222rEcT|tj|dSri)rrSetColumnComment)rfrs rDrz!DefaultImpl.create_column_comments% 6*62233333rEc P|tj|fi|dSri)rr DropIndexrs rD drop_indexzDefaultImpl.drop_indexs- 6#E00R0011111rETTableClause | Tablerows 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 rD z+DefaultImpl.bulk_insert..sx   !%1(2$%{'E("("!' >$%q !"!"!"!" &'   rE)rrI) rlistrrrXrinsertinlinevaluesitems)rfrr r rows ` rD bulk_insertzDefaultImpl.bulk_inserts $%% =O,, ,  =*T!Wd33 =;<< < ; J   ELLNNVXX    ),      ( JJJJu||~~4466DJIIIII#JJ #A5<<>>#8#8#:#:#A#H#HC#H#HIIII  J JJJrEr Paramscj|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[^(),]+=)rU type_compilerprocessr?lowerrefindallmatchappendrsplitrkwargsargs) rfr definitiontokens term_tokens paren_termtokenrtermkeyvals rD_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 rEinspector_paramsmetadata_paramscX|j|jkrdSd|jD}d|jg|jz}d|jg|jz}|D]=}||h|s!|j|jh|rdS>dS)NTc&g|]}d|DS)c6h|]}|SrI)r )rts rD z=DefaultImpl._column_types_match...1s ...1QWWYY...rErI)rbatchs rD z3DefaultImpl._column_types_match..1s'OOO5.....OOOrE F)token0rPjoinr)issubset)rfr1r2synonymsinspector_all_termsmetadata_all_termsr8s rD_column_types_matchzDefaultImpl._column_types_match+s  "o&< < <4OOD%?%? ? ? $4$9995xx(/006688xx *++1133(  C3%%B3%%B b RXXa[[BHHQKK77uutrEinspector_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)r0rArM)rfrNrOr1r2s rD compare_typezDefaultImpl.compare_typebsm 556FGG44_EE''(8/JJ 4&&'7II 4urEc||kSrirI)rfrNrOrendered_metadata_defaultrendered_inspector_defaults rDcompare_server_defaultz"DefaultImpl.compare_server_defaultrs*-FFFrE conn_uniquesset[UniqueConstraint] conn_indexes set[Index]metadata_unique_constraintsmetadata_indexescdSrirI)rfrVrXrZr[s rDcorrect_for_autogen_constraintsz+DefaultImpl.correct_for_autogen_constraints{s  rEcb|jj|jurt|d||d<dSdS)Nexpr)r?_type_affinityr )rfexistingexisting_transfernew_types rDcast_for_batch_migratez"DefaultImpl.cast_for_batch_migratesC = 'x/F F F(,!&)8)) f % % % G FrEr_r'is_server_defaultc ^ddd}t||j|S)zdRender a SQL expression that is typically a server default, index expression, etc. TF)rb include_table)rUr)r8rrU)rfr_rertrs rDrender_ddl_sql_exprzDefaultImpl.render_ddl_sql_exprs8(,eDD  LLjL I I   rE inspectorr&rc|jSri)autogen_column_reflect)rfris rD_compat_autogen_column_reflectz*DefaultImpl._compat_autogen_column_reflects **rEconn_fksset[ForeignKeyConstraint] metadata_fkscdSrirI)rfrmros rDcorrect_for_autogen_foreignkeysz+DefaultImpl.correct_for_autogen_foreignkeyss rEcdS)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. NrI)rfrir column_infos rDrkz"DefaultImpl.autogen_column_reflectrrEcdS)zA hook called when :meth:`.EnvironmentContext.run_migrations` is called. Implementations can set up per-migration-run state here. NrIrs rDstart_migrationszDefaultImpl.start_migrationsrrEc@|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`. BEGINNrorrs rD emit_beginzDefaultImpl.emit_begins& 7T%<<=====rEc@|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`. COMMITNrxrs rD emit_commitzDefaultImpl.emit_commits& 8d&==>>>>>rEtype_objr0autogen_contextr2str | Literal[False]cdSNFrI)rfr}r~s rD render_typezDefaultImpl.render_types urEct||tjdh\}}t|dd}t|dd}t |t |kr|d||j||t |pt |t |kfS)Nalways)skip)_compare_identity_optionsrIdentitygetattrrYadddifference_updaterT)rfmetadata_identityinspector_identitydiffignored meta_alwaysinspector_alwayss rD_compare_identity_defaultz%DefaultImpl._compare_identity_defaults2   O      g/4@@ "#5xFF   %5 6 6 6 6 HHX    t9:::   JJ M$011T:L5M5MM   rEmetadata_indexreflected_indexcrt|j}t|j}||krd|d|SdS)Nzunique=z to unique=)rYunique)rfrr conn_unique meta_uniques rD_compare_index_uniquez!DefaultImpl._compare_index_uniquesJ?122 >011 + % %B[BB[BB B4rE constraint _autogen._Copts_constraint_sig[_autogen._C]c *tjd||fi|S)NTrfrom_constraintrfrrs rD_create_metadata_constraint_sigz+DefaultImpl._create_metadata_constraint_sigs!.tT:NNNNNrEc *tjd||fi|Srrrs rD _create_reflected_constraint_sigz,DefaultImpl._create_reflected_constraint_sigs!.udJOO$OOOrErc$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)rfrrmsg unique_msgm_sigr_sigsigs rDcompare_indexeszDefaultImpl.compare_indexessB// O    # JJz " " "44^DD55oFF$U+++++$U+++++%<  C"  :I   !3 3 3 JJJe0JJe6HJJ     ,#-c22 2#)++ +rEmetadata_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)rfrr metadata_tup reflected_tupmeta_sigconn_sigs rDcompare_unique_constraintz%DefaultImpl.compare_unique_constraint's;;   ==    ' ( x  #-6h66H66 $)++ +rEr*c(tjSri)rr)rfrrs rDcompare_check_constraintz$DefaultImpl.compare_check_constraintCs  %'''rEc d|D}t|D]i}|j|vr tj|}|rGt jd|jd|jdtjd||jdS)Nci|] }|j| SrIrz)rrs rDrz8DefaultImpl._skip_functional_indexes..Ks@@@a@@@rEz@autogenerate skipping metadata-specified expression-based index z ; dialect z under SQLAlchemy z6 can't reflect these indexes so they can't be compared) rrjris_expression_indexrrr>sqlalchemy_versiondiscard)rfr[rXconn_indexes_by_nameidxiexs rD_skip_functional_indexesz$DefaultImpl._skip_functional_indexesJs@@<@@@()) . .Cx///1#66C . FxFF-1-=FF#."@FFF!((--- . .rEreflected_object_ReflectedConstraintkindc.|diS)Ndialect_options)rc)rfrrs rD adjust_reflected_dialect_optionsz,DefaultImpl.adjust_reflected_dialect_options[s ##$5r:::rE)rUrrVrWrXrYrZr[r\r]r^r<r_r`)rUrr_rg)rr8r_r`) rpr8rqrrrsrYrtrr_r)r~r4r_rY)rr3rrr_r`)r_rW) rrrrrrrrr_rri)rrrrr_r`) rr8rr8rxr[rr1rjrrrrrrrrr[rrrrrrrrr1rr[rr[rtrr_r`)rr8rrrrrr[rr[rr[r_r`) rr8rrrrrrr[r_r`)rrrtrr_r`)rr+rtrr_r`)rr8rrrrr_r`)rrrtrr_r`)rr-rtrr_r`)rrr_r`)rrr_r`)T)rrr r r rYr_r`)rr r_r)r1rr2rr_rY)rBrrCrr_rY)rNrrOr r_rY) rVrWrXrYrZrWr[rYr_r`)F)r_r'rerYrtrr_r8)rir&r_r)rmrnrornr_r`)r_r`)r}r0r~r2r_r)rr-rr-r_rr)rrrrr_r)rr-rr-r_r)rr.rr.r_r)rr*rr*r_r)rrrr8r_r<)>rFrGrH__doc__r>rZrrP__annotations__rQrTr@ classmethodrkror}rrpropertyrr immutabledictrrrrrrrrrrrrrrrrr0rArMrQrUr]rdrhrlrqrkruryr|rrrrrrrrrrrIrErDrKrKWs  K,5y+A*CMCCCC&((((( .BAAAA4$$$[$####    D         X 7;:>$6D$6$8$8 +7+7+7+7+7`48+++++!%16#'!%)/4'++/:>)-.2!ooooool,0%))-*.      4"!%       ::::7777,0      3333.    44442222333344442222! $J$J$J$J$JL&&&&P,B GGG    >C      ++++           >>>>????    8OOOO PPPP ),),),),V,,,,8((((...";;;;;;rErK) metaclassc8eZdZUded<ded<ded<ded<dS) rr8r;z list[str]r)r'zdict[str, str]r&N)rFrGrHrrIrErDrras<KKKOOOrEr metadata_io(schema.Identity | schema.Sequence | None inspector_io default_io!schema.Identity | schema.Sequencerset[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|ydSri)set differencercr) rrrrattr meta_value insp_value default_valuer ignored_attrrs rD check_dictsz._compare_identity_options..check_dictsws JJ))$// # #D"t,,J"t,,JZ'' , 0 0 6 6 .. $$T****HHTNNN # #rEdialect_kwargs)rrrrrrrr)r_get_identity_options_dictrunionidentity_has_dialect_kwargshasattrrr) rrrrmeta_dinsp_drrrs ` @@rDrrhs 3K @ @F  3L A AF 55D55L######## K.z:: F &!!  .  z#344444  K!12 6 6 L"2B 7 7  % L"2B 7 7      rE)rrrrrrrr)L __future__rcollections.abcrrrloggingr!typingrrr r sqlalchemyr r r rrrrrrrrrrrrrsqlalchemy.enginerrsqlalchemy.engine.cursorr sqlalchemy.engine.interfacesr!r"r#r$r%sqlalchemy.engine.reflectionr&sqlalchemy.sqlr'r(sqlalchemy.sql.elementsr)sqlalchemy.sql.schemar*r+r,r-r.sqlalchemy.sql.selectabler/sqlalchemy.sql.type_apir0r1autogenerate.apir2operations.batchr3r4r getLoggerrFrr?r6rArrKrrrIrErDrsr#""""""$$$$$$######$$$$$$  ++++++888888::::::#,,,,,,))))))555555EEEEEEJJJJJJ;;;;;;JJJJJJFFFFFF666666,,,,,,))))))333333555555000000::::::++++++666666555555222222,,,,,,111111111111666666 ! ' ( ' (   $  $g!!     t   (*))))G ;G ;G ;G ;G ;HG ;G ;G ;G ;TZ000000rE