ruạṛ
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
cải xoăn