a
Me- @ s d Z ddlZddlZddlmZmZmZmZmZ ddl Z
ddl Z
ddl Z
ddl
Z
ddlZ
ddlZ
ddlZ
ddlZ
ddlZ
G dd de
jjZe
jjZe
jjZe
jjZe
jj Z e
jj Z e
jj Z G dd de
jj Z dS )zDNS Versioned Zones. N)CallableDequeOptionalSetUnionc @ 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 dZeZejjddfe e
ej j e
f ejjee ed ege e f d fdd
Zd7e e e e eddd Zd8eed d d
Zdd Zdd Zdd Ze e ddddZe ed ege e f ddddZdd 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 dd1d2d3Z,e
ej j e
f ej(j)dd4d5d6Z- Z.S ):Zone) _versionsZ_versions_lock
_write_txn_write_waiters _write_event_pruning_policy_readersTN)originrdclass
relativizepruning_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__ collectionsdequer threadingLock
_version_lock_default_pruning_policyr r r r setr _commit_version_unlockedWritableVersion)selfr r r r __class__r r
r . s
z
Zone.__init__)idserialreturnc C s, |d ur|d urt d| j |d urbd }t| jD ]}|j|kr8|} qPq8|d u rtdn|d ur| jrztjj }n| j
}d }t| jD ]@}|j |}|r|
| jtjj}|r|d j|kr|} qq|d u rtdn
| jd }t| d|}| j| |W d S 1 s 0 Y d S )Nz!cannot specify both id and serialzversion not foundr zserial not foundF)
ValueErrorr reversedr r( KeyErrorr dnsnameemptyr nodesget get_rdatasetr rdatatypeZSOAr) Transactionr add) r% r( r) versionvZonamenZrdstxnr r r
readerS s:
z Zone.readerF)r r* c C s d }| j \ | jd u rD|| jkrDt| |dd| _d | _W d qt }| j| W d n1 sl0 Y | q| j | jS )NT)Zmake_immutable)
r r r r6 r Eventr appendwaitZ_setup_version)r% r Zeventr r r
writerw s *
z Zone.writerc C s( t | jdkr$| 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 | jdkr$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
z0Zone._prune_versions_unlocked..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..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 | jdkr | 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 rdtypecoversr[ r* c s$ |rt t |||}tj|S rE )r r
find_rdatasetr/ rdatasetImmutableRdatasetr% 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
rdataclassINr r r0 NamestrZ
RdataClassboolr 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
s(