Edit File: revision.cpython-37.pyc
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