관리-도구
편집 파일: versioned.cpython-38.pyc
U /?�f�- � @ 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 �;/opt/hc_python/lib/python3.8/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 . s �z Zone.__init__)�id�serial�returnc C s, |d k r|d k rt d��| j�� |d k rbd }t| j�D ]}|j|kr8|} qPq8|d kr�td��n�|d k r�| jrztjj }n| j d k s�t �| j }d }t| j�D ]@}|j � |�}|r�|�| jtjj�}|r�|d j|kr�|} q�q�|d kr�td��n | jd }t| d|�}| j�|� |W 5 Q R � 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 �readerS s< z Zone.readerF)r r* c C sz d }| j �P | jd krB|| jkrBt| |dd�| _d | _W 5 Q R � qjt�� }| j�|� W 5 Q R X |�� q| j� � | jS )NT)Zmake_immutable) r r r r7 r �Eventr �append�waitZ_setup_version)r% r �eventr 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% �zoner9 r r r r! � s z Zone._default_pruning_policyc C st t | j�dkst�t | j�dkr6tdd� | jD ��}n | jd j}| jd j|k rp| �| | jd �rp| j�� qBd S )Nr c s s | ]}|j jV qd S �N)r9 r( )�.0r< r r r � <genexpr>� s z0Zone._prune_versions_unlocked.<locals>.<genexpr>r+ )rC r r2 r �minr( r rD )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 �rF �_r r r �policy� s z%Zone.set_max_versions.<locals>.policyc s t | j�� kS rG )rC r rN �rL r r rP � s )r, �set_pruning_policy)r% rL rP r rQ r �set_max_versions� s zZone.set_max_versions)rP r* c C s2 |dkr| j }| j� || _| �� W 5 Q 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 rK )r% rP r r r rR � s zZone.set_pruning_policyc C s* | j � | j�|� | �� W 5 Q R X d S rG )r r �removerK �r% r< r r r � _end_read� s zZone._end_readc C s | j |kst�d | _ | �� d S rG )r r2 rE rU r r r �_end_write_unlocked� s zZone._end_write_unlockedc C s | j � | �|� W 5 Q R X d S rG )r rW rU r r r � _end_write� s zZone._end_writec C sB | j �|� | �� |j| _| jd kr,|| _|d k r>| �|� d S rG )r r? rK r3 r rW �r% r< r9 r r r r r# � s z Zone._commit_version_unlockedc C s$ | j � | �|||� W 5 Q R X d S rG )r r# rY r r r �_commit_version s zZone._commit_versionc C s( t | j�dkr | jd jd }nd}|S )Nr r+ rM )rC r r( )r% r( r r r �_get_next_version_id s zZone._get_next_version_id)r0 �creater* c s |rt �t� �|�S rG )r r � find_node)r% r0 r\ r& r r r] s zZone.find_node)r0 r* c C s t �d S rG �r )r% r0 r r r � delete_node s zZone.delete_node)r0 �rdtype�coversr\ r* c s$ |rt �t� �|||�}tj�|�S rG )r r � find_rdatasetr/ �rdataset�ImmutableRdataset�r% r0 r` ra r\ rc r& r r rb s zZone.find_rdatasetc s4 |rt �t� �|||�}|d k r,tj�|�S d S d S rG )r r r5 r/ rc rd re r& r r r5 $ s zZone.get_rdataset)r0 r` ra r* c C s t �d S rG r^ )r% r0 r` ra r r r �delete_rdataset3 s zZone.delete_rdataset)r0 r r* c C s t �d S rG 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 �intr7 r= rB rE r! rK rS rR rV rW rX r# rZ r[ �noder] r_ r6 �NONEZ RdataTyperc ZRdatasetrb r5 rf rg � __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 rF Z VersionedNoderi ZImmutableVersionedNodeZ ImmutableNodern r$ ZImmutableVersionr7 r r r r r �<module> s(