Google
Edit File: node.cpython-39.pyc
a �M�ew1 � @ 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� rdatatypeZRRSIG)Zrdtypes�rdtype�covers� r �,/usr/lib/python3.9/site-packages/dns/node.py� _matches_type_or_its_signature, s r c @ sR e Zd ZdZdZdZdZeej j ej j d d�dd��Z eej j d d�d d ��Zd S ) �NodeKindzRdatasets in nodesr � � )r r �returnc C s. t t||�rtjS t t||�r$tjS tjS d S r )r � _cname_typesr �CNAME�_neutral_types�NEUTRAL�REGULAR)�clsr r r r r �classify8 s zNodeKind.classify)�rdatasetr c 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 @ s( e Zd ZdZdgZdd� Zejje e e f e d�dd�Z dd � Z d d � Zd d � Zdd� Zdd� Zdd� Zejjdfejjejjejjeejjd�dd�Zejjdfejjejjejjee ejj d�dd�Z ejjfejjejjejjdd�d d �Z ejjdd �d d �Z e d!�d"d#�Z!ed!�d$d%�Z"dS )&�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__Z s z Node.__init__)�name�kwr c K sR t �� }| jD ]2}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 r/ ^ s z Node.to_textc C s dt t| �� d S )Nz <DNS node �>)�str�idr$ r r r �__repr__r s z Node.__repr__c C s< | j D ]}||j vr dS q|j D ]}|| j vr" dS q"dS )NFTr# )r% �otherZrdr r r �__eq__u s z Node.__eq__c C s | � |� S r )r8 )r% r7 r r r �__ne__� s z Node.__ne__c C s t | j�S r )r- r" r$ r r r �__len__� s z Node.__len__c C s t | j�S r )�iterr" r$ r r r �__iter__� s z Node.__iter__c C sb t | j�dkrRt�|�}|tjkr6dd� | jD �| _n |tjkrRdd� | 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 s g | ]}t �|�t jkr|�qS r )r r r ��.0r2 r r r � <listcomp>� s �z)Node._append_rdataset.<locals>.<listcomp>c S s g | ]}t �|�t jkr|�qS r )r r r r= r r r r? � s �N)r- r" r r r r �append�r% r �kindr r r �_append_rdataset� s � �zNode._append_rdatasetF��rdclassr r �creater c C sH | j D ]}|�|||�r| S q|s*t�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 rC �r% rE r r rF r2 r r r � find_rdataset� s zNode.find_rdatasetc C s0 z| � ||||�}W n ty* d}Y n0 |S )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)rJ rH rI r r r � get_rdataset� s zNode.get_rdatasetN�rE r r r c C s& | � |||�}|dur"| j�|� 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)rK r" �remove)r% rE r r r2 r r r �delete_rdataset� s zNode.delete_rdataset�� replacementr c C sN t |tjj�std��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 � ValueErrorZrrsetZRRsetZ to_rdatasetrN rE r r rC �r% rP r r r �replace_rdataset s �zNode.replace_rdataset�r c 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 rA r r r r $ s z Node.classifyc C s dS )NFr r$ r r r � is_immutable9 s zNode.is_immutable)#r r r r � __slots__r& r r' �Namer r4 r r/ r6 r8 r9 r: r< rC r �NONE� rdataclass� RdataClassr �boolr r rJ r rK rN rT r r rV r r r r r! H sL �� -� � )�� r! c s� e Zd Z� fdd�Zejjdfejjejj ejj e ej j d�� fdd� Z ejjdfejjejj ejj e eej j d�� fdd� Zejjfejjejj ejj d d �d d �Zej j d d �dd�Ze d�dd�Z� ZS )� ImmutableNodec s$ t � �� tdd� |jD ��| _d S )Nc S s g | ]}t j�|��qS r )r r ZImmutableRdatasetr= r r r r? B � z*ImmutableNode.__init__.<locals>.<listcomp>)�superr& �tupler" )r% Znode�� __class__r r r& ? s �zImmutableNode.__init__FrD c s |r t d��t� �|||d�S �N� immutableF)� TypeErrorr_ rJ �r% rE r r rF ra r r rJ E s zImmutableNode.find_rdatasetc s |r t d��t� �|||d�S rc )re r_ rK rf ra r r rK P s zImmutableNode.get_rdatasetNrL c C s t d��d S �Nrd �re )r% rE r r r r r rN [ s z ImmutableNode.delete_rdatasetrO c C s t d��d S rg rh rS r r r rT c s z ImmutableNode.replace_rdatasetrU c C s dS )NTr r$ r r r rV f s zImmutableNode.is_immutable)r r r r& r r rY rZ r[ r r\ r r rJ r rK rN rT rV � __classcell__r r ra r r] = s8 ��� ��� r] ) r �enumr+ �typingr r r Z dns.immutabler Zdns.nameZdns.rdataclassZ dns.rdatasetZ dns.rdatatypeZ dns.rendererZ dns.rrsetr r r ZNSECZNSEC3ZKEYr r �unique�Enumr r! rd r] r r r r �<module> s. �� v