JezK
Edit File: transaction.cpython-39.pyc
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���