File: //usr/share/netplan/netplan/cli/__pycache__/utils.cpython-38.pyc
U
��~fd' � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZdZdZ dd� Z
dd� Z dd � Z e
ed
�d d �Zd
d� Zdd� Zd0dd�Zd1dd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd2d"d#�Zd$d%� Zd&d'� Z d(d)� Z d*d+� Z d,d-� Z G d.d/� d/ej �Z!dS )3� NzNetworkManager.servicez+snap.network-manager.networkmanager.servicec C s t j�dd�S )NZNETPLAN_GENERATE_PATHz/lib/netplan/generate)�os�environ�get� r r �'/usr/share/netplan/netplan/cli/utils.py�get_generator_path s r c C s t jdddtgt jd�dkS )N� systemctl�--quietz
is-enabled)�stderrr )�
subprocess�call�NM_SNAP_SERVICE_NAME�DEVNULLr r r r �is_nm_snap_enabled# s r c C s t jdg| t jt jd� d S )N�nmcli��stdoutr
�r �
check_callr ��argsr r r r ' s r )r �returnc C s t jdg| dd�S )Nr T�Zuniversal_newlines)r � check_outputr r r r � nmcli_out- s r c C s2 zt dg� W dS ttjfk
r, Y dS X dS )z"Check if NetworkManager is runningZgeneralTFN)r �OSErrorr ZSubprocessErrorr r r r �
nm_running3 s
r c C sl t �d�}t� }| D ]R}t|d��>}|D ]2}|�|�}|r(|�tt�||�d���� q\q(W 5 Q R X q|S )Nz^interface-name=(.*)$�r� ) �re�compile�set�open�match�update�fnmatch�filter�group)�pathsZdevicesZpat�
interfaces�path�f�line�mr r r �
nm_interfaces= s
r. Fc C s" t � rt| tg|�S t| tg|�S �N)r r r
�NM_SERVICE_NAME)�action�syncr r r �systemctl_network_managerK s r3 c C s: t |�dkr6d| g}|s"|�d� |�|� t�|� d S )Nr r z
--no-block)�len�append�extendr r )r1 �servicesr2 Zcommandr r r r S s
r c C s` t � } tjdddgdd�}|�� D ]8}|�� �d�}|d �� r"|d d kr"| �|d � q"| S )
N�
networkctlz
--no-pagerz --no-legendTr � r ���)Z unmanagedZlinger)r! r r �
splitlines�strip�split� isnumeric�add)r) �outr, �sr r r �networkd_interfaces_ s rB c C s t �ddg� d S )Nr8 �reload�r r r r r r �networkctl_reloadi s rE c C s& t | �dkr"t�ddgt| � � d S )Nr r8 � reconfigure)r4 r r �list)r) r r r �networkctl_reconfigurem s rH c C s t �ddd| g�dkrdS dS )z4Return True if at least one matching unit is runningr r z is-activer TF)r r )Z unit_patternr r r �systemctl_is_activer s rI c C s t �ddg� dS )zEReload systemd unit files from disk and re-calculate its dependenciesr z
daemon-reloadNrD r r r r �systemctl_daemon_reloady s rJ c C s t jddd| gt jt jd� dS )z8Flush all IP addresses of a given interface via iproute2Zip�addr�flushr Nr �Zifacer r r �
ip_addr_flush~ s rN c
C s t j�d| �}|r�zTtt j�|d���8}|�� �� }|dkrZt�d| |� W 5 Q R � W d S W 5 Q R X W n: tk
r� } z t� d| t
|�� W Y �d S d }~X Y nX z(t j� t j�|dd��}t j� |�}W n> tk
�r } z t�d| |t
|�� W Y �d S d }~X Y nX |S ) Nz/sys/class/netZ operstateZdownz'device %s operstate is %s, not changingz$Cannot determine operstate of %s: %sZdevice�driverz0Cannot replug %s: cannot read link %s/device: %s)
r r* �joinr" �readr< �logging�debug�IOError�error�str�realpath�basename)� interfaceZ only_downZdevdirr+ �state�erO � driver_namer r r �get_interface_driver_name� s$ r] c C s& t �| ��t ji g�d }|�dd�S )Nr rK � )� netifacesZ ifaddressesr ZAF_LINK)rY �linkr r r �get_interface_macaddress� s ra c C s t �| |�S r/ �r% Z fnmatchcase)rY Z
match_namer r r �is_interface_matching_name� s rc c s2 |}t |t�r|g}t| �� t� fdd�|D ��S )Nc 3 s | ]}t �� |�V qd S r/ rb )�.0�pattern�r\ r r � <genexpr>� s �z4is_interface_matching_driver_name.<locals>.<genexpr>)�
isinstancerV r] �any)rY Z match_driverZ driver_globsr rf r �!is_interface_matching_driver_name� s
�rj c C s t | �}|�� |�� kS r/ )ra �lower)rY Z match_mac�
macaddressr r r � is_interface_matching_macaddress� s rm c s� t � t�st�� �dd�r$� �d�nd}t�| |�}t|�dkr`� �d�r`tt� fdd�|��}t|�dkr�� �d�r�tt� fd d�|��}t|�dkr�t� |� d S |d
S ) N�nameF�*r rl c s t | � �d��S )Nrl )rm r rM �r# r r �<lambda>� � z%find_matching_iface.<locals>.<lambda>rO c s t | � �d��S )NrO )rj r rM rp r r rq � rr r )
rh �dict�AssertionErrorr r% r& r4 rG rR �info)r) r# Z name_globZmatchesr rp r �find_matching_iface� s
rv c @ sF e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dS )�NetplanCommandTFc C s� || _ || _|| _|| _d | _d| _d | _i | _d | _d | _ t
j dt j
d |f |dd�| _| jjdddd � |s�| jjd
d d d
�| _| jjdddd�}|j| jd� d S )NFz%s %sr T)�prog� description�add_helpz--debug�
store_truezEnable debug messages)r1 �helpzAvailable commandsr^ �
subcommand)�title�metavar�destr| zShow this help message)ry r| )�func)�
command_idry � leaf_command�testing�_argsrS � commandclass� subcommandsr} r� �argparse�ArgumentParser�sys�argv�parser� add_argumentZadd_subparsers�
subparsers�
add_parser� set_defaults� print_usage)�selfr� ry �leafr� Zp_helpr r r �__init__� s6 �
� ��zNetplanCommand.__init__c C s
|| _ d S r/ )r� )r� r r r r r$ � s zNetplanCommand.updatec C s> | j j| j| d�\}| _| js:| js:tdtjd� | �� d S )N)r � namespacez You need to specify a command��file) r� Zparse_known_argsr� r} r� �printr� r
r� )r� �nsr r r �
parse_args� s zNetplanCommand.parse_argsc C s8 | j r| j �| j� | jr,d| jkr,| �� | �� d S )Nr| )r� r$ r� r� r� r� �r� r r r � run_command� s
zNetplanCommand.run_commandc C s | j jtjd� t�tj� d S )Nr� )r� Z
print_helpr� r
�exitr �EX_USAGEr� r r r r� � s zNetplanCommand.print_usagec C s� |� }i | j |< || j | d< || j | d< |jrDtj�dd �sDd S | jj|j|j|jdd�}|j |j
|d� || j | d< d S )N�class�instanceZENABLE_TEST_COMMANDSF)ry r| rz )r� r� r� ) r� r� r r r r� r� r� ry r� �run)r� rn r� r� �pr r r �_add_subparser_from_class s
�z(NetplanCommand._add_subparser_from_classc C s@ dd l }|�|�D ](\}}|�|�rt|t�r| �||� qd S )Nr )�inspectZ
getmembersZisclass�
issubclassrw r� )r� Z
submodulesr� rn �objr r r �_import_subcommands s z"NetplanCommand._import_subcommandsN)TF)
�__name__�
__module__� __qualname__r� r$ r� r� r� r� r� r r r r rw � s
rw )F)F)F)"r� r rR r� r r_ r% r r0 r
r r r rG rV r r r. r3 r rB rE rH rI rJ rN r] ra rc rj rm rv Z Namespacerw r r r r �<module> s<