Ojz.UddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!dd l!m"Z"erdd!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*m.Z.dd*l/m0Z0dd+l1m2Z2dd,l1m3Z3dd-l4m5Z5dd.l6m7Z7dd/l6m8Z8dd0l6m9Z9dd1l6m:Z:dd2l;mZ>dd4lm?Z?dd5l@mAZAdd6lBmCZCdd7lBmDZDe+e-ze,ze.zZEejFeGZHGd8d9eIZJiZKd:eLd;<Gd<d=eJ>ZMGd?d@e ZNdIdHZOdS)J) annotationsN)Any)Callable)Dict)Iterable)List)Mapping) NamedTuple)Optional)Sequence)Set)Tuple)Type) TYPE_CHECKING)Union)cast)Column)MetaData)PrimaryKeyConstraintschema)String)Table)text)_autogen)base)_constraint_sig)ComparisonResult)util) sqla_compat)Literal)TextIO) Connection)Dialect) CursorResult)ReflectedForeignKeyConstraint)ReflectedIndex)ReflectedPrimaryKeyConstraint)ReflectedUniqueConstraint) Inspector) ClauseElement) Executable) quoted_name) Constraint)ForeignKeyConstraint)Index)UniqueConstraint) TableClause) TypeEngine)_ServerDefaultType)AutogenContext)ApplyBatchImpl)BatchOperationsImplceZdZd dZdS) ImplMeta classnamestrbasesTuple[Type[DefaultImpl]]dict_Dict[str, Any]cht||||}d|vr|t|d<|S)N __dialect__)type__init___impls)clsr<r>r@newtypes d/lsinfo/ai/hellotax_ai/training_center/backend/venv/lib/python3.11/site-packages/alembic/ddl/impl.pyrEzImplMeta.__init__Ls: --Yu== E ! !+.F5' (N)r<r=r>r?r@rA)__name__ __module__ __qualname__rErJrIr;r;Ks(      rJr;zDict[str, Type[DefaultImpl]]rFc 8eZdZUdZdZdZdZddhfZded<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:dZ;ddZ(>(>     rJbatch_opr9cdS)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. FrN)rkrs rIrequires_recreate_in_batchz&DefaultImpl.requires_recreate_in_batchs urJ batch_implr8tablecdS)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. NrN)rkrrs rIprep_table_for_batchz DefaultImpl.prep_table_for_batchrJc|jSrn)r[rks rIbindzDefaultImpl.binds rJN 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_sqlrgT)compile_kwargsrZ z z2Can't send params and multiparams at the same timerN) isinstancer=rr] TypeErrorrgr DDLElementdictrr-compilerZrtreplacestripcommand_terminatorr[rexecute)rkrrrr compile_kwcompiledconns rI_execzDefaultImpl._execs i % % (YI ;" 7&&& HIII! *6,++ "/41HIII   <!)];;;;;(y(LLLLLH   H %%dF3399;;)*   4?D###  C-t-BB0ABB +1H 7||I{;;;||Iv666rJsqlOptional[dict[str, Any]]c2|||dSrn)r)rkrrs rIrzDefaultImpl.executes 3)*****rJ) r}server_defaultrotype_r autoincrementcommentexisting_comment existing_typeexisting_server_defaultexisting_nullableexisting_autoincrement table_name column_namer}r3Optional[Union[_ServerDefaultType, Literal[False]]]rorOptional[TypeEngine]rrr$Optional[Union[str, Literal[False]]]rrrrrc ||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)rkrrr}rrorrrrrrrrrrycls_s rI alter_columnzDefaultImpl.alter_column s5.  $(>(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       rJr if_not_existsinline_referencesinline_primary_keycolumn Column[Any]!Optional[Union[str, quoted_name]]rrrc `|tj||||||dS)Nr)rr AddColumn)rkrrrrrrs rI add_columnzDefaultImpl.add_column~sJ N+"3#5     rJr if_existsrc \|tj||||dS)Nr)rr DropColumn)rkrrrrrys rI drop_columnzDefaultImpl.drop_columnsA OF6Y        rJconstc |j||rItjr|dd|t j|fi|dSdS)Nisolate_from_tableT) _create_ruler"sqla_2_1 setdefaultrr AddConstraintrkrrys rIadd_constraintzDefaultImpl.add_constraintsq   %););D)A)A %# : 2D999 JJv+E88R88 9 9 9 9 9 & %rJr0c P|tj|fi|dSrn)rrDropConstraintrs rIdrop_constraintzDefaultImpl.drop_constraints- 6(55"5566666rJold_table_namenew_table_nameUnion[str, quoted_name]cZ|tj|||dS)Nr)rr RenameTable)rkrrrs rI rename_tablezDefaultImpl.rename_tables9  ^^F K K K     rJc |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_creater[rr CreateTable after_createindexes CreateIndexrZsupports_commentsinline_commentsrcreate_table_commentcolumnscreate_column_comment)rkrryindex with_commentrrs rI 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 3rJc |j||jd||t j|fi||j||jd|dSr)r before_dropr[rr DropTable after_drop)rkrrys rI drop_tablezDefaultImpl.drop_tables "" 4?u$ #    6#E00R00111 !! 4?u$ "     rJrr2c P|tj|fi|dSrn)rrrrkrrys rI create_indexzDefaultImpl.create_indexs- 6%e22r2233333rJcT|tj|dSrn)rrSetTableCommentrkrs rIrz DefaultImpl.create_table_comments% 6)%0011111rJcT|tj|dSrn)rrDropTableCommentrs rIdrop_table_commentzDefaultImpl.drop_table_comments% 6*51122222rJcT|tj|dSrn)rrSetColumnComment)rkrs rIrz!DefaultImpl.create_column_comments% 6*62233333rJc P|tj|fi|dSrn)rr DropIndexrs rI drop_indexzDefaultImpl.drop_indexs- 6#E00R0011111rJTUnion[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_bindparamcrD).0kvrs rI z+DefaultImpl.bulk_insert..sx   !%1(2$%{'E("("!' >$%q !"!"!"!" &'   rJ)rrN) rlistrrr]rinsertinlinevaluesitems)rkrrrrows ` rI bulk_insertzDefaultImpl.bulk_inserts $%% =O,, ,  =*T!Wd33 =;<< < ; J   ELLNNVXX    ),      ( JJJJu||~~4466DJIIIII#JJ #A5<<>>#8#8#:#:#A#H#HC#H#HIIII  J JJJrJrParamscj|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[^(),]+=)rZ type_compilerprocessrDlowerrefindallmatchappendr!splitrkwargsargs) rkr definitiontokens term_tokens paren_termtokenrtermkeyvals rI_tokenize_column_typez!DefaultImpl._tokenize_column_type s*\/77 DDJJLL J':JGG!#   * *Ex U++ *" ""5)))) A ABBR@@  5 9j99 5 5$;;#zz#HC14FM#))++..K&&tzz||4444 rJinspector_paramsmetadata_paramscX|j|jkrdSd|jD}d|jg|jz}d|jg|jz}|D]=}||h|s!|j|jh|rdS>dS)NTc&g|]}d|DS)c6h|]}|SrN)r&)rts rI z=DefaultImpl._column_types_match...9s ...1QWWYY...rJrN)rbatchs rI z3DefaultImpl._column_types_match..9s'OOO5.....OOOrJ F)token0rUjoinr/issubset)rkr7r8synonymsinspector_all_termsmetadata_all_termsr>s rI_column_types_matchzDefaultImpl._column_types_match3s  "o&< < <4OOD%?%? ? ? $4$9995xx(/006688xx *++1133(  C3%%B3%%B b RXXa[[BHHQKK77uutrJinspector_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)r6rGrS)rkrTrUr7r8s rI compare_typezDefaultImpl.compare_typejsm 556FGG44_EE''(8/JJ 4&&'7II 4urJc||kSrnrN)rkrTrUrendered_metadata_defaultrendered_inspector_defaults rIcompare_server_defaultz"DefaultImpl.compare_server_defaultzs*-FFFrJ conn_uniquesSet[UniqueConstraint] conn_indexes Set[Index]metadata_unique_constraintsmetadata_indexescdSrnrN)rkr\r^r`ras rIcorrect_for_autogen_constraintsz+DefaultImpl.correct_for_autogen_constraintss  rJcb|jj|jurt|d||d<dSdS)Nexpr)rD_type_affinityr)rkexistingexisting_transfernew_types rIcast_for_batch_migratez"DefaultImpl.cast_for_batch_migratesC = 'x/F F F(,!&)8)) f % % % G FrJrer-is_server_defaultc ^ddd}t||j|S)zdRender a SQL expression that is typically a server default, index expression, etc. TF)rg include_table)rZr)r=rrZ)rkrerkryrs rIrender_ddl_sql_exprzDefaultImpl.render_ddl_sql_exprs8(,eDD  LLjL I I   rJ inspectorr,rc|jSrn)autogen_column_reflect)rkros rI_compat_autogen_column_reflectz*DefaultImpl._compat_autogen_column_reflects **rJconn_fksSet[ForeignKeyConstraint] metadata_fkscdSrnrN)rkrsrus rIcorrect_for_autogen_foreignkeysz+DefaultImpl.correct_for_autogen_foreignkeyss rJcdS)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. NrN)rkror column_infos rIrqz"DefaultImpl.autogen_column_reflectrrJcdS)zA hook called when :meth:`.EnvironmentContext.run_migrations` is called. Implementations can set up per-migration-run state here. NrNrs rIstart_migrationszDefaultImpl.start_migrationsrrJc@|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`. BEGINNrtrrs rI emit_beginzDefaultImpl.emit_begins& 7T%<<=====rJc@|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`. COMMITNr~rs rI emit_commitzDefaultImpl.emit_commits& 8d&==>>>>>rJtype_objr5autogen_contextr7Union[str, Literal[False]]cdSNFrN)rkrrs rI render_typezDefaultImpl.render_types urJct||tjdh\}}t|dd}t|dd}t |t |kr|d||j||t |pt |t |kfS)Nalways)skip)_compare_identity_optionsrIdentitygetattrr^adddifference_updaterY)rkmetadata_identityinspector_identitydiffignored meta_alwaysinspector_alwayss rI_compare_identity_defaultz%DefaultImpl._compare_identity_defaults2   O      g/4@@ "#5xFF   %5 6 6 6 6 HHX    t9:::   JJ M$011T:L5M5MM   rJmetadata_indexreflected_indexcrt|j}t|j}||krd|d|SdS)Nzunique=z to unique=)r^unique)rkrr conn_unique meta_uniques rI_compare_index_uniquez!DefaultImpl._compare_index_uniquesJ?122 >011 + % %B[BB[BB B4rJ constraint _autogen._Copts_constraint_sig[_autogen._C]c *tjd||fi|S)NTrfrom_constraintrkrrs rI_create_metadata_constraint_sigz+DefaultImpl._create_metadata_constraint_sigs!.tT:NNNNNrJc *tjd||fi|Srrrs rI _create_reflected_constraint_sigz,DefaultImpl._create_reflected_constraint_sigs!.udJOO$OOOrJrc$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)rkrrmsg unique_msgm_sigr_sigsigs rIcompare_indexeszDefaultImpl.compare_indexessB// O    # JJz " " "44^DD55oFF$U+++++$U+++++%<  C"  :I   !3 3 3 JJJe0JJe6HJJ     ,#-c22 2#)++ +rJmetadata_constraintr3reflected_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)rkrr metadata_tup reflected_tupmeta_sigconn_sigs rIcompare_unique_constraintz%DefaultImpl.compare_unique_constraint/s;;   ==    ' ( x  #-6h66H66 $)++ +rJc d|D}t|D]i}|j|vr tj|}|rGt jd|jd|jdtjd||jdS)Nci|] }|j| SrNr)rrs rIrz8DefaultImpl._skip_functional_indexes..Ls@@@a@@@rJz@autogenerate skipping metadata-specified expression-based index z ; dialect z under SQLAlchemy z6 can't reflect these indexes so they can't be compared) rror"is_expression_indexr!rrCsqlalchemy_versiondiscard)rkrar^conn_indexes_by_nameidxiexs rI_skip_functional_indexesz$DefaultImpl._skip_functional_indexesKs@@<@@@()) . .Cx///1#66C . FxFF-1-=FF#."@FFF!((--- . .rJreflected_object_ReflectedConstraintkindc.|diS)Ndialect_options)rh)rkrrs rI adjust_reflected_dialect_optionsz,DefaultImpl.adjust_reflected_dialect_options\s ##$5r:::rJ)rZr&r[r\r]r^r_r`rarbrcrArdre)rZr&rdrl)rr=rdre) rur=rvrwrxr^ryrrdr)rr9rdr^)rr8rrrdre)rdr\) rrrrrrrrrdrrn)rrrrrdre) rr=rr=r}r`rrrorwrrrrwrr`rrrrwrrrrrr`rr`ryrrdre)rr=rrrrrr`rr`rr`rdre) rr=rrrrwrr`rdre)rrryrrdre)rr0ryrrdre)rr=rrrrrdre)rrryrrdre)rr2ryrrdre)rrrdre)rrrdre)T)rrrrrr^rdre)rrrdr!)r7r!r8r!rdr^)rHr!rIr!rdr^)rTrrUrrdr^) r\r]r^r_r`r]rar_rdre)F)rer-rkr^ryrrdr=)ror,rdr)rsrtrurtrdre)rdre)rr5rr7rdr)rr2rr2rdrw)rrrrrdr)rr2rr2rdr)rr3rr3rdr)rrrr=rdrA)=rKrLrM__doc__rCr_rrU__annotations__rVrYrE classmethodrprtrrrpropertyrr! immutabledictrrrrrrrrrrrrrrr r r6rGrSrWr[rcrjrnrrrwrqr{rrrrrrrrrrrrNrJrIrPrP[s  K,5y+A*CMCCCC&((((( .BAAAA4$$$[$####    D         X :>=A$6D$6$8$8 +7+7+7+7+7`7;+++++$( "&* $(,8=*..2 ,015)sssssst59(,,0-1      4!%$(       ::::777759      3333.    44442222333344442222! $J$J$J$J$JL&&&&P,B GGG    >C      ++++           >>>>????    8OOOO PPPP ),),),),V,,,,8...";;;;;;rJrP) metaclassc8eZdZUded<ded<ded<ded<dS) r!r=rAz List[str]r/r-zDict[str, str]r,N)rKrLrMrrNrJrIr!r!bs<KKKOOOrJr! 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|ydSrn)set differencerhr) rrrrattr meta_value insp_value default_valuer ignored_attrrs rI check_dictsz._compare_identity_options..check_dictsxs JJ))$// # #D"t,,J"t,,JZ'' , 0 0 6 6 .. $$T****HHTNNN # #rJdialect_kwargs)rrrrrrrr)r"_get_identity_options_dictrunionidentity_has_dialect_kwargshasattrrr) rrrrmeta_dinsp_drrrs ` @@rIrris 3K @ @F  3L A AF 55D55L######## K.z:: F &!!  .  z#344444  K!12 6 6 L"2B 7 7  % L"2B 7 7      rJ)rrrrrrrr)P __future__rloggingr'typingrrrrrr r r r r rrrr sqlalchemyrrrrrrrrrrrrr!r"r#r$sqlalchemy.enginer%r&sqlalchemy.engine.cursorr'sqlalchemy.engine.interfacesr(r)r*r+sqlalchemy.engine.reflectionr,sqlalchemy.sqlr-r.sqlalchemy.sql.elementsr/sqlalchemy.sql.schemar0r1r2r3sqlalchemy.sql.selectabler4sqlalchemy.sql.type_apir5r6autogenerate.apir7operations.batchr8r9r getLoggerrKrrDr;rFrrPr!rrNrJrIrs#""""""  ++++++888888:::::: ,,,,,,))))))555555JJJJJJ;;;;;;JJJJJJFFFFFF666666,,,,,,))))))333333000000::::::++++++666666555555222222((((((111111111111666666 & ' (   $ $ g!!     t   (*))))D ;D ;D ;D ;D ;HD ;D ;D ;D ;NZ000000rJ