File: //usr/lib/python3/dist-packages/cloudinit/distros/parsers/__pycache__/resolv_conf.cpython-38.pyc
U
Ӈgw � @ sH d dl Z d dlmZ d dlmZ d dlmZ e �e�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 �G/usr/lib/python3/dist-packages/cloudinit/distros/parsers/resolv_conf.py�__init__ s zResolvConf.__init__c C s | j d kr| �| j�| _ d S r )r �_parser �r r r r �parse s
zResolvConf.parsec C s | � � | �d�S )N�
nameserver�r � _retr_optionr r r r � nameservers s zResolvConf.nameserversc C s" | � � | �d�}|r |d S d S )N�domainr r )r Zdmr r r � local_domain s
zResolvConf.local_domainc C s0 | � � | �d� | j�ddt|�dgf� |S )Nr �option� )r �_remove_optionr �append�str)r r r r r r ' s
c C sB | � � | �d�}g }|D ]"}|�d �D ]}|r(|�|� q(q|S )N�search)r r �splitr )r Z current_sds�flat_sdsZsdlistZsdr r r �search_domains. s
zResolvConf.search_domainsc C s� | � � t� }| jD ]t\}}|dkr0|�d� q|dkrL|�d|d � q|dkr|\}}}d||f }t|�rz||7 }|�d| � q|�� 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 }| j D ],\}}|dkr
|\}}}||kr
|�|� q
|S )Nr �r r )r �opt_name�foundr' r( r) r* �
_comment_tailr r r r I s
zResolvConf._retr_optionc C st | � � | �d�}t|�}|�t|�� t�|�}t|�t|�krF|S | �d� |D ]}| j �dd|dgf� qT|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 s8 � fdd�}g }| j D ]}||�s|�|� q|| _ d S )Nc s. | \}}|dkrdS |\}}}|� kr*dS dS )Nr FTr )�itemr' r( r) Z
_cfg_valuer0 �r. r r �
remove_opt` s
z-ResolvConf._remove_option.<locals>.remove_optr- )r r. r8 Z new_contents�cr r7 r r _ s
zResolvConf._remove_optionc C s� | j }t|�}|�t|�� t�|�}t|�t|�kr:|S t|�dkrRtd| ��d�|�}t|�dkrttd| ��| � d� | j
�dd|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 r1 r r r r2 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 }t |�� �D ]�\}}|�� }|s6|�d|gf� qt|d�\}}t|�� �sb|�d|gf� q|sjd}z|�d d�\}} W n8 ttfk
r� }
zt d|d �|
�W 5 d }
~
X Y nX |dkr�t d| ��|�d || |gf� q|S )
Nr z;#r" r � z)Incorrectly formatted resolv.conf line %s)r r r ZsortlistZoptionsz Unexpected resolv.conf option %sr )
� enumerate�
splitlines�stripr r r$ r �
IndexErrorr= �IOError) r r& �entries�ir+ Zsline�head�tailr) Z
cfg_values�er r r r � s0
�� zResolvConf._parseN)�__name__�
__module__� __qualname__r
r �propertyr r �setterr r, r r5 r r? r r r r r r s
r ) Zlogging�ior Z cloudinitr Zcloudinit.distros.parsersr Z getLoggerrK ZLOGr r r r r �<module> s