File: //opt/alt/python37/lib/python3.7/site-packages/dns/__pycache__/versioned.cpython-37.pyc
B
��f . � @ s� d Z ddlmZmZmZmZmZ ddlZddlZddl Z
ddl Z
ddl Z
ddl
Z
ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
G dd� de
jj�Ze
jjZe
jjZe
jjZe
jj Z e
jj Z e
jj Z G dd� de
jj �Z dS )zDNS Versioned Zones.� )�Callable�Deque�Optional�Set�UnionNc @ s e Zd ZdZdS )�UseTransactionz-To alter a versioned zone, use a transaction.N)�__name__�
__module__� __qualname__�__doc__� r r �>/opt/alt/python37/lib/python3.7/site-packages/dns/versioned.pyr s r c sb e Zd ZdddddddgZeZejjdd fe e
ej j e
f ejjee ed ege e f d
�� fd d �
Zd=e e e e ed
�dd�Zd>eed�dd�Zdd� Zdd� Zdd� Ze e d d�dd �Ze ed ege e f d d �d d �Zd d!� Z d"d#� Z d$d%� Z d&d'� Z d(d)� Z d*d+� Z!d?e
ej j e
f eej"jd,�� fd-d.�
Z#e
ej j e
f d d/�d0d1�Z$ej%j&dfe
ej j e
f e
ej%j'e
f e
ej%j'e
f eej(j)d2�� fd3d4�
Z*ej%j&dfe
ej j e
f e
ej%j'e
f e
ej%j'e
f ee ej(j) d2�� fd5d6�
Z+ej%j&fe
ej j e
f e
ej%j'e
f e
ej%j'e
f d d7�d8d9�Z,e
ej j e
f ej(j)d d:�d;d<�Z-� Z.S )@�Zone� _versionsZ_versions_lock�
_write_txn�_write_waiters� _write_event�_pruning_policy�_readersTN)�origin�rdclass�
relativize�pruning_policyc st t � �|||� t�� | _t�� | _|dkr6| j| _ n|| _ d| _
d| _ t�� | _ t
� | _| �dt| dd�|� dS )a� Initialize a versioned zone object.
*origin* is the origin of the zone. It may be a ``dns.name.Name``,
a ``str``, or ``None``. If ``None``, then the zone's origin will
be set by the first ``$ORIGIN`` line in a zone file.
*rdclass*, an ``int``, the zone's rdata class; the default is class IN.
*relativize*, a ``bool``, determine's whether domain names are
relativized to the zone's origin. The default is ``True``.
*pruning policy*, a function taking a ``Zone`` and a ``Version`` and returning
a ``bool``, or ``None``. Should the version be pruned? If ``None``,
the default policy, which retains one version is used.
NT)� replacement)�super�__init__� collections�dequer � threading�Lock�
_version_lock�_default_pruning_policyr r r r �setr �_commit_version_unlocked�WritableVersion)�selfr r r r )� __class__r r
r 0 s
z
Zone.__init__)�id�serial�returnc C s" |d k r|d k rt d��| j�� |d k rbd }x"t| j�D ]}|j|kr8|}P q8W |d kr�td��n�|d k r�| jrztjj }n| j
d k s�t �| j
}d }xLt| j�D ]>}|j �
|�}|r�|�| jtjj�}|r�|d j|kr�|}P q�W |d kr�td��n
| jd }t| d|�}| j�|� |S Q R X d S )Nz!cannot specify both id and serialzversion not foundr zserial not found���F)�
ValueErrorr �reversedr r' �KeyErrorr �dns�name�emptyr �AssertionError�nodes�get� get_rdatasetr � rdatatypeZSOAr( � Transactionr �add) r% r' r( �version�vZoname�nZrds�txnr r r
�readerU s<
z Zone.readerF)r r) c C st d }x^| j �F | jd kr:|| jkr:t| |dd�| _d | _P t�� }| j�|� W d Q R X |�� qW | j� � | jS )NT)Zmake_immutable)
r r r r6 r �Eventr �append�waitZ_setup_version)r% r �eventr r r
�writery s
z Zone.writerc C s( t | j�dkr$| j�� | _| j�� d S )Nr )�lenr �popleftr r" )r% r r r
�!_maybe_wakeup_one_waiter_unlocked� s z&Zone._maybe_wakeup_one_waiter_unlockedc C s dS )NTr )r% �zoner8 r r r
r! � s z Zone._default_pruning_policyc C sx t | j�dkst�t | j�dkr6tdd� | jD ��}n | jd j}x0| jd j|k rr| �| | jd �rr| j�� qDW d S )Nr c s s | ]}|j jV qd S )N)r8 r' )�.0r; r r r
� <genexpr>� s z0Zone._prune_versions_unlocked.<locals>.<genexpr>r* )rB r r1 r �minr' r rC )r% Z
least_keptr r r
�_prune_versions_unlocked� s z Zone._prune_versions_unlocked)� max_versionsr) c sD � dk r� dk rt d��� dkr*dd� }n � fdd�}| �|� dS )zYSet a pruning policy that retains up to the specified number
of versions
N� z max versions must be at least 1c S s dS )NFr )rE �_r r r
�policy� s z%Zone.set_max_versions.<locals>.policyc s t | j�� kS )N)rB r )rE rL )rJ r r
rM � s )r+ �set_pruning_policy)r% rJ rM r )rJ r
�set_max_versions� s
zZone.set_max_versions)rM r) c C s2 |dkr| j }| j� || _| �� W dQ R X dS )a� Set the pruning policy for the zone.
The *policy* function takes a `Version` and returns `True` if
the version should be pruned, and `False` otherwise. `None`
may also be specified for policy, in which case the default policy
is used.
Pruning checking proceeds from the least version and the first
time the function returns `False`, the checking stops. I.e. the
retained versions are always a consecutive sequence.
N)r! r r rI )r% rM r r r
rN � s
zZone.set_pruning_policyc C s* | j � | j�|� | �� W d Q R X d S )N)r r �removerI )r% r; r r r
� _end_read� s zZone._end_readc C s | j |kst�d | _ | �� d S )N)r r1 rD )r% r; r r r
�_end_write_unlocked� s zZone._end_write_unlockedc C s | j � | �|� W d Q R X d S )N)r rR )r% r; r r r
�
_end_write� s zZone._end_writec C sB | j �|� | �� |j| _| jd kr,|| _|d k r>| �|� d S )N)r r> rI r2 r rR )r% r; r8 r r r r
r# � s
z Zone._commit_version_unlockedc C s$ | j � | �|||� W d Q R X d S )N)r r# )r% r; r8 r r r r
�_commit_version s zZone._commit_versionc C s( t | j�dkr | jd jd }nd}|S )Nr r* rK )rB r r' )r% r' r r r
�_get_next_version_id s zZone._get_next_version_id)r/ �creater) c s |rt �t� �|�S )N)r r � find_node)r% r/ rV )r& r r
rW s zZone.find_node)r/ r) c C s t �d S )N)r )r% r/ r r r
� delete_node s zZone.delete_node)r/ �rdtype�coversrV r) c s$ |rt �t� �|||�}tj�|�S )N)r r �
find_rdatasetr. �rdataset�ImmutableRdataset)r% r/ rY rZ rV r\ )r& r r
r[ s zZone.find_rdatasetc s4 |rt �t� �|||�}|d k r,tj�|�S d S d S )N)r r r4 r. r\ r] )r% r/ rY rZ rV r\ )r& r r
r4 & s zZone.get_rdataset)r/ rY rZ r) c C s t �d S )N)r )r% r/ rY rZ r r r
�delete_rdataset5 s zZone.delete_rdataset)r/ r r) c C s t �d S )N)r )r% r/ r r r r
�replace_rdataset= s zZone.replace_rdataset)NN)F)F)/r r r
� __slots__�NodeZ node_factoryr. Z
rdataclass�INr r r/ �Name�strZ
RdataClass�boolr �Versionr �intr6 r< rA rD r! rI rO rN rQ rR rS r# rT rU �noderW rX r5 �NONEZ RdataTyper\ ZRdatasetr[ r4 r^ r_ �
__classcell__r r )r& r
r " sN D!"1
0
0
2r ) r �typingr r r r r r r Z
dns.exceptionr. Z
dns.immutableZdns.nameZdns.nodeZdns.rdataclassZ dns.rdatasetZ
dns.rdatatypeZdns.rdtypes.ANY.SOAZdns.zone� exceptionZ DNSExceptionr rE Z
VersionedNodera ZImmutableVersionedNodeZ
ImmutableNoderf r$ ZImmutableVersionr6 r r r r r
�<module> s(