Ziz.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.zZEejeGZHGd8d9eIZJiZKd:eLd;<Gd<d=eJ>ZMGd?d@e ZN dBdAZOy)C) 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)BatchOperationsImplc eZdZ ddZy)ImplMetacVtj||||}d|vr |t|d<|S)N __dialect__)type__init___impls)cls classnamebasesdict_newtypes L/home/portmadness/app/.venv/lib/python3.12/site-packages/alembic/ddl/impl.pyr?zImplMeta.__init__Ls4 --Yu= E !+.F5' (N)rBstrrCzTuple[Type[DefaultImpl]]rDDict[str, Any])__name__ __module__ __qualname__r?rGrFr;r;Ks#  (  rGr;zDict[str, Type[DefaultImpl]]r@c eZdZUdZdZdZdZddhfZded<d Z d ed <d Z d ed< dAdZ e dBdZ dCdZ dDdZ dEdZ dFdZedGdZddej*f dHdZ dI dJdZddddddddddddd dKdZddddd dLdZddd dMdZdNdZdOd Z dI dPd!ZdQd"ZdQd#ZdRd$Z dSd%Z!dSd&Z"dTd'Z#dRd(Z$ dU dVd)Z%dWd*Z& dXd+Z' dYd,Z( dZd-Z)d.Z* d[d/Z+d0Z, d\ d]d1Z-d^d2Z. d_d3Z/d4Z0d`d5Z1d`d6Z2d`d7Z3 dad8Z4d9Z5 dbd:Z6 dcd;Z7 dcd<Z8 ddd=Z9 ded>Z:d?Z; dfd@Z!!---   /   "rGc t|ttdtdd|}|r|j t d|d|S)aGenerate a :class:`.Table` object which will be used as the structure for the Alembic version table. Third party dialects may override this hook to provide an alternate structure for this :class:`.Table`; requirements are only that it be named based on the ``version_table`` parameter and contains at least a single string-holding column named ``version_num``. .. versionadded:: 1.14 version_num F)nullabler_pkc)rg)rrrrappend_constraintr)rd version_tableversion_table_schemaversion_table_pkkwvts rFversion_table_implzDefaultImpl.version_table_implsW&  J =&*u ='     $!=/(>   rGcy)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. FrM)rdbatch_ops rFrequires_recreate_in_batchz&DefaultImpl.requires_recreate_in_batchsrGcy)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. NrM)rd batch_impltables rFprep_table_for_batchz DefaultImpl.prep_table_for_batchrGc|jSrf)r\rds rFbindzDefaultImpl.binds rGNct|tr t|}|jr||r t d|j r)t|t jstddi}ni}trt|tsJ|jd d|ji|}|jt|jddj|j zy|j"}|J|r|j$d i|}|r | t d|r|j'||S|j'||S) Nz&SQL parameters not allowed with as_sqlrZT)compile_kwargsr[ z z2Can't send params and multiparams at the same timerM) isinstancerHrr] TypeErrorrZr DDLElementdictrr-compiler[rlreplacestripcommand_terminatorr\execution_optionsexecute)rd constructr multiparamsparams compile_kwcompiledconns rF_execzDefaultImpl._execsA i %YI ;;&& HII!!*6,,+"/41HI  !)];;;(y((LLLH   H %%dF399;))* ??D# ## -t--B0AB+1H||I{;;||Iv66rGc(|j||yrf)r)rdsqlrs rFrzDefaultImpl.executes 3)*rG) rpserver_defaultrgtype_r autoincrementcommentexisting_comment existing_typeexisting_server_defaultexisting_nullableexisting_autoincrementc ||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  rGr if_not_existsinline_referencesinline_primary_keyc X|jtj||||||y)Nr)rr AddColumn)rdrcolumnrrrrs rF add_columnzDefaultImpl.add_column~s. NN+"3#5   rGr if_existsc T|jtj||||y)Nr)rr DropColumn)rdrrrrrvs rF drop_columnzDefaultImpl.drop_columns% OOF6Y  rGc |j|j|rHtjr|jdd|j t j |fi|yy)Nisolate_from_tableT) _create_ruler"sqla_2_1 setdefaultrr AddConstraintrdconstrvs rFadd_constraintzDefaultImpl.add_constraintsU    %););D)A## 2D9 JJv++E8R8 9 *BrGc N|jtj|fi|yrf)rrDropConstraintrs rFdrop_constraintzDefaultImpl.drop_constraints 6((5"56rGcR|jtj|||y)Nr)rr RenameTable)rdold_table_namenew_table_namers rF rename_tablezDefaultImpl.rename_tables"   ^^F K rGc d|jj||jd||jt j |fi||jj ||jd||jD]&}|jt j|(|jjxr|jj }|j}|r|r|j||jD]%}|j}|s|s|j|'yNF) checkfirst _ddl_runner)dispatch before_creater\rr CreateTable after_createindexes CreateIndexr[supports_commentsinline_commentsrcreate_table_commentcolumnscreate_column_comment)rdr~rvindex with_commentrrs rF create_tablezDefaultImpl.create_tables $$ 4??u$ %  6%%e2r23 ## 4??u$ $ ]] 2E JJv))%0 1 2 LL * * O4<<3O3O/O -- |  % %e ,mm 3FnnG<**62 3rGc |jj||jd||jt j |fi||jj ||jd|yr)r before_dropr\rr DropTable after_drop)rdr~rvs rF drop_tablezDefaultImpl.drop_tablesh "" 4??u$ #  6##E0R01 !! 4??u$ " rGc N|jtj|fi|yrf)rrrrdrrvs rF create_indexzDefaultImpl.create_indexs 6%%e2r23rGcL|jtj|yrf)rrSetTableCommentrdr~s rFrz DefaultImpl.create_table_comments 6))%01rGcL|jtj|yrf)rrDropTableCommentrs rFdrop_table_commentzDefaultImpl.drop_table_comments 6**512rGcL|jtj|yrf)rrSetColumnComment)rdrs rFrz!DefaultImpl.create_column_comments 6**623rGc N|jtj|fi|yrf)rr DropIndexrs rF drop_indexzDefaultImpl.drop_indexs 6##E0R01rGct|ts td|rt|dts td|jr|D]}|j |j jjdi|jDcic]Q\}}|t|tjs/tj|||j|jn|Sc}}y|rw|r0|j |j j|y|D]?}|j |j jjdi|Ayycc}}w)Nz List expectedrzList of dictionaries expected)r)rrM)rlistrrr]rinsertinlinevaluesitemsr"_literal_bindparamcr>)rdr~rows multiinsertrowkvs rF bulk_insertzDefaultImpl.bulk_insertsN $%O, , *T!Wd3;< < ;;  ELLNVXV),  !%1(2$%{'E'E("!, > >$%q !" &''  (JJu||~446DJI#J #A5<<>#8#8#:#A#A#HC#HIJ  sAE2cR|jjj|jj }t j d|}g}d}|D],}t jd|r|}|j|.t|d|ddgi}|rt j d|D]o}d|vr@|jd\} } | j|j| j<G|jj|jq|S)Nz[\w\-_]+|\(.+?\)z^\(.*\)$rrz[^(),]+=)r[ type_compilerprocessr>lowerrefindallmatchappendParamssplitrkwargsargs) rdr definitiontokens term_tokens paren_termtokenrtermkeyvals rF_tokenize_column_typez!DefaultImpl._tokenize_column_type s\\//77 DJJL JJ':JG!#   *Exx U+" ""5)  *  A ABR@  9j9 5$;#zz#HC14FMM#))+.KK&&tzz|4  5 rGc |j|jk(ry|jDcgc] }|Dchc]}|jc}"}}}dj|jg|jz}dj|jg|jz}|D]>}||hj |s(|j|jhj |s>yycc}wcc}}w)NT F)token0rTr joinrissubset)rdinspector_paramsmetadata_paramsbatchtsynonymsinspector_all_termsmetadata_all_termss rF_column_types_matchzDefaultImpl._column_types_match3s  " "o&<&< <<@)>%? ?  $4$9$99xx(//0668xx **+113(( C3%B3%BbRXXa[BHHQK7  rGc|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)rr(r4)rdinspector_columnmetadata_columnr!r"s rF compare_typezDefaultImpl.compare_typejsQ 556FG44_E''(8/J&&'7IrGc ||k7SrfrM)rdr6r7rendered_metadata_defaultrendered_inspector_defaults rFcompare_server_defaultz"DefaultImpl.compare_server_defaultzs*-FFFrGcyrfrM)rd conn_uniques conn_indexesmetadata_unique_constraintsmetadata_indexess rFcorrect_for_autogen_constraintsz+DefaultImpl.correct_for_autogen_constraintss rGcn|jj|jurt|d||d<yy)Nexpr)r>_type_affinityr)rdexistingexisting_transfernew_types rFcast_for_batch_migratez"DefaultImpl.cast_for_batch_migrates: == ' 'x/F/F F(,!&)8) f % GrGc Xddd}t|j|j|S)zdRender a SQL expression that is typically a server default, index expression, etc. TF)rZ include_table)r[r)rHrr[)rdrDis_server_defaultrvrs rFrender_ddl_sql_exprzDefaultImpl.render_ddl_sql_exprs/(,eD  LLjL I  rGc|jSrf)autogen_column_reflect)rd inspectors rF_compat_autogen_column_reflectz*DefaultImpl._compat_autogen_column_reflects***rGcyrfrM)rdconn_fks metadata_fkss rFcorrect_for_autogen_foreignkeysz+DefaultImpl.correct_for_autogen_foreignkeyss rGcy)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. NrM)rdrPr~ column_infos rFrOz"DefaultImpl.autogen_column_reflectrrGcy)zA hook called when :meth:`.EnvironmentContext.run_migrations` is called. Implementations can set up per-migration-run state here. NrMrs rFstart_migrationszDefaultImpl.start_migrationsrrGc@|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`. BEGINNrlrrs rF emit_beginzDefaultImpl.emit_begins 7T%<%<<=rGc@|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`. COMMITNr\rs rF emit_commitzDefaultImpl.emit_commits 8d&=&==>rGcyNFrM)rdtype_objautogen_contexts rF render_typezDefaultImpl.render_typesrGcTt||tjdh\}}t|dd}t|dd}t |t |k7r|j d|j |j||t |xst |t |k7fS)Nalways)skip)_compare_identity_optionsrIdentitygetattrbooladddifference_updaterX)rdmetadata_identityinspector_identitydiffignored meta_alwaysinspector_alwayss rF_compare_identity_defaultz%DefaultImpl._compare_identity_defaults2   OO   g/4@ "#5xF  %5 6 6 HHX  t99:   J M$01T:L5MM   rGcrt|j}t|j}||k7rd|d|Sy)Nzunique=z to unique=)rlunique)rdmetadata_indexreflected_index conn_unique meta_uniques rF_compare_index_uniquez!DefaultImpl._compare_index_uniquesB?112 >001 + %[MB BrGc 2tjd||fi|S)NTrfrom_constraintrd constraintoptss rF_create_metadata_constraint_sigz+DefaultImpl._create_metadata_constraint_sigs..tT:NNNrGc 2tjd||fi|Srbr~rs rF _create_reflected_constraint_sigz,DefaultImpl._create_reflected_constraint_sigs..udJO$OOrGc: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 )r|rrrr is_index_sighas_expressionslogwarningr column_namesr DifferentEqual)rdrxrymsg unique_msgm_sigr_sigsigs rFcompare_indexeszDefaultImpl.compare_indexess// O   JJz "44^D55oF$$U+++$$U+++%< C"" :II     !3!3 3 JJe001e6H6H5IJ  #--c2 2#))+ +rGc|j|}|j|}|j}|j}||k7rtjd|d|Stj S)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)rdmetadata_constraintreflected_constraint metadata_tup reflected_tupmeta_sigconn_sigs rFcompare_unique_constraintz%DefaultImpl.compare_unique_constraint/s;;  ==   '' (( x #--hZtH:6 $))+ +rGc T|Dcic]}|j|}}t|D]{}|j|vrtj|}|s*t j d|jd|j dtjd|j|}ycc}w)Nz@autogenerate skipping metadata-specified expression-based index z ; dialect z under SQLAlchemy z6 can't reflect these indexes so they can't be compared) rgrr"is_expression_indexr!rr=sqlalchemy_versiondiscard)rdrAr?rconn_indexes_by_nameidxiexs rF_skip_functional_indexesz$DefaultImpl._skip_functional_indexesKs3?@a @@() .Cxx//11#6C .xxl*T-=-=,@A""-"@"@!ABFF!((- . AsB%c&|jdiS)Ndialect_options)r^)rdreflected_objectkinds rF adjust_reflected_dialect_optionsz,DefaultImpl.adjust_reflected_dialect_options\s ##$5r::rG)r[r&r\Optional[Connection]r]rlrbOptional[bool]r_zOptional[TextIO]rarIreturnNone)r[r&rzType[DefaultImpl])rrHrr) rsrHrt Optional[str]rurlrvrrr)rzr9rrl)r}r8r~rrr)rr) rUnion[Executable, str]rzOptional[Mapping[str, Any]]rz%Optional[Sequence[Mapping[str, Any]]]rMapping[str, Any]rzOptional[CursorResult]rf)rrrzOptional[dict[str, Any]]rr) rrHrrHrprr3Optional[Union[_ServerDefaultType, Literal[False]]]rgrrOptional[TypeEngine]rrrrrz$Optional[Union[str, Literal[False]]]rrrrrrrrrrrvrrr)rrHr Column[Any]r!Optional[Union[str, quoted_name]]rrrrrrrr) rrHrrrrrrrr)rrrvrrr)rr0rvrrr)rrHrzUnion[str, quoted_name]rrrr)r~rrvrrr)rr2rvrrr)r~rrr)rrrr)T)r~zUnion[TableClause, Table]rz List[dict]rrlrr)rrrr)r!rr"rrrl)r-rr.rrrl)r6rr7rrrl) r>Set[UniqueConstraint]r? Set[Index]r@rrArrr)F)rDr-rLrlrvrrrH)rPr,rr)rSSet[ForeignKeyConstraint]rTrrr)rr)rcr5rdr7rzUnion[str, Literal[False]])rxr2ryr2rr)rz _autogen._Crrrz_constraint_sig[_autogen._C])rxr2ryr2rr)rr3rr3rr)r_ReflectedConstraintrrHrrI)=rJrKrL__doc__r=rbrrT__annotations__rUrXr? classmethodrhrlrxr{rpropertyrr! immutabledictrrrrrrrrrrrrrrrrrr(r4r8r<rBrIrMrQrUrOrYr]r`rerur|rrrrrrrMrGrFrOrO[s K,5y+A*CM'C&(m( .B?A)  *  ( % 4$$#  ,       D +     (  16     :>=A$6D$6$6$8 +7)+77+7; +7 " +7 +7`7;+ #+4+  +$( "&* $(,8=*..2 ,015)sss ! s  ss$ss&s6s(s,s " !s&*'s(!/)s*+s, -st59(,,0-1   2  &  * +   4!%$(        "     :759   0 2    3. 42342! $J($J$J $J  $JL&P &9? , &5; B +>D  G + ! &;  %    >C  !  6:  JM    + + 0      >?"5C #  8#6; O%O/2O %O P%P/2P %P ),),),  ),V,-,/,  ,8."; 4;.check_dictsxsx J))$/ #D"t,J"t,JZ' , 0 0 6 . $$T*HHTN #rGdialect_kwargs)rrrrrrrz Iterable[str])r"_get_identity_options_dictrunionidentity_has_dialect_kwargshasattrrkr) metadata_io inspector_io default_iorhmeta_dinsp_drrqrs ` @@rFririis 3 3K @F  3 3L AF 5D5L#$#$#(# # ..z: F &!  ..z#3444  K!12 6 L"2B 7  % % L"2B 7    rG)r-Union[schema.Identity, schema.Sequence, None]rrrz'Union[schema.Identity, schema.Sequence]rhzSet[str])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 getLoggerrJrr>r;r@rrOrrirMrGrFrsA#  +8:,)5J;JF6,)30:+652(116 & ' (   $ $ g! t (*$)D ;HD ;NZ0>0?080  0rG