Current File : //snap/core18/current/usr/lib/python3/dist-packages/probert/__pycache__/utils.cpython-36.pyc
3
��ZXu � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZddddddd d
d d d
ddddddddddddgZddd d d d gZd d!d"d#d$d%d&d'gZ d(d)� Z
ej d?k r�d+d,� Z nd-d,� Z e
d.fd/d0�Zd1d2� Zd3d4� Zd5d6� Zd7d8� Zd9d:� Zd;d<� Zd=d>� ZdS )@� )�deepcopyNZaddressZnetmaskZ broadcastZnetworkZmetricZgatewayZ pointtopointZmediaZmtuZhostnameZ
leasehoursZ leasetimeZvendorZclientZbootfile�serverZhwaddrZprovider�frameZnetnumZendpointZlocalZttlzpre-upZupzpost-upZdownzpre-downz post-downZ
bridge_ageingZbridge_bridgeprioZ bridge_fdZ
bridge_gcinitZ bridge_helloZ
bridge_maxageZbridge_maxwaitZ
bridge_stpc C s� t | �}t| t�r*t|t�r*|j|� |S x�|j� D ]v\}}||kr�t|| t�rpt|t�rpt|| |�||< q�t|| t�r�t|t�r�tt|| | ��||< q4|||< q4W |S )N)r �
isinstance�list�extend�items�dict�
dict_merge�set)Zonto�source�target�key�value� r �//usr/lib/python3/dist-packages/probert/utils.pyr
s
r
� c C sB i }x8| j D ].}| j j|�}t|t�r2|jdd�}|||< q W |S )Nzutf-8�replace)�
attributes�getr �bytes�decode)�device�rr �valr r r �udev_get_attributes- s
r c C sD i }x:| j jD ].}| j j|�}t|t�r4|jdd�}|||< qW |S )Nzutf-8r )r Zavailable_attributesr r r r )r r r r r r r r 6 s
� c s( dd� t tj� fdd�| D �|��D �S )Nc s s | ]\}}d d� ||�V qdS )c s � fdd�|D �S )Nc 3 s | ]\}}|� kr|V qd S )Nr )�.0Zpred�item)�ir r � <genexpr>B s z:partitionn2.<locals>.<genexpr>.<lambda>.<locals>.<genexpr>r )r �teer )r r �<lambda>B s z'partitionn2.<locals>.<genexpr>.<lambda>Nr )r �x�tr r r r B s z partitionn2.<locals>.<genexpr>c 3 s | ]}� |�|fV qd S )Nr )r r )� predicater r r C s )� enumerate� itertoolsr! )r r% �nr )r% r � partitionn2A s r) c s> | j � � t� � fdd�dd�}dd� |D �}|\\}}||fS )z{Unpack a partition into a tuple of (first partition, second partition)
param: partition iterator from partitionn2
c s � j | �dkS )Nr )�index)r# )r r r r" Q s z#partition_to_pair.<locals>.<lambda>r )r r% r( c S s g | ] }t |��qS r )r )r �pr r r �
<listcomp>S s z%partition_to_pair.<locals>.<listcomp>)�splitr) )�inputZ
partitions�datar r r )r r �partition_to_pairJ s
r0 c C s dd� | j d�D �S )Nc S s& g | ] }t |�r|jd �d j� �qS )�;r )�lenr- �strip)r r# r r r r, Z s z3disentagle_data_from_whitespace.<locals>.<listcomp>�
)r- )r/ r r r � disentagle_data_from_whitespaceX s r5 c C s~ i }i }xbt | �D ]V}t|�dkr$qt|�\}}|dkrP|j|d |d i� qdj|�}|j||i� qW |jd|i� |S )zuTransform lease string into dictionary of attributes
params: lease: string if a dhcp lease structure { to }
r �option� � �options)r5 r2 r0 �update�join)�leaseZ
lease_dictr9 �liner r r r r �
dictify_lease^ s
r> c C s dd� t jd| jdd��D �S )z�Parses dhclient leases file data, returning dictionary of leases
:param leasesdata: string of lease data read from leases file
c S s g | ] }t |��qS r )r> )r r< r r r r, y s z.parse_dhclient_leases_file.<locals>.<listcomp>z
{([^{}]*)}�"� )�re�findallr )� leasedatar r r �parse_dhclient_leases_filet s rD c C sR i }xH| j d�D ]:}|jd�st|�dk r,q|j d�}|d ||d j� < qW |S )z6Parses systemd/networkd/netif lease data, returns dictr4 �#r7 �=r )r- �
startswithr2 �lower)rC r< r= Zkeyvaluer r r �parse_networkd_lease_file} s
rI c C s* ddg} x | D ]}t jj|�r|S qW d S )Nz
/var/lib/dhcpz/var/lib/dhclient)�os�path�exists)Zsupported_dirs�dr r r �get_dhclient_d� s
rN c C s6 d}|}�x�|j � D �]�}|j� }|jd�r.q|jd�}|d }|dkr\tjj||d �}q|dkr�tjj||d �}xPtj|�D ]B} t| d��}
|
j � j� } W dQ R X t
| | tjj tjj | ��� q�W q|d k�rx0|dd� D ] } | | kr�i | | < d
| | d < q�W q|d k�rl|dd � \} }
}| | k�r8i | | < nd
| | k�rNt
d��|
| | d
< || | d< | }q|dk�r�|d | | d< q|tk�r�|d | | |< q|tk�r�|| | k�r�g | | |< | | | jdj|dd� �� q|jd��r�d| | k�ri | | d< |dk�rZg | | d d< xv|dd� D ] }| | d d j|� �q8W nH|dk�r g | | d d< x,|dd� D ] }| | d d j|� �q�W q|jd�rd| | k�r�i | | d< |tk�r�|jdd�}|d | | d |< q|dk�r>g | | d d< x�|dd� D ] } | | d d j| � �q W q|d k�rp|d j� d k�rp|d | | d d < q|d k�r�d | | d k�r�i | | d d < |d | | d d |d < q|d!krd"| | d k�r�i | | d d"< |d | | d d" |d < qW x,| j� D ] } d | | k�rd#| | d < �qW dS )$z�Parses the file contents, placing result into ifaces.
:param ifaces: interface dictionary
:param contents: contents of interfaces file
:param path: directory interfaces file was located
NrE r8 r zsource-directoryr7 r r �autoT�iface� �familyz!Cannot define %s interface again.�methodZ hwaddresszdns-Zdnsz
dns-search�searchzdns-nameserversZ nameserversZbridge_Zbridger@ Z bridge_portsZportsZ bridge_hwZmacr Zbridge_pathcostZpathcostZbridge_portprioZportprioF)�
splitlinesr3 rG r- rJ rK r; �glob�open�read� parse_etc_network_interfaces�dirname�abspath� Exception�NET_CONFIG_OPTIONS�NET_CONFIG_COMMANDS�append�NET_CONFIG_BRIDGE_OPTIONSr rH �keys)Zifaces�contentsrK ZcurrifZsrc_dirr= r- r6 Zsrc_pathZsrc_file�fpZsrc_datarP rR rS Zdomainr Z
bridge_optionr r r rY � s�
"
rY )r r )�copyr rV r' rJ rA Zpyudevr] r^ r` r
Z__version_info__r �intr) r0 r5 r> rD rI rN rY r r r r �<module> s0
Mini Shell