3
\¼me} ã @ sH d dl Z d dlmZ d dlmZ d dlmZ e jeƒZ G dd„ dƒZ
dS )é N)ÚStringIO)Úutil)Ú chop_commentc @ s~ e Zd Zdd„ Zdd„ Zedd„ ƒZedd„ ƒZejd d„ ƒZed
d „ ƒZ d d
„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ ZdS )Ú
ResolvConfc C s || _ d | _d S )N)Ú_textÚ _contents)ÚselfÚtext© r
ú!/usr/lib/python3.6/resolv_conf.pyÚ__init__ s zResolvConf.__init__c C s | j d kr| j| jƒ| _ d S )N)r Ú_parser )r r
r
r Úparse s
zResolvConf.parsec C s | j ƒ | jdƒS )NÚ
nameserver)r Ú _retr_option)r r
r
r Ú nameservers s zResolvConf.nameserversc C s" | j ƒ | jdƒ}|r |d S d S )NÚdomainr )r r )r Zdmr
r
r Ú local_domain s
zResolvConf.local_domainc C s0 | j ƒ | jdƒ | jjddt|ƒdgfƒ |S )Nr ÚoptionÚ )r Ú_remove_optionr ÚappendÚstr)r r r
r
r r ' s
c C sJ | j ƒ | jdƒ}g }x.|D ]&}x |jd ƒD ]}|r,|j|ƒ q,W q W |S )NÚsearch)r r Úsplitr )r Z current_sdsÚflat_sdsZsdlistZsdr
r
r Úsearch_domains. s
zResolvConf.search_domainsc C s– | j ƒ tƒ }x~| jD ]t\}}|dkr2|jdƒ q|dkrN|jd|d ƒ q|dkr|\}}}d||f }t|ƒr|||7 }|jd| ƒ qW |jƒ S )NÚblankÚ
Ú all_commentz%s
r r z%s %s)r r r ÚwriteÚlenÚgetvalue)r ÚcontentsÚ line_typeÚ
componentsÚcfg_optÚ cfg_valueZ comment_tailÚliner
r
r Ú__str__9 s
zResolvConf.__str__c C s@ g }x6| j D ],\}}|dkr |\}}}||kr |j|ƒ q W |S )Nr )r r )r Úopt_nameÚfoundr$ r% r& r' Ú
_comment_tailr
r
r r I s
zResolvConf._retr_optionc C sx | j ƒ | jdƒ}t|ƒ}|jt|ƒƒ tj|ƒ}t|ƒt|ƒkrF|S | jdƒ x"|D ]}| j jdd|dgfƒ qVW |S )Nr r r )
r r Úlistr r r Ú uniq_listr! r r )r ÚnsZ
current_nsZnew_nsÚnr
r
r Úadd_nameserverR s
zResolvConf.add_nameserverc s< ‡ fdd„}g }x | j D ]}||ƒs|j|ƒ qW || _ d S )Nc s. | \}}|dkrdS |\}}}|ˆ kr*dS dS )Nr FTr
)Úitemr$ r% r& Z
_cfg_valuer, )r* r
r Ú
remove_opt` s
z-ResolvConf._remove_option..remove_opt)r r )r r* r3 Z new_contentsÚcr
)r* r r _ s zResolvConf._remove_optionc C s˜ | j }t|ƒ}|jt|ƒƒ tj|ƒ}t|ƒt|ƒkr:|S t|ƒdkrRtd| ƒ‚dj|ƒ}t|ƒdkrttd| ƒ‚| j dƒ | j
jdd|dgfƒ |S ) Né z8Adding %r would go beyond the '6' maximum search domainsú é zEAdding %r would go beyond the 256 maximum search list character limitr r r ) r r- r r r r. r! Ú
ValueErrorÚjoinr r )r Z
search_domainr Znew_sdsZs_listr
r
r Úadd_search_domaino s$
z ResolvConf.add_search_domainc C sê g }xàt |jƒ ƒD ]Ð\}}|jƒ }|s8|jd|gfƒ qt|dƒ\}}t|jƒ ƒsd|jd|gfƒ q|sld}y|jd dƒ\}} W n8 ttfk
r¸ }
zt d|d ƒ|
‚W Y d d }
~
X nX |dkrÎt d | ƒ‚|jd
|| |gfƒ qW |S )Nr z;#r r é z)Incorrectly formatted resolv.conf line %sr r r ÚsortlistÚoptionsz Unexpected resolv.conf option %sr )r r r r< r= )
Ú enumerateÚ
splitlinesÚstripr r r! r Ú
IndexErrorr8 ÚIOError) r r# ZentriesÚir( ZslineÚheadÚtailr& Z
cfg_valuesÚer
r
r r
‡ s6 zResolvConf._parseN)Ú__name__Ú
__module__Ú __qualname__r r Úpropertyr r Úsetterr r) r r1 r r: r
r
r
r
r r s
r ) ZloggingÚior Z cloudinitr Zcloudinit.distros.parsersr Z getLoggerrG ZLOGr r
r
r
r Ú s