U î¤ùdx1ã@sÊdZddlZddlZddlmZmZmZddlZddl Zddl Zddl Zddl Zddl ZddlZejjhZejjejjejjhZdd„ZejGdd„dejƒƒZGdd „d ƒZejjGd d „d eƒƒZdS) z)DNS nodes. A node is a set of rdatasets.éN)ÚAnyÚDictÚOptionalcCs||kp|tjjko||kS©N)ÚdnsÚ rdatatypeZRRSIG)ZrdtypesÚrdtypeÚcovers©r ú,/tmp/pip-unpacked-wheel-pk5slln3/dns/node.pyÚ_matches_type_or_its_signature,sr c@sReZdZdZdZdZdZeej j ej j ddœdd„ƒZ eej j ddœd d „ƒZd S) ÚNodeKindzRdatasets in nodesréé)rr ÚreturncCs.tt||ƒrtjStt||ƒr$tjStjSdSr)r Ú _cname_typesr ÚCNAMEÚ_neutral_typesÚNEUTRALÚREGULAR)Úclsrr r r r Úclassify8s   zNodeKind.classify)ÚrdatasetrcCs| |j|j¡Sr)rrr )rrr r r Úclassify_rdatasetCszNodeKind.classify_rdatasetN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrÚ classmethodrrÚ RdataTyperrÚRdatasetrr r r r r 0sþ r c@s(eZdZdZdgZdd„Zejje e e fe dœdd„Z dd „Z d d „Zd d „Zdd„Zdd„Zdd„Zejjdfejjejjejjeejjdœdd„Zejjdfejjejjejjeeejjdœdd„Zejjfejjejjejjddœdd„Zejjddœdd „Ze 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. Ú rdatasetscCs g|_dSr©r"©Úselfr r r Ú__init__[sz Node.__init__)ÚnameÚkwrcKsNt ¡}|jD].}t|ƒdkr| |j|f|Ž¡| 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_textcCsdtt|ƒƒdS)Nz )ÚstrÚidr$r r r Ú__repr__ssz Node.__repr__cCs<|jD]}||jkrdSq|jD]}||jkr"dSq"dS)NFTr#)r%ÚotherÚrdr r r Ú__eq__vs    z Node.__eq__cCs | |¡ Sr)r9)r%r7r r r Ú__ne__‚sz Node.__ne__cCs t|jƒSr)r-r"r$r r r Ú__len__…sz Node.__len__cCs t|jƒSr)Úiterr"r$r r r Ú__iter__ˆsz Node.__iter__cCsbt|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. rcSs g|]}t |¡tjkr|‘qSr )r rr©Ú.0r2r r r Ú ˜sþz)Node._append_rdataset..cSs g|]}t |¡tjkr|‘qSr )r rrr>r r r r@žsþN)r-r"r rrrÚappend©r%rÚkindr r r Ú_append_rdataset‹s   þ  þzNode._append_rdatasetF©Úrdclassrr ÚcreatercCsH|jD]}| |||¡r|Sq|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ÚKeyErrorrrr rD©r%rFrr rGr2r r r Ú find_rdataset§s   zNode.find_rdatasetcCs2z| ||||¡}Wntk r,d}YnX|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)rKrIrJr r r Ú get_rdatasetÐs  zNode.get_rdatasetN©rFrr rcCs&| |||¡}|dk r"|j |¡dS)aDDelete 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)rLr"Úremove)r%rFrr r2r r r Údelete_rdatasetõszNode.delete_rdataset©Ú replacementrcCsNt|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``. zreplacement is not an rdatasetN) Ú isinstancerrr Ú ValueErrorZrrsetZRRsetZ to_rdatasetrOrFrr rD©r%rQr r r Úreplace_rdataset sÿzNode.replace_rdataset©rcCs4|jD]&}t |j|j¡}|tjkr|SqtjS)aTClassify 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 rrr rrBr r r r%s    z Node.classifycCsdS)NFr r$r r r Ú is_immutable:szNode.is_immutable)#rrrrÚ __slots__r&rr'ÚNamerr4rr/r6r9r:r;r=rDrÚNONEÚ rdataclassÚ RdataClassrÚboolrr rKrrLrOrUr rrWr r r r r!HsL  ûú -û ú )üû r!csÒeZdZ‡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)Ú ImmutableNodecs$tƒ ¡tdd„|jDƒƒ|_dS)NcSsg|]}tj |¡‘qSr )rrZImmutableRdatasetr>r r r r@Csz*ImmutableNode.__init__..)Úsuperr&Útupler")r%Únode©Ú __class__r r r&@s ÿzImmutableNode.__init__FrEcs|r tdƒ‚tƒ |||d¡S©NÚ immutableF)Ú TypeErrorr_rK©r%rFrr rGrbr r rKFszImmutableNode.find_rdatasetcs|r tdƒ‚tƒ |||d¡Srd)rfr_rLrgrbr r rLQszImmutableNode.get_rdatasetNrMcCs tdƒ‚dS©Nre©rf)r%rFrr r r r rO\szImmutableNode.delete_rdatasetrPcCs tdƒ‚dSrhrirTr r r rUdszImmutableNode.replace_rdatasetrVcCsdS)NTr r$r r r rWgszImmutableNode.is_immutable)rrrr&rrrZr[r\rr]rr rKrrLrOrUrWÚ __classcell__r r rbr r^>s8 ûúû úüû r^)rÚenumr+ÚtypingrrrZ dns.immutablerZdns.nameZdns.rdataclassZ dns.rdatasetZ dns.rdatatypeZ dns.rendererZ dns.rrsetrrrZNSECZNSEC3ZKEYrr ÚuniqueÚEnumr r!rer^r r r r Ús.ÿýw