OwlCyberSecurity - MANAGER
Edit File: node.cpython-36.pyc
3 �b�W� � @ s< d Z ddlmZ ddlZddlZddlZG dd� de�ZdS )z)DNS nodes. A node is a set of rdatasets.� )�StringIONc @ s� e Zd ZdZdgZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z e j jdfdd�Ze j jdfdd�Ze j jfdd�Zdd� ZdS )�Nodez�A DNS node. A node is a set of rdatasets @ivar rdatasets: the node's rdatasets @type rdatasets: list of dns.rdataset.Rdataset objects� rdatasetsc C s g | _ dS )z Initialize a DNS node. N)r )�self� r �/usr/lib/python3.6/node.py�__init__$ s z Node.__init__c K sP t � }x8| jD ].}t|�dkr|j|j|f|�� |jd� qW |j� 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. @param name: the owner name of the rdatasets @type name: dns.name.Name object @rtype: string r � N� ���)r r �len�write�to_text�getvalue)r �name�kw�s�rdsr r r r * s z Node.to_textc C s dt t| �� d S )Nz <DNS node �>)�str�id)r r r r �__repr__; s z Node.__repr__c C s@ x | j D ]}||j krdS qW x |j D ]}|| j kr&dS q&W dS )zSTwo nodes are equal if they have the same rdatasets. @rtype: bool FT)r )r �otherZrdr r r �__eq__> s z Node.__eq__c C s | j |� S )N)r )r r r r r �__ne__N s z Node.__ne__c C s t | j�S )N)r r )r r r r �__len__Q s z Node.__len__c C s t | j�S )N)�iterr )r r r r �__iter__T s z Node.__iter__Fc C sH x | j D ]}|j|||�r|S qW |s*t�tjj||�}| j j|� |S )a Find an rdataset matching the specified properties in the current node. @param rdclass: The class of the rdataset @type rdclass: int @param rdtype: The type of the rdataset @type rdtype: int @param covers: 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. @type covers: int @param create: If True, create the rdataset if it is not found. @type create: bool @raises KeyError: An rdataset of the desired type and class does not exist and I{create} is not True. @rtype: dns.rdataset.Rdataset object )r �match�KeyError�dns�rdataset�Rdataset�append)r �rdclass�rdtype�covers�creater r r r � find_rdatasetW s zNode.find_rdatasetc C s2 y| j ||||�}W n tk r, d}Y nX |S )aE 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 I{create} is not True. @param rdclass: The class of the rdataset @type rdclass: int @param rdtype: The type of the rdataset @type rdtype: int @param covers: The covered type. @type covers: int @param create: If True, create the rdataset if it is not found. @type create: bool @rtype: dns.rdataset.Rdataset object or None N)r( r )r r$ r% r& r' r r r r � get_rdatasety s zNode.get_rdatasetc C s& | j |||�}|dk r"| jj|� dS )az Delete the rdataset matching the specified properties in the current node. If a matching rdataset does not exist, it is not an error. @param rdclass: The class of the rdataset @type rdclass: int @param rdtype: The type of the rdataset @type rdtype: int @param covers: The covered type. @type covers: int N)r) r �remove)r r$ r% r& r r r r �delete_rdataset� s zNode.delete_rdatasetc C s: t |tjj�std��| j|j|j|j� | j j |� dS )a8 Replace an rdataset. It is not an error if there is no rdataset matching I{replacement}. Ownership of the I{replacement} object is transferred to the node; in other words, this method does not store a copy of I{replacement} at the node, it stores I{replacement} itself. z replacement is not an rdatasetN) � isinstancer r! r" � ValueErrorr+ r$ r% r& r r# )r Z replacementr r r �replace_rdataset� s zNode.replace_rdatasetN)�__name__� __module__� __qualname__�__doc__� __slots__r r r r r r r r Z rdatatypeZNONEr( r) r+ r. r r r r r s ! r ) r2 �ior Z dns.rdatasetr Z dns.rdatatypeZ dns.renderer�objectr r r r r �<module> s