File: //snap/core22/2133/usr/lib/python3/dist-packages/probert/__pycache__/utils.cpython-310.pyc
o
?�^)# � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZg d�Zg d�Zg d�Z dZ
dd� Z ej d k r6d
d � Z
nd d � Z
ed
fdd�Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� ZdS )$� )�deepcopyN)�address�netmask� broadcast�network�metric�gateway� pointtopoint�media�mtu�hostname�
leasehours� leasetime�vendor�client�bootfile�server�hwaddr�provider�frame�netnum�endpoint�local�ttl)zpre-up�upzpost-up�downzpre-downz post-down)�
bridge_ageing�bridge_bridgeprio� bridge_fd�
bridge_gcinit� bridge_hello�
bridge_maxage�bridge_maxwait�
bridge_stpi c C s� t | �}t| t�rt|t�r|�|� |S |�� D ];\}}||v rPt|| t�r7t|t�r7t|| |�||< qt|| t�rOt|t�rOtt|| | ��||< q|||< q|S �N)r �
isinstance�list�extend�items�dict�
dict_merge�set)�onto�source�target�key�value� r1 �//usr/lib/python3/dist-packages/probert/utils.pyr* s
�
r* )r � c C s> i }| j D ]}| j �|�}t|t�r|�dd�}|||< q|S �Nzutf-8�replace)�
attributes�getr% �bytes�decode��device�rr/ �valr1 r1 r2 �udev_get_attributes1 s
r> c C s@ i }| j jD ]}| j �|�}t|t�r|�dd�}|||< q|S r4 )r6 �available_attributesr7 r% r8 r9 r: r1 r1 r2 r> : s
� c s( dd� t t�� fdd�| D �|��D �S )Nc s s" � | ] \}}d d� ||�V qdS )c s � fdd�|D �S )Nc 3 s � | ] \}}|� kr|V qd S r$ r1 )�.0�pred�item��ir1 r2 � <genexpr>F s � z:partitionn2.<locals>.<genexpr>.<lambda>.<locals>.<genexpr>r1 )rE �teer1 rD r2 �<lambda>F s z'partitionn2.<locals>.<genexpr>.<lambda>Nr1 )rA �x�tr1 r1 r2 rF F s � �z partitionn2.<locals>.<genexpr>c 3 s � | ] }� |�|fV qd S r$ r1 )rA rC �� predicater1 r2 rF G s � �)� enumerate� itertoolsrG �r( rL �nr1 rK r2 � partitionn2E s ���rQ c s> | � � � 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 � � | �dkS )Nr )�index)rI �r( r1 r2 rH U s z#partition_to_pair.<locals>.<lambda>r@ rO c S � g | ]}t |��qS r1 )r& )rA �pr1 r1 r2 �
<listcomp>W � z%partition_to_pair.<locals>.<listcomp>)�splitrQ )�input�
partitions�datar/ r0 r1 rS r2 �partition_to_pairN s
�
r\ c C s dd� | � d�D �S )Nc S s& g | ]}t |�r|�d �d �� �qS )�;r )�lenrX �strip)rA rI r1 r1 r2 rV ^ s �z3disentagle_data_from_whitespace.<locals>.<listcomp>�
)rX )r[ r1 r1 r2 � disentagle_data_from_whitespace\ s ra c C sz i }i }t | �D ]+}t|�dkrqt|�\}}|dkr'|�|d |d i� qd�|�}|�||i� q|�d|i� |S )zuTransform lease string into dictionary of attributes
params: lease: string if a dhcp lease structure { to }
r �option� � �options)ra r^ r\ �update�join)�lease�
lease_dictre �liner/ r0 r1 r1 r2 �
dictify_leaseb s
rk c C s dd� t �d| �dd��D �S )z�Parses dhclient leases file data, returning dictionary of leases
:param leasesdata: string of lease data read from leases file
c S rT r1 )rk )rA rh r1 r1 r2 rV } rW z.parse_dhclient_leases_file.<locals>.<listcomp>z
{([^{}]*)}�"� )�re�findallr5 )� leasedatar1 r1 r2 �parse_dhclient_leases_filex s �rq c C sN i }| � d�D ] }|�d�st|�dk rq|� d�}|d ||d �� < q|S )z6Parses systemd/networkd/netif lease data, returns dictr` �#rc �=r )rX �
startswithr^ �lower)rp rh rj �keyvaluer1 r1 r2 �parse_networkd_lease_file� s
rw c C s* ddg} | D ] }t j�|�r| S qd S )Nz
/var/lib/dhcpz/var/lib/dhclient)�os�path�exists)�supported_dirs�dr1 r1 r2 �get_dhclient_d� s �r} c
C s d}|}|� � D �]�}|�� }|�d�rq|�d�}|d }|dkr,tj�||d �}q|dkrktj�||d �}t�|�D ]+} t| d��}
|
� � �� } W d � n1 sVw Y t
| | tj� tj� | ��� q>q|d kr�|dd� D ]} | | vri | | < d
| | d < quq|d kr�|dd � \} }
}| | vr�i | | < n
d
| | v r�t
d��|
| | d
< || | d< | }q|dkr�|d | | d< q|tv r�|d | | |< q|tv r�|| | vr�g | | |< | | | �d�|dd� �� q|�d��rJd| | v�ri | | d< |dk�r'g | | d d< |dd� D ]}| | d d �|� �qq|dk�rIg | | d d< |dd� D ]}| | d d �|� �q:q|�d��r�d| | v�r]i | | d< |tv �rs|�dd�}|d | | d |< q|dk�r�g | | d d< |dd� D ]} | | d d �| � �q�q|d k�r�|d �� d k�r�|d | | d d < q|d k�r�d | | d v�r�i | | d d < |d | | d d |d < q|d!k�r�d"| | d v�r�i | | d d"< |d | | d d" |d < q| �� D ]} d | | v�r d#| | d < �q�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
Nrr rd r zsource-directoryrc r- r< �autoT�iface� �familyz!Cannot define %s interface again.�method� hwaddresszdns-�dnsz
dns-search�searchzdns-nameservers� nameservers�bridge_�bridgerm � bridge_ports�ports� bridge_hw�macr@ �bridge_pathcost�pathcost�bridge_portprio�portprioF)�
splitlinesr_ rt rX rx ry rg �glob�open�read� parse_etc_network_interfaces�dirname�abspath� Exception�NET_CONFIG_OPTIONS�NET_CONFIG_COMMANDS�append�NET_CONFIG_BRIDGE_OPTIONSr5 ru �keys)�ifaces�contentsry �currif�src_dirrj rX rb �src_path�src_file�fp�src_datar r� r� �domainr �
bridge_optionr1 r1 r2 r� � s�
����
"
�
�
�
� ��r� c C sf t j�dt j�| ��}t j�|d�}t|��}t|�� �� �t }W d � |S 1 s,w Y |S )zA /sys/class/block/<device>/size and return integer value in bytes�/sys/class/block�sizeN) rx ry rg �basenamer� �intr� r_ �SECTOR_SIZE_BYTES)r; �
device_dir�
blockdev_sizer| r� r1 r1 r2 �read_sys_block_size_bytes� s
��r� c C s* t j�dt j�| ��}t �t j�|d��S )Nr� �slaves)rx ry rg r� �listdir)r; r� r1 r1 r2 �read_sys_block_slaves� s r� c c s( � | j dd�D ]
}d|vrq|V qd S )N�block)� subsystem�MAJOR)� list_devices)�contextr; r1 r1 r2 �sane_block_devices� s ��r� )�copyr r� rN rx rn �pyudevr� r� r� r� r* �__version_info__r> r� rQ r\ ra rk rq rw r} r� r� r� r� r1 r1 r1 r2 �<module> s2
S