>il'N d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d lm
Z
d d
l m Z d d l
mZ e r6d d lmZ d d
lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ G d d ZdS ) ) annotations)Any)Callable)Iterator)List)Optional)Type)
TYPE_CHECKING)Union)util)ops) AddColumnOp)
AlterColumnOp)
CreateTableOp)MigrateOperation)MigrationScript)ModifyTableOps) OpContainer)MigrationContext)Revisionc X e Zd ZU dZ ej ZdZded< d"dZ d#d Z
d$d
Z d%dZ d&dZ
e ej d'd Ze ej d(d Ze ej d)d Zd*d Zd+d Zd&d!ZdS ),Rewritera5 A helper object that allows easy 'rewriting' of ops streams.
The :class:`.Rewriter` object is intended to be passed along
to the
:paramref:`.EnvironmentContext.configure.process_revision_directives`
parameter in an ``env.py`` script. Once constructed, any number
of "rewrites" functions can be associated with it, which will be given
the opportunity to modify the structure without having to have explicit
knowledge of the overall structure.
The function is passed the :class:`.MigrationContext` object and
``revision`` tuple that are passed to the :paramref:`.Environment
Context.configure.process_revision_directives` function normally,
and the third argument is an individual directive of the type
noted in the decorator. The function has the choice of returning
a single op directive, which normally can be the directive that
was actually passed, or a new directive to replace it, or a list
of zero or more directives to replace it.
.. seealso::
:ref:`autogen_rewriter` - usage example
NzOptional[Rewriter]_chainedreturnNonec 6 t j | _ d S N)r
Dispatcherdispatch)selfs o/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/alembic/autogenerate/rewriter.py__init__zRewriter.__init__9 s ))
otherc | j | j }|j | j ||_ |S )a Produce a "chain" of this :class:`.Rewriter` to another.
This allows two rewriters to operate serially on a stream,
e.g.::
writer1 = autogenerate.Rewriter()
writer2 = autogenerate.Rewriter()
@writer1.rewrites(ops.AddColumnOp)
def add_column_nullable(context, revision, op):
op.column.nullable = True
return op
@writer2.rewrites(ops.AddColumnOp)
def add_column_idx(context, revision, op):
idx_op = ops.CreateIndexOp(
"ixc", op.table_name, [op.column.name]
)
return [op, idx_op]
writer = writer1.chain(writer2)
:param other: a :class:`.Rewriter` instance
:return: a new :class:`.Rewriter` that will run the operations
of this writer, then the "other" writer, in succession.
) __class____new____dict__updater )r r$ wrs r! chainzRewriter.chain< s? < ^
#
#DN
3
3
4=))) r# operatorpUnion[Type[AddColumnOp], Type[MigrateOperation], Type[AlterColumnOp], Type[CreateTableOp], Type[ModifyTableOps]]r c 6 | j | S )a Register a function as rewriter for a given type.
The function should receive three arguments, which are
the :class:`.MigrationContext`, a ``revision`` tuple, and
an op directive of the type indicated. E.g.::
@writer1.rewrites(ops.AddColumnOp)
def add_column_nullable(context, revision, op):
op.column.nullable = True
return op
)r dispatch_for)r r, s r! rewriteszRewriter.rewrites_ s , }))(333r# contextr revisionr directiver Iterator[MigrateOperation]c # K | j | }| |j v r|V d S t j |||| g D ]&}|j | g |_ |V 'd S # t
$ r
d }|V Y d S w xY wr )r
_mutationsr to_listunion
ValueError)r r1 r2 r3 _rewriter r_directives r! _rewritezRewriter._rewritew s &
..y99I
y+++ #'< Igx;;R$ $ & &K .9-C-I-I . .K* &%%%%
& & IOOOOOO s A6 6B
B
directivesList[MigrationScript]c t | ||| | j r| ||| d S d S r )process_revision_directivesr r r1 r2 r= s r! __call__zRewriter.__call__ sL
(((JGGG = 9 MM'8Z 8 8 8 8 8 9 9r# r c g }|j D ]V}| ||| }t | dk rt d | |d W||_ g }|j D ]V}| ||| }t | dk rt d | |d W||_ d S )N z5Can only return single object for UpgradeOps traverser z7Can only return single object for DowngradeOps traverse)upgrade_ops_list
_traverse_forlenr9 append upgrade_opsdowngrade_ops_list
downgrade_ops) r r1 r2 r3 rE rI retrJ rK s r! _traverse_scriptzRewriter._traverse_script s $5 , ,K$$Wh DDC3xx1}} K
# #CF + + + + 0 & 9 . .M$$Wh
FFC3xx1}} M
% %c!f - - - -"4 r# r c > | |||j d S r )_traverse_listr
r r1 r2 r3 s r! _traverse_op_containerz Rewriter._traverse_op_container s$
GXy}=====r# c d S r rP s r! _traverse_any_directivez Rewriter._traverse_any_directive s
r# r c t | ||| }|D ]*}| j | } || ||| +|S r )listr< _traverser )r r1 r2 r3 r= traversers r! rF zRewriter._traverse_for sd $--9EEFF
# : :I // ::I IdGXy 9 9 9 9r# c v g }|D ],}| | ||| -||d d < d S r )extendrF )r r1 r2 r= destr3 s r! rO zRewriter._traverse_list sR # J JI KK**7HiHH I I I I
111
r# c 4 | ||| d S r )rO rA s r! r@ z$Rewriter.process_revision_directives s"
GXz:::::r# )r r )r$ r r r )r, r- r r )r1 r r2 r r3 r r r4 )r1 r r2 r r= r> r r )r1 r r2 r r3 r r r )r1 r r2 r r3 r r r )r1 r r2 r r3 r r r )r1 r r2 r r3 r r r )r1 r r2 r r= r r r )__name__
__module__ __qualname____doc__r r rW r __annotations__r" r+ r0 r< rB r/ r
r rM r rQ r rT rF rO r@ rS r# r! r r sx 2 !!I#'H''''* * * *! ! ! !F4 4 4 40& & & &.9 9 9 9 C /005 5 5 1054 CO,,> > > -,> C 011
21
; ; ; ; ; ;r# r N)
__future__r typingr r r r r r r
r alembicr alembic.operationsr
alembic.operations.opsr r r r r r r alembic.runtime.migrationr alembic.script.revisionr r rS r# r! ri s " " " " " " " " " " " " 1222222444444444444777777666666555555222222::::::000000H; H; H; H; H; H; H; H; H; H;r#