| Server IP : 192.158.238.246 / Your IP : 18.189.143.114 Web Server : LiteSpeed System : Linux uniform.iwebfusion.net 4.18.0-553.27.1.lve.1.el8.x86_64 #1 SMP Wed Nov 20 15:58:00 UTC 2024 x86_64 User : jenniferflocom ( 1321) PHP Version : 8.1.32 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : OFF | Pkexec : OFF Directory : /opt/alt/python37/lib/python3.7/site-packages/alembic/script/__pycache__/ |
Upload File : |
B
�M!V'u � @ s� d dl Z d dlZddlmZ d dlmZ ddlmZ e �d�ZG dd� de �Z
G dd � d e
�Z G d
d � d e
�Z G d d
� d
e
�Z
G dd� de�ZG dd� de�Zdd� ZdS )� N� )�util)�compatz (?:(.+?)@)?(\w+)?((?:\+|-)\d+)c @ s e Zd ZdS )�
RevisionErrorN)�__name__�
__module__� __qualname__� r r �H/opt/alt/python37/lib/python3.7/site-packages/alembic/script/revision.pyr s r c s e Zd Z� fdd�Z� ZS )�RangeNotAncestorErrorc s0 || _ || _tt| ��d|p d|p$df � d S )Nz-Revision %s is not an ancestor of revision %s�base)�lower�upper�superr �__init__)�selfr
r )� __class__r r
r s
z RangeNotAncestorError.__init__)r r r r �
__classcell__r r )r r
r s r c s e Zd Z� fdd�Z� ZS )�
MultipleHeadsc s. || _ || _tt| ��d|d�|�f � d S )Nz6Multiple heads are present for given argument '%s'; %sz, )�heads�argumentr r r �join)r r r )r r r
r s
zMultipleHeads.__init__)r r r r r r r )r r
r s r c s e Zd Z� fdd�Z� ZS )�ResolutionErrorc s t t| ��|� || _d S )N)r r r r )r �messager )r r r
r $ s zResolutionError.__init__)r r r r r r r )r r
r # s r c @ s e Zd ZdZdd� Zejdd� �Zejdd� �Zejdd � �Z ejd
d � �Z
ejd d
� �Z dd� Z d7dd�Z
dd� Zd8dd�Zd9dd�Zdd � Zd d � Zd d � Zd!d"� Zd:d#d$�Zd;d%d&�Zd<d'd(�Zd)d*� Zd+d,� Zd=d-d.�Zd>d/d0�Zd?d1d2�Z d@d3d4�Z dAd5d6�Z dS )B� RevisionMapz�Maintains a map of :class:`.Revision` objects.
:class:`.RevisionMap` is used by :class:`.ScriptDirectory` to maintain
and traverse the collection of :class:`.Script` objects, which are
themselves instances of :class:`.Revision`.
c C s
|| _ dS )a
Construct a new :class:`.RevisionMap`.
:param generator: a zero-arg callable that will generate an iterable
of :class:`.Revision` instances to be used. These are typically
:class:`.Script` subclasses within regular Alembic use.
N)�
_generator)r � generatorr r r
r 2 s zRevisionMap.__init__c C s | j | jS )z�All "head" revisions as strings.
This is normally a tuple of length one,
unless unmerged branches are present.
:return: a tuple of string revision numbers.
)�
_revision_mapr )r r r r
r < s
zRevisionMap.headsc C s | j | jS )z�All "base" revisions as strings.
These are revisions that have a ``down_revision`` of None,
or empty tuple.
:return: a tuple of string revision numbers.
)r �bases)r r r r
r I s
zRevisionMap.basesc C s | j | jS )zeAll "real" head revisions as strings.
:return: a tuple of string revision numbers.
)r � _real_heads)r r r r
r V s zRevisionMap._real_headsc C s | j | jS )zeAll "real" base revisions as strings.
:return: a tuple of string revision numbers.
)r � _real_bases)r r r r
r ` s zRevisionMap._real_basesc
C s� i }t �� }t �� }d| _d| _t� }t� }x�| �� D ]�}|j|krTt�d|j � |||j< |j rn|�
|� |j r~|�
|� |�
|j� |�
|j� |j r�| j|jf7 _|j
r6| j|jf7 _q6W x|D ]}| �||� q�W x|D ]}| �||� q�W xr|�� D ]f}x^|jD ]T}||k�r4t�d||f � || } | �|� ||jk�r\|�|� |�|� �qW �qW d |d< |d< t|�| _t|�| _x |D ]}| j||dd� �q�W |S )z_memoized attribute, initializes the revision map from the
initial collection.
r z%Revision %s is present more than oncez-Revision %s referenced from %s is not presentNF)�map_branch_labels)�sqlautil�
OrderedSetr r �setr �revisionr �warn�
branch_labels�add� dependencies�is_base�
_is_real_base�_map_branch_labels�_add_depends_on�values�_all_down_revisions� add_nextrev�_versioned_down_revisions�discard�tupler r �
_add_branches)
r �map_r r Zhas_branch_labelsZhas_depends_onr% �rev�downrev�
down_revisionr r r
r j sT
zRevisionMap._revision_mapc C sD |j r@x8|jD ].}||kr4td||j|| jf ��|||< qW d S )Nz;Branch name '%s' in revision %s already used by revision %s)r' �_orig_branch_labelsr r% )r r% r5 � branch_labelr r r
r, � s z RevisionMap._map_branch_labelsTc C s� |r| � ||� |jr�|j�|j� x(| j|g|dd�D ]}|j�|j� q8W |}x6|r�|js�|js�|j�|j� |jr�||j }qTP qTW d S )NF)�include_dependencies)r, r' �update�_get_descendant_nodes�_is_real_branch_point�is_merge_pointr8 )r r% r5 r! �node�parentr r r
r4 � s zRevisionMap._add_branchesc s* |j r&t� fdd�t�|j �D ��|_d S )Nc 3 s | ]}� | j V qd S )N)r% )�.0�dep)r5 r r
� <genexpr>� s z.RevisionMap._add_depends_on.<locals>.<genexpr>)r) r3 r �to_tuple�_resolved_dependencies)r r% r5 r )r5 r
r- � s
zRevisionMap._add_depends_onFc s" | j }|s&� j|kr&t�d� j � n |rB� j|krBtd� j ��� |� j< | �� |� | �� |� � jr|| j� jf7 _� j r�| j
� jf7 _
x6� j D ],}||kr�t�d|� f � || � � � q�W � j
r�t� fdd�| jD ��� jf | _� j�r t� fdd�| jD ��� jf | _dS )z�add a single revision to an existing map.
This method is for single-revision use cases, it's not
appropriate for fully populating an entire revision map.
z%Revision %s is present more than oncezrevision %s not in mapz-Revision %s referenced from %s is not presentc 3 s* | ]"}|t � j��� jg�kr|V qd S )N)r$ r/ �unionr% )rB �head)r% r r
rD � s z+RevisionMap.add_revision.<locals>.<genexpr>c 3 s* | ]"}|t � j��� jg�kr|V qd S )N)r$ r1 rG r% )rB rH )r% r r
rD � s N)r r% r r& � Exceptionr4 r- r* r r+ r r/ r0 �
_is_real_headr3 r �is_headr )r r% �_replacer5 r7 r )r% r
� add_revision� s6
zRevisionMap.add_revisionNc C sL | j }|r| �||�}t|�dkr8t||r2d| nd��|rD|d S dS dS )a� Return the current head revision.
If the script directory has multiple heads
due to branching, an error is raised;
:meth:`.ScriptDirectory.get_heads` should be
preferred.
:param branch_label: optional branch name which will limit the
heads considered to those which include that branch_label.
:return: a string revision number.
.. seealso::
:meth:`.ScriptDirectory.get_heads`
� z%s@headrH r N)r �filter_for_lineage�lenr )r r: Z
current_headsr r r
�get_current_head� s z RevisionMap.get_current_headc C s | � | j|�S )N)rO r )r �
identifierr r r
�_get_base_revisions s z RevisionMap._get_base_revisionsc sT t |ttttf�r*t�fdd�|D �d�S ��|�\}� t� �fdd�|D ��S dS )a� Return the :class:`.Revision` instances with the given rev id
or identifiers.
May be given a single identifier, a sequence of identifiers, or the
special symbols "head" or "base". The result is a tuple of one
or more identifiers, or an empty tuple in the case of "base".
In the cases where 'head', 'heads' is requested and the
revision map is empty, returns an empty tuple.
Supports partial identifiers, where the given identifier
is matched against all identifiers that start with the given
characters; if there is exactly one match, that determines the
full revision.
c s g | ]}� � |��qS r )�
get_revisions)rB Zid_elem)r r r
�
<listcomp>+ s z-RevisionMap.get_revisions.<locals>.<listcomp>r c 3 s | ]}�� |� �V qd S )N)�_revision_for_ident)rB �rev_id)r: r r r
rD / s z,RevisionMap.get_revisions.<locals>.<genexpr>N)�
isinstance�listr3 r$ � frozenset�sum�_resolve_revision_number)r �id_� resolved_idr )r: r r
rT s zRevisionMap.get_revisionsc C s> | � |�\}}t|�dkr&t||��n |r2|d }| �||�S )a Return the :class:`.Revision` instance with the given rev id.
If a symbolic name such as "head" or "base" is given, resolves
the identifier into the current head or base revision. If the symbolic
name refers to multiples, :class:`.MultipleHeads` is raised.
Supports partial identifiers, where the given identifier
is matched against all identifiers that start with the given
characters; if there is exactly one match, that determines the
full revision.
rN r )r\ rP r rV )r r] r^ r: r r r
� get_revision2 s zRevisionMap.get_revisionc C sb y| j | }W nJ tk
rX y| �|�}W n" tk
rN td| |��Y nX |S Y nX |S d S )NzNo such branch: '%s')r �KeyErrorrV r )r r: �
branch_revZ
nonbranch_revr r r
�_resolve_branchH s
zRevisionMap._resolve_branchc s� |r| � |�}nd }y| j� }W n� tk
r� � fdd�| jD �}|rV| �||�}|sjtd� � ��nFt|�dkr�td� d�dd� |d d
� D ��f � ��n| j|d }Y nX |r�|d k r�| �|j|j�s�td |j|f � ��|S ) Nc s g | ]}|r|� � �r|�qS r )�
startswith)rB �x)r^ r r
rU ` s z3RevisionMap._revision_for_ident.<locals>.<listcomp>z No such revision or branch '%s'rN z)Multiple revisions start with '%s': %s...z, c s s | ]}d | V qdS )z'%s'Nr )rB �rr r r
rD m s z2RevisionMap._revision_for_ident.<locals>.<genexpr>r � z*Revision %s is not a member of branch '%s') rb r r` rO r rP r �_shares_lineager% )r r^ Z check_branchra r% �revsr )r^ r
rV V s6 z RevisionMap._revision_for_identc sD �� |�\}}g �|r ��|� |r.��|� � ��fdd�|D �S )Nc s g | ]}�j |�� d �r|�qS ))r; )rg )rB �tg)r; r �sharesr r
rU � s z2RevisionMap.filter_for_lineage.<locals>.<listcomp>)r\ �append�extend)r �targetsZ
check_againstr; r] r: r )r; r rj r
rO z s
z RevisionMap.filter_for_lineagec sh |sdS t |t�s � �|�}� fdd�tj|dd�D �}tt� j|g|d���� j |g|d���
|��S )NTc s$ g | ] }t |t�s � �|�n|�qS r )rX �RevisionrV )rB Ztest_against_rev)r r r
rU � s z/RevisionMap._shares_lineage.<locals>.<listcomp>r )�default)r; ) rX rn rV r rE �boolr$ r= rG �_get_ancestor_nodes� intersection)r �targetZtest_against_revsr; r )r r
rg � s
zRevisionMap._shares_lineagec C s� t |tj�r&d|kr&|�dd�\}}nd }| j |dkrZ|rN| �| j|�|fS | j|fS nT|dkr�| �|�}|rz|f|fS d|fS n*|dks�|d kr�d|fS t j
|d d�|fS d S )N�@rN r rH r r )ro ) rX r � string_types�splitr rO r r rQ r rE )r r] r: Z current_headr r r
r\ � s"
z$RevisionMap._resolve_revision_numberc C s t |tj�r t�|�}|s$d S nd S t|�d��}|�d�} |�d�}
|rR| sRdnd} |r�|
rhd|
} n"| r�| �d�r|| } q�d| } nd} |}
n$|
r�d|
}
n| r�d| }
nd}
|} t| j | |
||d ��}| �r\|
r�| �
d
|
| f �}n
| �
| �}| �d��rd}nV| dk�r t |�d }n>t� t |�d dd �}x,|D ]}|j
|| j
k�r8P �q8W d}nd}|�r�||| | d � }|�s�|�r�t |�t|| �k �r�td |t|�f ��nJ|d|| | � }|�s�|�r�t |�t|�| k�r�td |t|�f ��t|�S )
Nrf r rN r z%s@headrH z%s@baser )� inclusive�
implicit_basez%s@%s���z1Relative revision %s didn't produce %d migrations)rX r ru �_relative_destination�match�int�grouprc rY �_iterate_revisionsr_ rP �ranger% �absr �iter)r Z destination�sourceZ
is_upwardsrx rw �assert_relative_lengthr{ Zrelative�symbolr: Zreldelta�from_�to_rh Z
symbol_rev�indexZrange_r r r
�_relative_iterate� sr
z RevisionMap._relative_iteratec C sJ | � ||d|||�}|r |S | � ||d|||�}|r8|S | j||||d�S )au Iterate through script revisions, starting at the given
upper revision identifier and ending at the lower.
The traversal uses strictly the `down_revision`
marker inside each migration script, so
it is a requirement that upper >= lower,
else you'll get nothing back.
The iterator yields :class:`.Revision` objects.
TF)rw rx )r� r~ )r r r
rx rw r� Zrelative_upperZrelative_lowerr r r
�iterate_revisions s z RevisionMap.iterate_revisionsc C s( |rdd� }ndd� }| j ||||d�S )Nc S s | j S )N)� _all_nextrev)r6 r r r
�<lambda>* � z3RevisionMap._get_descendant_nodes.<locals>.<lambda>c S s | j S )N)�nextrev)r6 r r r
r� , r� )r5 �check)�_iterate_related_revisions)r rm r5 r� r; �fnr r r
r= & s
z!RevisionMap._get_descendant_nodesc C s( |rdd� }ndd� }| j ||||d�S )Nc S s | j S )N)r/ )r6 r r r
r� 6 r� z1RevisionMap._get_ancestor_nodes.<locals>.<lambda>c S s | j S )N)r1 )r6 r r r
r� 8 r� )r5 r� )r� )r rm r5 r� r; r� r r r
rq 2 s
z RevisionMap._get_ancestor_nodesc
# s� � d kr| j � t� }t�� }x�|D ]�}|�|� |r:t� }xT|r�|�� } |rV|�| � | |kr`q<|�| � |�� fdd�|| �D �� | V q<W |r"|�|�� |g�r"t
d|j ��q"W d S )Nc 3 s | ]}� | V qd S )Nr )rB rW )r5 r r
rD S s z9RevisionMap._iterate_related_revisions.<locals>.<genexpr>z=Requested revision %s overlaps with other requested revisions) r r$ � collections�dequerk �popr( rl rr �
differencer r% )
r r� rm r5 r� �seen�todors Z
per_targetr6 r )r5 r
r� > s,
z&RevisionMap._iterate_related_revisionsc # sb �� |�}t|tj�o |�d�}t��� |��}|s>|s>t� �t�j |dd��}|rf�� ��
|��} n�|r�|r�t�� |��}
t�� |��} t� } |� |
�� | �}
x6|
D ].}x(|j
D ]}�j| |
kr�P q�W | �|� q�W | �|�} n>|�rt�� �j��} | �|�} n |�s t�� �j��} n|} tdd� |D ���dd� �j | dd�D �����s^t||��t�fdd��fdd��D �D ��� t��fd d�|D ��}d}x���rR|�s�td
��d }|�s�|�t�fd d�� D �d
d� d�� � �|� x`|�rL|�� }��|j � d}|�� ��fdd�t |j
�D �� |�sB||k�rB�q�|V �q�W �q�W � �r^t �dS )z�iterate revisions from upper to lower.
The traversal is depth-first within branches, and breadth-first
across branches as a whole.
z@baseT)r� c s s | ] }|j V qd S )N)r% )rB r6 r r r
rD � s z1RevisionMap._iterate_revisions.<locals>.<genexpr>c s s | ] }|j V qd S )N)r% )rB r6 r r r
rD � s c 3 s, | ]$}|j rt� �|j��d kr|V qdS )rN N)r> rP rr r� )rB r6 )� total_spacer r
rD � s c 3 s | ]}� j | V qd S )N)r )rB r6 )r r r
rD � s c 3 s | ]}|j � kr|V qd S )N)r% )rB re )r� r r
rD � s z5Dependency resolution failed; iteration can't proceedFc 3 s | ]}|j �� �s|V qd S )N)r� rr )rB r6 )r� r r
rD � s c S s | j r
dS dS )Nr rN )�is_branch_point)r6 r r r
r� � r� z0RevisionMap._iterate_revisions.<locals>.<lambda>)�keyc s, g | ]$}�j | � kr|�kr�j | �qS r )r )rB r7 )� branch_todor r� r r
rU � s z2RevisionMap._iterate_revisions.<locals>.<listcomp>N) rT rX r ru �endswithr Z dedupe_tuple�
StopIterationr$ rq rS r= r� r/ r r( rG r rr r r� r� r �
extendleft�sorted�difference_update�popleft�remover% �reversed�AssertionError)r r r
rw rx Zrequested_lowersZlimit_to_lower_branchZuppersZupper_ancestorsZlowersZlower_ancestorsZlower_descendantsZ base_lowersZcandidate_lowersr6 r7 r� Ztotal_space_modifiedr )r� r r� r
r~ Z s~
z RevisionMap._iterate_revisions)T)F)N)N)F)F)FFT)NFT)NFT)F)TF) r r r �__doc__r r Zmemoized_propertyr r r r r r, r4 r- rM rQ rS rT r_ rb rV rO rg r\ r� r� r= rq r� r~ r r r r
r ) s6
;
*
%
N
r c @ s� e Zd ZdZe� Ze� ZdZdZdZ dZ
ddd�Z dd� Z dd� Z
ed d
� �Zed d � �Zed
d� �Zedd� �Zedd� �Zedd� �Zedd� �Zedd� �Zedd� �ZdS ) rn ao Base class for revisioned objects.
The :class:`.Revision` class is the base of the more public-facing
:class:`.Script` object, which represents a migration script.
The mechanics of revision management and traversal are encapsulated
within :class:`.Revision`, while :class:`.Script` applies this logic
to Python files in a version directory.
Nc C s@ || _ t|�| _t|�| _d| _tj|dd�| _t| j�| _ d S )Nr )ro )
r% �tuple_rev_as_scalarr8 r) rF r rE r9 r$ r' )r r% r8 r) r' r r r
r � s
zRevision.__init__c C sV t | j�t | j�g}| jr*|�d| j � | jr@|�d| j � d| jjd�|�f S )Nzdependencies=%rzbranch_labels=%rz%s(%s)z, ) �reprr% r8 r) rk r' r r r )r �argsr r r
�__repr__ s zRevision.__repr__c C s4 | j �|jg�| _ | j|jkr0| j�|jg�| _d S )N)r� rG r% r1 r� )r r% r r r
r0 s zRevision.add_nextrevc C s t j| jdd�| j S )Nr )ro )r rE r8 rF )r r r r
r/ s z Revision._all_down_revisionsc C s t j| jdd�S )Nr )ro )r rE r8 )r r r r
r1 ! s z"Revision._versioned_down_revisionsc C s t | j� S )a Return True if this :class:`.Revision` is a 'head' revision.
This is determined based on whether any other :class:`.Script`
within the :class:`.ScriptDirectory` refers to this
:class:`.Script`. Multiple heads can be present.
)rp r� )r r r r
rK % s zRevision.is_headc C s t | j� S )N)rp r� )r r r r
rJ 0 s zRevision._is_real_headc C s
| j dkS )z<Return True if this :class:`.Revision` is a 'base' revision.N)r8 )r r r r
r* 4 s zRevision.is_basec C s | j dko| jdkS )zrReturn True if this :class:`.Revision` is a "real" base revision,
e.g. that it has no dependencies either.N)r8 r) )r r r r
r+ : s zRevision._is_real_basec C s t | j�dkS )a+ Return True if this :class:`.Script` is a branch point.
A branchpoint is defined as a :class:`.Script` which is referred
to by more than one succeeding :class:`.Script`, that is more
than one :class:`.Script` has a `down_revision` identifier pointing
here.
rN )rP r� )r r r r
r� D s
zRevision.is_branch_pointc C s t | j�dkS )zzReturn True if this :class:`.Script` is a 'real' branch point,
taking into account dependencies as well.
rN )rP r� )r r r r
r> P s z Revision._is_real_branch_pointc C s t | j�dkS )z6Return True if this :class:`.Script` is a merge point.rN )rP r1 )r r r r
r? X s zRevision.is_merge_point)NN)r r r r� rZ r� r� r% r8 r) r' r r� r0 �propertyr/ r1 rK rJ r* r+ r� r> r? r r r r
rn � s&
rn c C s$ | sd S t | �dkr | d S | S d S )NrN r )rP )r6 r r r
r� _ s
r� )�rer� � r Z
sqlalchemyr" r �compilerz rI r r r r �objectr rn r� r r r r
�<module> s$
3