a
�M�e�- � @ s� d Z ddlZddlZddlmZmZmZmZmZ ddl 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.� N)�Callable�Deque�Optional�Set�Unionc @ s e Zd ZdZdS )�UseTransactionz-To alter a versioned zone, use a transaction.N)�__name__�
__module__� __qualname__�__doc__� r r �1/usr/lib/python3.9/site-packages/dns/versioned.pyr s r c sX e Zd Zg d�ZeZejjddfe e
ej j e
f ejjee ed ege e f d�� fdd�
Zd7e e e e ed�dd �Zd8eed �d d
�Zdd� Zdd� Zdd� Ze e dd�dd�Ze ed ege e f dd�dd�Zdd� Z d d � Z d d � Z d d!� Z d"d#� Z d$d%� Z!d9e
ej j e
f eej"jd&�� fd'd(�
Z#e
ej j e
f dd)�d*d+�Z$ej%j&d
fe
ej j e
f e
ej%j'e
f e
ej%j'e
f eej(j)d,�� fd-d.�
Z*ej%j&d
fe
ej j e
f e
ej%j'e
f e
ej%j'e
f ee ej(j) d,�� fd/d0�
Z+ej%j&fe
ej j e
f e
ej%j'e
f e
ej%j'e
f dd1�d2d3�Z,e
ej j e
f ej(j)dd4�d5d6�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�� | _|du r6| 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 . s
�z
Zone.__init__)�id�serial�returnc C s, |d ur|d urt d��| j� �� |d urbd }t| j�D ]}|j|kr8|} qPq8|d u r�td��n�|d ur�| jrztjj }n| j
}d }t| j�D ]@}|j � |�}|r�|�
| jtjj�}|r�|d j|kr�|} q�q�|d u r�td��n
| jd }t| d|�}| j�|� |W d � S 1 �s 0 Y d S )Nz!cannot specify both id and serialzversion not foundr zserial not found���F)�
ValueErrorr �reversedr r( �KeyErrorr �dns�name�emptyr �nodes�get� get_rdatasetr � rdatatypeZSOAr) � Transactionr �add) r% r( r) �version�vZoname�nZrds�txnr r r
�readerS s:
z Zone.readerF)r r* c C s� d }| j �\ | jd u rD|| jkrDt| |dd�| _d | _W d � q�t�� }| j�|� W d � n1 sl0 Y |�� q| j� � | jS )NT)Zmake_immutable)
r r r r6 r �Eventr �append�waitZ_setup_version)r% r Zeventr r r
�writerw 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 sb t | j�dkr$tdd� | jD ��}n | jd j}| jd j|k r^| �| | jd �r^| j�� q0d S )Nr c s s | ]}|j jV qd S �N)r8 r( )�.0r; r r r
� <genexpr>� � z0Zone._prune_versions_unlocked.<locals>.<genexpr>r+ )rA r �minr r( r rB )r% Z
least_keptr r r
�_prune_versions_unlocked� s
�z Zone._prune_versions_unlocked)� max_versionsr* c sD � dur� dk rt d��� du r*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 �rD �_r r r
�policy� s z%Zone.set_max_versions.<locals>.policyc s t | j�� kS rE )rA r rM �rK r r
rO � s )r, �set_pruning_policy)r% rK rO r rP r
�set_max_versions� s
zZone.set_max_versions)rO r* c C sF |du r| j }| j� || _| �� W d � n1 s80 Y 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 rJ )r% rO r r r
rQ � s
zZone.set_pruning_policyc C s> | j �$ | j�|� | �� W d � n1 s00 Y d S rE )r r �removerJ �r% r; r r r
� _end_read� s zZone._end_readc C s d | _ | �� d S rE )r rC rT r r r
�_end_write_unlocked� s zZone._end_write_unlockedc C s4 | j � | �|� W d � n1 s&0 Y d S rE )r rV rT r r r
�
_end_write� s zZone._end_writec C sB | j �|� | �� |j| _| jd u r,|| _|d ur>| �|� d S rE )r r> rJ r2 r rV �r% r; r8 r r r r
r# � s
z Zone._commit_version_unlockedc C s8 | j � | �|||� W d � n1 s*0 Y d S rE )r r# rX r r r
�_commit_version s zZone._commit_versionc C s( t | j�dkr | jd jd }nd}|S )Nr r+ rL )rA r r( )r% r( r r r
�_get_next_version_id s zZone._get_next_version_id)r0 �creater* c s |rt �t� �|�S rE )r r � find_node)r% r0 r[ r& r r
r\ s zZone.find_node)r0 r* c C s t �d S rE �r )r% r0 r r r
� delete_node s zZone.delete_node)r0 �rdtype�coversr[ r* c s$ |rt �t� �|||�}tj�|�S rE )r r �
find_rdatasetr/ �rdataset�ImmutableRdataset�r% r0 r_ r` r[ rb r& r r
ra s zZone.find_rdatasetc s4 |rt �t� �|||�}|d ur,tj�|�S d S d S rE )r r r4 r/ rb rc rd r& r r
r4 $ s zZone.get_rdataset)r0 r_ r` r* c C s t �d S rE r] )r% r0 r_ r` r r r
�delete_rdataset3 s zZone.delete_rdataset)r0 r r* c C s t �d S rE r] )r% r0 r r r r
�replace_rdataset; s zZone.replace_rdataset)NN)F)F)/r r r
� __slots__�NodeZ node_factoryr/ Z
rdataclass�INr r r0 �Name�strZ
RdataClass�boolr �Versionr �intr6 r< r@ rC r! rJ rR rQ rU rV rW r# rY rZ Znoder\ r^ r5 ZNONEZ RdataTyperb ZRdatasetra r4 re rf �
__classcell__r r r& r
r ! s�
��& � � $1�
�� ���
��� �r ) r r r �typingr r r r r Z
dns.exceptionr/ Z
dns.immutableZdns.nameZdns.nodeZdns.rdataclassZ dns.rdatasetZ
dns.rdatatypeZdns.rdtypes.ANY.SOAZdns.zone� exceptionZ DNSExceptionr rD Z
VersionedNoderh ZImmutableVersionedNodeZ
ImmutableNoderm r$ ZImmutableVersionr6 r r r r r
�<module> s(