File: //home/arjun/projects/env/lib/python3.10/site-packages/dns/__pycache__/node.cpython-310.pyc
o
!wex1 � @ s� d Z ddlZddlZddlmZmZmZ ddlZddl Zddl
Zddl Zddl Zddl
ZddlZejjhZejjejjejjhZdd� ZejG dd� dej��ZG dd � d �ZejjG d
d � d e��Z dS ) z)DNS nodes. A node is a set of rdatasets.� N)�Any�Dict�Optionalc C s || v p
|t jjko
|| v S �N)�dns� rdatatype�RRSIG)�rdtypes�rdtype�covers� r �A/home/arjun/projects/env/lib/python3.10/site-packages/dns/node.py� _matches_type_or_its_signature, s r c @ sX e Zd ZdZdZdZdZedej j
dej j
dd fdd ��Z ed
ej j
dd fd d ��Zd
S )�NodeKindzRdatasets in nodesr � � r
r �returnc C s* t t||�r tjS t t||�rtjS tjS r )r � _cname_typesr �CNAME�_neutral_types�NEUTRAL�REGULAR)�clsr
r r r r
�classify8 s
zNodeKind.classify�rdatasetc C s | � |j|j�S r )r r
r )r r r r r
�classify_rdatasetC s zNodeKind.classify_rdatasetN)�__name__�
__module__� __qualname__�__doc__r r r � classmethodr r � RdataTyper r �Rdatasetr r r r r
r 0 s �� �
r c @ sD e Zd ZdZdgZdd� Zdejjde e
e f de
fdd �Z d
d � Z
d d
� Zdd� Zdd� Zdd� Zdd� Zejjdfdejjdejjdejjdedejjf
dd �Zejjdfdejjdejjdejjdede ejj f
d d �Z ejjfdejjdejjdejjdd fd d!�Z d"ejjdd fd#d$�Z de fd%d&�Z!defd'd(�Z"d S ))�Nodea� A Node is a set of rdatasets.
A node is either a CNAME node or an "other data" node. A CNAME
node contains only CNAME, KEY, NSEC, and NSEC3 rdatasets along with their
covering RRSIG rdatasets. An "other data" node contains any
rdataset other than a CNAME or RRSIG(CNAME) rdataset. When
changes are made to a node, the CNAME or "other data" state is
always consistent with the update, i.e. the most recent change
wins. For example, if you have a node which contains a CNAME
rdataset, and then add an MX rdataset to it, then the CNAME
rdataset will be deleted. Likewise if you have a node containing
an MX rdataset and add a CNAME rdataset, the MX rdataset will be
deleted.
� rdatasetsc C s
g | _ d S r �r$ ��selfr r r
�__init__[ s
z
Node.__init__�name�kwr c K sR t �� }| jD ]}t|�dkr |�|j|fi |��� |�d� q|�� dd� S )a" Convert a node to text format.
Each rdataset at the node is printed. Any keyword arguments
to this method are passed on to the rdataset's to_text() method.
*name*, a ``dns.name.Name``, the owner name of the
rdatasets.
Returns a ``str``.
r �
N���)�io�StringIOr$ �len�write�to_text�getvalue)r' r) r* �s�rdsr r r
r1 _ s
�z Node.to_textc C s dt t| �� d S )Nz
<DNS node �>)�str�idr& r r r
�__repr__s s z
Node.__repr__c C s< | j D ]
}||j vr
dS q|j D ]
}|| j vr dS qdS )NFTr% )r' �other�rdr r r
�__eq__v s
�
�z Node.__eq__c C s | � |� S r )r; )r' r9 r r r
�__ne__� s z Node.__ne__c C �
t | j�S r )r/ r$ r& r r r
�__len__� �
z Node.__len__c C r= r )�iterr$ r& r r r
�__iter__� r? z
Node.__iter__c C sb t | j�dkr)t�|�}|tjkrdd� | jD �| _n|tjkr)dd� | jD �| _| j�|� dS )a� Append rdataset to the node with special handling for CNAME and
other data conditions.
Specifically, if the rdataset being appended has ``NodeKind.CNAME``,
then all rdatasets other than KEY, NSEC, NSEC3, and their covering
RRSIGs are deleted. If the rdataset being appended has
``NodeKind.REGULAR`` then CNAME and RRSIG(CNAME) are deleted.
r c S � g | ] }t �|�t jkr|�qS r )r r r ��.0r4 r r r
�
<listcomp>� �
�z)Node._append_rdataset.<locals>.<listcomp>c S rB r )r r r rC r r r
rE � rF N)r/ r$ r r r r �append�r' r �kindr r r
�_append_rdataset� s
�
�zNode._append_rdatasetF�rdclassr
r �createc C sH | j D ]
}|�|||�r| S q|st�tj�|||�}| �|� |S )a* Find an rdataset matching the specified properties in the
current node.
*rdclass*, a ``dns.rdataclass.RdataClass``, the class of the rdataset.
*rdtype*, a ``dns.rdatatype.RdataType``, the type of the rdataset.
*covers*, a ``dns.rdatatype.RdataType``, the covered type.
Usually this value is ``dns.rdatatype.NONE``, but if the
rdtype is ``dns.rdatatype.SIG`` or ``dns.rdatatype.RRSIG``,
then the covers value will be the rdata type the SIG/RRSIG
covers. The library treats the SIG and RRSIG types as if they
were a family of types, e.g. RRSIG(A), RRSIG(NS), RRSIG(SOA).
This makes RRSIGs much easier to work with than if RRSIGs
covering different rdata types were aggregated into a single
RRSIG rdataset.
*create*, a ``bool``. If True, create the rdataset if it is not found.
Raises ``KeyError`` if an rdataset of the desired type and class does
not exist and *create* is not ``True``.
Returns a ``dns.rdataset.Rdataset``.
)r$ �match�KeyErrorr r r" rJ �r' rK r
r rL r4 r r r
�
find_rdataset� s
�
zNode.find_rdatasetc C s0 z | � ||||�}W |S ty d}Y |S w )a� Get an rdataset matching the specified properties in the
current node.
None is returned if an rdataset of the specified type and
class does not exist and *create* is not ``True``.
*rdclass*, an ``int``, the class of the rdataset.
*rdtype*, an ``int``, the type of the rdataset.
*covers*, an ``int``, the covered type. Usually this value is
dns.rdatatype.NONE, but if the rdtype is dns.rdatatype.SIG or
dns.rdatatype.RRSIG, then the covers value will be the rdata
type the SIG/RRSIG covers. The library treats the SIG and RRSIG
types as if they were a family of
types, e.g. RRSIG(A), RRSIG(NS), RRSIG(SOA). This makes RRSIGs much
easier to work with than if RRSIGs covering different rdata
types were aggregated into a single RRSIG rdataset.
*create*, a ``bool``. If True, create the rdataset if it is not found.
Returns a ``dns.rdataset.Rdataset`` or ``None``.
N)rP rN rO r r r
� get_rdataset� s ��zNode.get_rdatasetNc C s* | � |||�}|dur| j�|� dS dS )aD Delete the rdataset matching the specified properties in the
current node.
If a matching rdataset does not exist, it is not an error.
*rdclass*, an ``int``, the class of the rdataset.
*rdtype*, an ``int``, the type of the rdataset.
*covers*, an ``int``, the covered type.
N)rQ r$ �remove)r' rK r
r r4 r r r
�delete_rdataset� s �zNode.delete_rdataset� replacementc C sN t |tjj�s td��t |tjj�r|�� }| �|j |j
|j � | � |� dS )a� Replace an rdataset.
It is not an error if there is no rdataset matching *replacement*.
Ownership of the *replacement* object is transferred to the node;
in other words, this method does not store a copy of *replacement*
at the node, it stores *replacement* itself.
*replacement*, a ``dns.rdataset.Rdataset``.
Raises ``ValueError`` if *replacement* is not a
``dns.rdataset.Rdataset``.
z replacement is not an rdatasetN)
�
isinstancer r r" �
ValueError�rrset�RRset� to_rdatasetrS rK r
r rJ �r' rT r r r
�replace_rdataset s �zNode.replace_rdatasetc C s4 | j D ]}t�|j|j�}|tjkr| S qtjS )aT Classify a node.
A node which contains a CNAME or RRSIG(CNAME) is a
``NodeKind.CNAME`` node.
A node which contains only "neutral" types, i.e. types allowed to
co-exist with a CNAME, is a ``NodeKind.NEUTRAL`` node. The neutral
types are NSEC, NSEC3, KEY, and their associated RRSIGS. An empty node
is also considered neutral.
A node which contains some rdataset which is not a CNAME, RRSIG(CNAME),
or a neutral type is a a ``NodeKind.REGULAR`` node. Regular nodes are
also commonly referred to as "other data".
)r$ r r r
r r rH r r r
r % s
�z
Node.classifyc C � dS )NFr r& r r r
� is_immutable: � zNode.is_immutable)#r r r r � __slots__r( r r) �Namer r6 r r1 r8 r; r< r> rA rJ r �NONE�
rdataclass�
RdataClassr! �boolr r" rP r rQ rS r[ r r r] r r r r
r# H sd " �����
�-�����
�)����
�r# c s� e Zd Z� fdd�Zejjdfdejjdejj dejj de
dej j f
� fd d
�
Z
ejjdfdejjdejj dejj de
deej j f
� fd d �
Zejjfdejjdejj dejj dd
fdd�Zdej j dd
fdd�Zde
fdd�Z� ZS )�
ImmutableNodec s$ t � �� tdd� |jD ��| _d S )Nc S s g | ]}t j�|��qS r )r r �ImmutableRdatasetrC r r r
rE C s z*ImmutableNode.__init__.<locals>.<listcomp>)�superr( �tupler$ )r' �node�� __class__r r
r( @ s
�zImmutableNode.__init__FrK r
r rL r c � |rt d��t� �|||d�S �N� immutableF)� TypeErrorrg rP �r' rK r
r rL rj r r
rP F � zImmutableNode.find_rdatasetc rl rm )ro rg rQ rp rj r r
rQ Q rq zImmutableNode.get_rdatasetNc C � t d���Nrn �ro )r' rK r
r r r r
rS \ s z ImmutableNode.delete_rdatasetrT c C rr rs rt rZ r r r
r[ d s z ImmutableNode.replace_rdatasetc C r\ )NTr r& r r r
r] g r^ zImmutableNode.is_immutable)r r r r( r r ra rb rc r! rd r r" rP r rQ rS r[ r] �
__classcell__r r rj r
re > sP
�����������
�����
�re ) r �enumr- �typingr r r �
dns.immutabler �dns.name�dns.rdataclass� dns.rdataset�
dns.rdatatype� dns.renderer� dns.rrsetr r r �NSEC�NSEC3�KEYr r �unique�Enumr r# rn re r r r r
�<module> s0 �� w