3 [_]g(@s`dZddlZddlZddlZddlZddlZddlZddlZddl m Z Gdddej j Z dS)zDNS Dynamic Update SupportN) string_typescsxeZdZejjddejjffdd ZdddZ ddZ dd Z d d Z d d Z ddZdddZdfdd ZZS)UpdateNcstt|j|jtjjtjjO_t|t r>tj j |}||_ t|t rZtj j |}||_|j|j|j |tjjddd|dk r|j|||ddS)aInitialize a new DNS Update object. See the documentation of the Message class for a complete description of the keyring dictionary. *zone*, a ``dns.name.Name`` or ``text``, the zone which is being updated. *rdclass*, an ``int`` or ``text``, the class of the zone. *keyring*, a ``dict``, the TSIG keyring to use. If a *keyring* is specified but a *keyname* is not, then the key used will be the first key in the *keyring*. Note that the order of keys in a dictionary is not defined, so applications should supply a keyname when a keyring is used, unless they know the keyring contains only one key. *keyname*, a ``dns.name.Name`` or ``None``, the name of the TSIG key to use; defaults to ``None``. The key must be defined in the keyring. *keyalgorithm*, a ``dns.name.Name``, the TSIG algorithm to use. T)createZ force_uniqueN) algorithm)superr__init__flagsdnsopcodeZto_flagsZUPDATE isinstancername from_textorigin rdataclass zone_rdclass find_rrsetZquestion rdatatypeZSOAZuse_tsig)selfzoneZrdclassZkeyringZkeynameZ keyalgorithm) __class__//tmp/pip-build-yz_zf6az/dnspython/dns/update.pyr!s     zUpdate.__init__c CsB|dkr|j}|j}|j|||j|j||dd}|j||dS)z&Add a single RR to the update section.NT) authoritycoversrrrdtypeadd)rr ttlrdZdeletingsectionrZrrsetrrr_add_rrFs  zUpdate._add_rrc Gs:t|trtjj|d}t|dtjjrnx@|D]8}|rF|j||jx |D]}|j ||j ||dqLWq0Wnt |}t |j d}t|dtjjr|r|j||djx|D]}|j ||||dqWnh|j d}t|trtjj|}|r|j||x6|D].} tjj|j|| |j}|j ||||dqWdS)aAdd records. *replace* is the replacement mode. If ``False``, RRs are added to an existing RRset; if ``True``, the RRset is replaced with the specified contents. The second argument is the section to add to. The third argument is always a name. The other arguments can be: - rdataset... - ttl, rdata... - ttl, rdtype, string... Nr)r)r rr r rrdatasetRdatasetdeleterr rlistintpoprdataRdatarrr) rreplacerr argsrdsrrrsrrr_addPs0         z Update._addcGs|jd|j|f|dS)zAdd records. The first argument is always a name. The other arguments can be: - rdataset... - ttl, rdata... - ttl, rdtype, string... FN)r-r)rr r*rrrr{s z Update.addc Gsft|trtjj|d}t|dkrR|j|j|tjj tj j tj j tj j ddnt|dtj j rx|D]&}x |D]}|j|d|tjj qtWqjWnt|}t|dtjjrx|D]}|j|d|tjj qWn|jd}t|trtj j|}t|dkr(|j|j||j|tj j tjj ddn:x8|D]0}tjj|j|||j}|j|d|tjj q.WdS)zDelete records. The first argument is always a name. The other arguments can be: - *empty* - rdataset... - rdata... - rdtype, [string...] NrT)r rr r rlenrrrANYrNONEr!r"r r$r'r(r&rr)rr r*r+rrr,rrrr#s6            z Update.deletecGs|jd|j|f|dS)abReplace records. The first argument is always a name. The other arguments can be: - rdataset... - ttl, rdata... - ttl, rdtype, string... 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. TN)r-r)rr r*rrrr)szUpdate.replacec Gst|trtjj|d}t|dkrL|j|j|tjj tj j tj j dddnt|dtj j s|t|dtjjs|t|dkrt|dtj j st|}|jdd|jd|j|f|n@|d}t|trtj j|}|j|j|tjj |tj j ddddS)aSRequire that an owner name (and optionally an rdata type, or specific rdataset) exists as a prerequisite to the execution of the update. The first argument is always a name. The other arguments can be: - rdataset... - rdata... - rdtype, string... NrTrF)r rr r rr.ranswerrr/rr0r!r"r'r(r$insertr-)rr r*rrrrpresents*          zUpdate.presentc Cst|trtjj|d}|dkrH|j|j|tjjtj j tj jdddn8t|tr^tj j|}|j|j|tjj|tj jddddS)zRequire that an owner name (and optionally an rdata type) does not exist as a prerequisite to the execution of the update.NT) r rr r rrr1rr0rr/)rr rrrrabsents       z Update.absentcs |dkr|j}tt|j||S)aLReturn a string containing the update in DNS compressed wire format. *origin*, a ``dns.name.Name`` or ``None``, the origin to be appended to any relative names. If *origin* is ``None``, then the origin of the ``dns.update.Update`` message object is used (i.e. the *zone* parameter passed when the Update object was created). *max_size*, an ``int``, the maximum size of the wire format output; default is 0, which means "the message's request payload, if nonzero, or 65535". Returns a ``binary``. N)rrrto_wire)rrmax_size)rrrr6szUpdate.to_wire)NN)N)Nr5)__name__ __module__ __qualname__r rINZtsigZdefault_algorithmrr r-rr#r)r3r4r6 __classcell__rr)rrrs$ +.' r)__doc__Z dns.messager Zdns.nameZ dns.opcodeZ dns.rdataZdns.rdataclassZ dns.rdatasetZdns.tsig_compatrmessageMessagerrrrrs