File: //home/arjun/projects/env/lib/python3.10/site-packages/dns/__pycache__/versioned.cpython-310.pyc
o
!we�- � @ 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 �F/home/arjun/projects/env/lib/python3.10/site-packages/dns/versioned.pyr s r c
s� e Zd Zg d�ZeZejjddfde e
ej j e
f dejjdede ed ege e f f� fdd �
Z d;d
e e d e e d efd
d�Zd<ded efdd�Zdd� Zdd� Zdd� Zde e d dfdd�Zd e ed ege e f d dfd d �Zd d � Z d!d"� Z d#d$� Z d%d&� Z d'd(� Z d)d*� Z! d<d+e
ej j e
f d,ed ej"jf� fd-d.�
Z#d+e
ej j e
f d dfd/d0�Z$ej%j&dfd+e
ej j e
f d1e
ej%j'e
f d2e
ej%j'e
f d,ed ej(j)f
� fd3d4�
Z*ej%j&dfd+e
ej j e
f d1e
ej%j'e
f d2e
ej%j'e
f d,ed e ej(j) f
� fd5d6�
Z+ej%j&fd+e
ej j e
f d1e
ej%j'e
f d2e
ej%j'e
f d dfd7d8�Z,d+e
ej j e
f dej(j)d dfd9d:�Z-� Z.S )=�Zone)� _versions�_versions_lock�
_write_txn�_write_waiters� _write_event�_pruning_policy�_readersTN�origin�rdclass�
relativize�pruning_policyc st t � �|||� t�� | _t�� | _|du r| 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 s6 |d ur |d ur t d��| j�� |d ur0d }t| j�D ] }|j|kr&|} nq|d u r/td��nL|d urw| jr<tjj }n
| j
d usCJ �| j
}d }t| j�D ] }|j � |�}|rm|�
| jtjj�}|rm|d j|krm|} nqM|d u rvtd��n| jd }t| d|�}| j�|� |W d � S 1 s�w 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 � rdatatype�SOAr* � Transactionr �add) r& r) r* �version�v�oname�n�rds�txnr r r
�readerS sF
��
��
$�z Zone.readerFr c C s� d } | j �/ | jd u r$|| jkr$t| |dd�| _d | _ W d � n t�� }| j�|� W d � n1 s8w Y |�� q| j� � | jS )NT)�make_immutable)
r! r r r8 r �Eventr �append�wait�_setup_version)r& r �eventr r r
�writerw s" �� �)�
,z Zone.writerc C s, t | j�dkr| j�� | _| j�� d S d S )Nr )�lenr �popleftr r# )r& r r r
�!_maybe_wakeup_one_waiter_unlocked� s �z&Zone._maybe_wakeup_one_waiter_unlockedc C � dS )NTr )r& �zoner: r r r
r" � � z Zone._default_pruning_policyc C s� t | j�dks J �t | j�dkrtdd� | jD ��}n| jd j}| jd j|k rJ| �| | jd �rN| j�� | jd j|k rL| �| | jd �s2d S d S d S d S )Nr c s s � | ]}|j jV qd S �N)r: r) )�.0r? r r r
� <genexpr>� s � z0Zone._prune_versions_unlocked.<locals>.<genexpr>r, )rH r r �minr) r rI )r& �
least_keptr r r
�_prune_versions_unlocked� s
�
�
�z Zone._prune_versions_unlocked� max_versionsc sD � dur � dk r t d��� du rdd� }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 rK )NFr �rL �_r r r
�policy� rM z%Zone.set_max_versions.<locals>.policyc s t | j�� kS rN )rH r rV �rT r r
rX � s )r- �set_pruning_policy)r& rT rX r rY r
�set_max_versions� s
zZone.set_max_versionsrX c C sH |du r| j }| j� || _| �� W d � dS 1 s w 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 rS )r& rX r r r
rZ � s
"�zZone.set_pruning_policyc C s@ | j � | j�|� | �� W d � d S 1 sw Y d S rN )r! r �removerS �r& r? r r r
� _end_read� s
"�zZone._end_readc C s | j |ksJ �d | _ | �� d S rN )r rJ r] r r r
�_end_write_unlocked� s zZone._end_write_unlockedc C s6 | j � | �|� W d � d S 1 sw Y d S rN )r! r_ r] r r r
�
_end_write� s "�zZone._end_writec C sF | j �|� | �� |j| _| jd u r|| _|d ur!| �|� d S d S rN )r rC rS r3 r r_ �r& r? r: r r r r
r$ � s
�z Zone._commit_version_unlockedc C s: | j � | �|||� W d � d S 1 sw Y d S rN )r! r$ ra r r r
�_commit_version s "�zZone._commit_versionc C s* t | j�dkr| jd jd }|S d}|S )Nr r, rU )rH r r) )r& r) r r r
�_get_next_version_id s
�zZone._get_next_version_idr1 �createc s |rt �t� �|�S rN )r r � find_node)r& r1 rd r' r r
re s zZone.find_nodec C � t �rN �r )r& r1 r r r
� delete_node rM zZone.delete_node�rdtype�coversc s$ |rt �t� �|||�}tj�|�S rN )r r �
find_rdatasetr0 �rdataset�ImmutableRdataset�r& r1 ri rj rd rl r' r r
rk s zZone.find_rdatasetc s0 |rt �t� �|||�}|d urtj�|�S d S rN )r r r5 r0 rl rm rn r' r r
r5 $ s zZone.get_rdatasetc C rf rN rg )r& r1 ri rj r r r
�delete_rdataset3 s zZone.delete_rdatasetc C rf rN rg )r& r1 r r r r
�replace_rdataset; s zZone.replace_rdataset)NN)F)/r r r
� __slots__�Node� node_factoryr0 �
rdataclass�INr r r1 �Name�str�
RdataClass�boolr �Versionr �intr8 r@ rG rJ r" rS r[ rZ r^ r_ r` r$ rb rc �nodere rh r6 �NONE� RdataTyperl �Rdatasetrk r5 ro rp �
__classcell__r r r' r
r ! s�
�����&���
�$1�
�
���� �����������
�����
����r ) r r r �typingr r r r r �
dns.exceptionr0 �
dns.immutable�dns.name�dns.node�dns.rdataclass� dns.rdataset�
dns.rdatatype�dns.rdtypes.ANY.SOA�dns.zone� exception� DNSExceptionr rL �
VersionedNoderr �ImmutableVersionedNode�
ImmutableNoderz r% �ImmutableVersionr8 r r r r r
�<module> s*