a
�M�elX � @ s. d dl Z d dlmZmZmZmZmZmZmZ d dl Z
d dl Z
d dl Z
d dl
Z
d dlZ
d dlZ
d dlZ
d dlZ
d dlZ
G dd� d�ZG dd� de
jj�ZG dd� de
jj�ZG d d
� d
e
jj�Zd d � Zd
d� Zede
jj e
j j gdf Z ede
jj e
j j!e
j j!gdf Z"ede
jj gdf Z#G dd� d�Z$dS )� N)�Any�Callable�Iterator�List�Optional�Tuple�Unionc @ s| e Zd Zdd�dd�Zdedd�dd�Zeeej j
eeej j
f d�d d
�Z ej j
d�d d �Zeej j
d�d
d�ZdS )�TransactionManager� Transaction��returnc C s t �dS )z Begin a read-only transaction.N��NotImplementedError��self� r �3/usr/lib/python3.9/site-packages/dns/transaction.py�reader s zTransactionManager.readerF)� replacementr c C s t �dS )a Begin a writable transaction.
*replacement*, a ``bool``. If `True`, the content of the
transaction completely replaces any prior content. If False,
the default, then the content of the transaction updates the
existing content.
Nr
)r r r r r �writer s zTransactionManager.writerc C s t �dS )aq Returns a tuple
(absolute_origin, relativize, effective_origin)
giving the absolute name of the default origin for any
relative domain names, the "effective origin", and whether
names should be relativized. The "effective origin" is the
absolute origin if relativize is False, and the empty name if
relativize is true. (The effective origin is provided even
though it can be computed from the absolute_origin and
relativize setting because it avoids a lot of code
duplication.)
If the returned names are `None`, then no origin information is
available.
This information is used by code working with transactions to
allow it to coordinate relativization. The transaction code
itself takes what it gets (i.e. does not change name
relativity).
Nr
r r r r �origin_information s z%TransactionManager.origin_informationc C s t �dS )z%The class of the transaction manager.Nr
r r r r � get_class; s z TransactionManager.get_classc C s | � � \}}}|r|S dS dS )z#Origin to use in from_wire() calls.N)r )r Zabsolute_originZ
relativize�_r r r �from_wire_origin? s z#TransactionManager.from_wire_originN)F)�__name__�
__module__� __qualname__r �boolr r r �dns�name�Namer Z
rdataclassZ
RdataClassr r r r r r r s � r c @ s e Zd ZdZdS )�DeleteNotExactz>Existing data did not match data specified by an exact delete.N�r r r �__doc__r r r r r! H s r! c @ s e Zd ZdZdS )�ReadOnlyz*Tried to write to a read-only transaction.Nr" r r r r r$ L s r$ c @ s e Zd ZdZdS )� AlreadyEndedz*Tried to use an already-ended transaction.Nr" r r r r r% P s r% c C s& | d u st | tjj�r| S tj�| �S �N)�
isinstancer �rdataset�ImmutableRdataset)r( r r r �_ensure_immutable_rdatasetT s r* c C s | d u s| � � r| S tj�| �S r& )Z is_immutabler �nodeZ
ImmutableNode)r+ r r r �_ensure_immutable_nodeZ s r, r
c @ sd e Zd Zd`eeed�dd�Zejjfe e
ej j e
f e
ejje
f e
ejje
f ejjd�dd�Zej j e ejj d�d d
�Zd d �d
d�Zed d�dd�Zed d�dd�Zed d�dd�Zed d�dd�Ze
ej j e
f ed�dd�Zddej j fe eej j d d �d d �Z d d � Z ed �d!d"�Z d d �d#d$�Z!d d �d%d&�Z"e#d d'�d(d)�Z$e%d d'�d*d+�Z&e'd d'�d,d-�Z(e)e*ej j ejjf d �d.d/�Z+e)ej j d �d0d1�Z,d2d3� Z-d4d5� Z.d6d7� Z/d8d9� Z0d:d;� Z1d<d=� Z2d>d?� Z3d@dA� Z4dBdC� Z5dDdE� Z6dFdG� Z7dHdI� Z8dJdK� Z9dLdM� Z:dNdO� Z;dPdQ� Z<dRdS� Z=dTdU� Z>dVdW� Z?dXdY� Z@dZd[� ZAd\d]� ZBd^d_� ZCd S )ar
F)�managerr � read_onlyc C s. || _ || _|| _d| _g | _g | _g | _d S �NF)r- r r. �_ended�_check_put_rdataset�_check_delete_rdataset�_check_delete_name)r r- r r. r r r �__init__k s zTransaction.__init__)r �rdtype�coversr c C sR | � � t|t�r tj�|d�}tjj�|�}tjj�|�}| � |||�}t
|�S )z�Return the rdataset associated with *name*, *rdtype*, and *covers*,
or `None` if not found.
Note that the returned rdataset is immutable.
N) � _check_endedr' �strr r � from_text� rdatatype� RdataType�make�
_get_rdatasetr* )r r r5 r6 r( r r r �get� s
zTransaction.get)r r c C s t | �|��S )z[Return the node at *name*, if any.
Returns an immutable node or ``None``.
)r, � _get_node�r r r r r �get_node� s zTransaction.get_nodeNr c C s | j r
t�d S r& )r. r$ r r r r �_check_read_only� s z Transaction._check_read_only)�argsr c G s | � � | �� | �d|� dS )z�Add records.
The arguments may be:
- rrset
- name, rdataset...
- name, ttl, rdata...
FN�r7 rB �_add�r rC r r r �add� s zTransaction.addc G s | � � | �� | �d|� dS )a� Replace the existing rdataset at the name with the specified
rdataset, or add the specified rdataset if there was no existing
rdataset.
The arguments may be:
- rrset
- name, rdataset...
- name, ttl, rdata...
Note that if you want to replace the entire node, you should do
a delete of the name followed by one or more calls to add() or
replace().
TNrD rF r r r �replace� s zTransaction.replacec G s | � � | �� | �d|� dS )a Delete records.
It is not an error if some of the records are not in the existing
set.
The arguments may be:
- rrset
- name
- name, rdatatype, [covers]
- name, rdataset...
- name, rdata...
FN�r7 rB �_deleterF r r r �delete� s zTransaction.deletec G s | � � | �� | �d|� dS )a4 Delete records.
The arguments may be:
- rrset
- name
- name, rdatatype, [covers]
- name, rdataset...
- name, rdata...
Raises dns.transaction.DeleteNotExact if some of the records
are not in the existing set.
TNrI rF r r r � delete_exact� s zTransaction.delete_exactc C s* | � � t|t�r tj�|d�}| �|�S )z Does the specified name exist?N)r7 r' r8 r r r9 � _name_existsr@ r r r � name_exists� s
zTransaction.name_exists� T)�value�relativer r c C s� | � � |dk rtd��t|t�r0tj�|d�}| �|tjj tjj
�}|du sZt |�dkr^t �|rztj
�|d j
�| }n tj
�|�}|j}|dkr�d}|d j|d�}tj�|j|�}| �||� dS )a� Update the serial number.
*value*, an `int`, is an increment if *relative* is `True`, or the
actual value to set if *relative* is `False`.
Raises `KeyError` if there is no SOA rdataset at *name*.
Raises `ValueError` if *value* is negative or if the increment is
so large that it would cause the new serial to be less than the
prior value.
r z negative update_serial() valueNrO )�serial)r7 �
ValueErrorr' r8 r r r9 r= r: �SOA�NONE�len�KeyErrorrR ZSerialrP rH r( �
from_rdata�ttl)r rP rQ r r( rR �rdataZ new_rdatasetr r r �
update_serial� s"
zTransaction.update_serialc C s | � � | �� S r& �r7 �_iterate_rdatasetsr r r r �__iter__ s zTransaction.__iter__c C s | � � | �� S )a Has this transaction changed anything?
For read-only transactions, the result is always `False`.
For writable transactions, the result is `True` if at some time
during the life of the transaction, the content was changed.
)r7 �_changedr r r r �changed s zTransaction.changedc C s | � d� dS )a� Commit the transaction.
Normally transactions are used as context managers and commit
or rollback automatically, but it may be done explicitly if needed.
A ``dns.transaction.Ended`` exception will be raised if you try
to use a transaction after it has been committed or rolled back.
Raises an exception if the commit fails (in which case the transaction
is also rolled back.
TN��_endr r r r �commit) s zTransaction.commitc C s | � d� dS )av Rollback the transaction.
Normally transactions are used as context managers and commit
or rollback automatically, but it may be done explicitly if needed.
A ``dns.transaction.AlreadyEnded`` exception will be raised if you try
to use a transaction after it has been committed or rolled back.
Rollback cannot otherwise fail.
FNra r r r r �rollback6 s
zTransaction.rollback)�checkr c C s | j �|� dS )a� Call *check* before putting (storing) an rdataset.
The function is called with the transaction, the name, and the rdataset.
The check function may safely make non-mutating transaction method
calls, but behavior is undefined if mutating transaction methods are
called. The check function should raise an exception if it objects to
the put, and otherwise should return ``None``.
N)r1 �append�r re r r r �check_put_rdatasetB s
z Transaction.check_put_rdatasetc C s | j �|� dS )a� Call *check* before deleting an rdataset.
The function is called with the transaction, the name, the rdatatype,
and the covered rdatatype.
The check function may safely make non-mutating transaction method
calls, but behavior is undefined if mutating transaction methods are
called. The check function should raise an exception if it objects to
the put, and otherwise should return ``None``.
N)r2 rf rg r r r �check_delete_rdatasetN s z!Transaction.check_delete_rdatasetc C s | j �|� dS )a� Call *check* before putting (storing) an rdataset.
The function is called with the transaction and the name.
The check function may safely make non-mutating transaction method
calls, but behavior is undefined if mutating transaction methods are
called. The check function should raise an exception if it objects to
the put, and otherwise should return ``None``.
N)r3 rf rg r r r �check_delete_name[ s
z Transaction.check_delete_namec C s | � � | �� S )a@ Iterate all the rdatasets in the transaction, returning
(`dns.name.Name`, `dns.rdataset.Rdataset`) tuples.
Note that as is usual with python iterators, adding or removing items
while iterating will invalidate the iterator and may raise `RuntimeError`
or fail to iterate over all entries.r\ r r r r �iterate_rdatasetsg s z Transaction.iterate_rdatasetsc C s | � � | �� S )z�Iterate all the names in the transaction.
Note that as is usual with python iterators, adding or removing names
while iterating will invalidate the iterator and may raise `RuntimeError`
or fail to iterate over all entries.)r7 �_iterate_namesr r r r �
iterate_namess s zTransaction.iterate_namesc C s t |�dkrtd|� ���d S )Nr zextra parameters to )rV � TypeError)r �methodrC r r r �_raise_if_not_empty� s z Transaction._raise_if_not_emptyc C s� z�|� � }t|tjj�r"|�� }n�t|tjj�r6|}nv|r@d}n@t|t�rj|}|tj j
krxt |� d���nt |� d���|� � }t|tj
j�r�tj�||�}nt |� d���|W S ty� |r�Y d S t |� d���Y n0 d S )Nr z: TTL value too bigz: expected a TTLz: expected an Rdataz: expected more arguments)�popleftr' r �rrset�RRset� to_rdatasetr( �Rdataset�intrY ZMAX_TTLrS rn rZ ZRdatarX �
IndexError)r ro ZdeletingrC �argr( rY r r r �_rdataset_from_args� s,
z Transaction._rdataset_from_argsc C s� �zVt �|�}|rd}nd}|�� }t|t�r<tj�|d �}t|tjj�r^|}| � |d|�}n0t|tj
j �r�|}|j}|� � }nt
|� d���|j| j�� kr�t|� d���|jtjjkr�| �� \}}} || kr�t|� d���| �||� |�sJ| �||j|j�}
|
d u�rJt|
tjj��r@tj�|
j|
j|
j�} | � |
� | }
|
� |�}| � ||� W n" t �yz t
d|� ���Y n0 d S )Nz replace()zadd()F�/ requires a name or RRset as the first argument� has objects of wrong RdataClassz has non-origin SOA�not enough parameters to ) � collections�dequerq r' r8 r r r9 r ry rr rs rt rn �rdclassr- r rS r5 r: rT �_origin_informationrp r= r6 r( r) ru �update�union�_checked_put_rdatasetrw ) r rH rC ro rx r r( rr r �origin�existingZtrdsr r r rE � sL
�
�
zTransaction._addc C sP �z&t �|�}|rd}nd}|�� }t|t�r<tj�|d �}t|tjj��r |}t |�dkr�t|d t
�sxt|d t�r�tj j �
|�� �}t |�dkr�tj j �
|�� �}ntj j}| �||� | �|||�}|d u r�|r�t|� d���n| �|||� W d S | �|d|�} n*t|tjj��r(|} | j}nt|� d���| �||� | �r�| j| j�� k�rht|� d���| �|| j| j �}|d u�r�|�r�|� | �}
|
| k�r�t|� d���|� | �} t | �dk�r�| � || j| j � n | � || � n|�r&t|� d���n*|�r | �!|��s t|� d ���| �"|� W n" t#�yJ td
|� ���Y n0 d S ) Nzdelete_exact()zdelete()r z: missing rdatasetTrz r{ z: missing rdatasz: name not knownr| )$r} r~ rq r' r8 r r r9 r rV rv r: r; r< rU rp r= r! �_delete_rdatasetry rr rs rn r r- r rS r5 r6 � intersection�
difference�_checked_delete_rdatasetr� rM �_checked_delete_namerw ) r �exactrC ro rx r r5 r6 r� r( r� r r r rJ � sp
� � �
�zTransaction._deletec C s | j r
t�d S r& )r0 r% r r r r r7 s zTransaction._check_endedc C s4 | � � | jrt�z| �|� W d| _nd| _0 d S )NT)r7 r0 r% �_end_transaction�r rc r r r rb s zTransaction._endc C s( | j D ]}|| ||� q| �||� d S r& )r1 �
_put_rdataset)r r r( re r r r r� s
z!Transaction._checked_put_rdatasetc C s, | j D ]}|| |||� q| �|||� d S r& )r2 r� )r r r5 r6 re r r r r� s
z$Transaction._checked_delete_rdatasetc C s$ | j D ]}|| |� q| �|� d S r& )r3 � _delete_name)r r re r r r r� % s
z Transaction._checked_delete_namec C s | S r& r r r r r � __enter__. s zTransaction.__enter__c C s$ | j s |d u r| �� n| �� dS r/ )r0 rc rd )r �exc_typeZexc_valZexc_tbr r r �__exit__1 s
zTransaction.__exit__c C s t �dS )zlReturn the rdataset associated with *name*, *rdtype*, and *covers*,
or `None` if not found.
Nr
�r r r5 r6 r r r r= > s zTransaction._get_rdatasetc C s t �dS )zStore the rdataset.Nr
)r r r( r r r r� D s zTransaction._put_rdatasetc C s t �dS )zhDelete all data associated with *name*.
It is not an error if the name does not exist.
Nr
r@ r r r r� H s zTransaction._delete_namec C s t �dS )z�Delete all data associated with *name*, *rdtype*, and *covers*.
It is not an error if the rdataset does not exist.
Nr
r� r r r r� O s z Transaction._delete_rdatasetc C s t �dS )z2Does name exist?
Returns a bool.
Nr
r@ r r r rM V s zTransaction._name_existsc C s t �dS )z&Has this transaction changed anything?Nr
r r r r r_ ] s zTransaction._changedc C s t �dS )z�End the transaction.
*commit*, a bool. If ``True``, commit the transaction, otherwise
roll it back.
If committing and the commit fails, then roll back and raise an
exception.
Nr
r� r r r r� a s z Transaction._end_transactionc C s t �dS )z�Set the origin.
This method is called when reading a possibly relativized
source, and an origin setting operation occurs (e.g. $ORIGIN
in a zone file).
Nr
)r r� r r r � _set_originl s zTransaction._set_originc C s t �dS )z7Return an iterator that yields (name, rdataset) tuples.Nr
r r r r r] u s z Transaction._iterate_rdatasetsc C s t �dS )z&Return an iterator that yields a name.Nr
r r r r rl y s zTransaction._iterate_namesc C s t �dS )zPReturn the node at *name*, if any.
Returns a node or ``None``.
Nr
r@ r r r r? } s zTransaction._get_nodec C s
| j �� S r& )r- r r r r r r� � s z Transaction._origin_information)FF)Dr r r r r r4 r r: rU r r r r r8 r; r( ru r> r+ ZNoderA rB r rG rH rK rL rN �emptyrv r[ r^ r` rc rd �CheckPutRdatasetTyperh �CheckDeleteRdatasetTyperi �CheckDeleteNameTyperj r r rk rm rp ry rE rJ r7 rb r� r� r� r� r� r= r� r� r� rM r_ r� r� r] rl r? r� r r r r r
j s~ �� �� �� $
�
,@
)%r} �typingr r r r r r r Z
dns.exceptionr Zdns.nameZdns.nodeZdns.rdataclassZ dns.rdatasetZ
dns.rdatatypeZ dns.rrsetZ
dns.serialZdns.ttlr � exceptionZ DNSExceptionr! r$ r% r* r, r r r( ru r� r: r; r� r� r
r r r r �<module> s4 $7���