File: //lib/python3.6/site-packages/cloudinit/distros/parsers/__pycache__/resolv_conf.cpython-36.pyc
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.<locals>.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 �<module> s