File: //usr/share/netplan/netplan/cli/__pycache__/utils.cpython-310.pyc
o
|hEc$ � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z
d dl m Z d dlm
Z
dZdZe e
efZdd� Zdd � Zd
d � Zd d
� Zdd� Zd*dd�Zd*dd�Zdd� Zdd� Zdd� Zdd � Z d d � Z d d � Z d*d!d"�Z d#d$� Z d%e!fd&d'�Z"G d(d)� d)ej#�Z$dS )+� N)�ConfigurationError)�LibNetplanExceptionzNetworkManager.servicez+snap.network-manager.networkmanager.servicec C s t j�dd�S )N�NETPLAN_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, d}t � rd}tj|g| tjtjd� d S )N�nmcliznetwork-manager.nmcli��stdoutr
)r r �
check_callr )�args� binary_namer r r r / s r c C s, zt dg� W dS ttjfy Y dS w )z"Check if NetworkManager is running�generalTF)r �OSErrorr �SubprocessErrorr r r r �
nm_running8 s
�r c C s� t �d�}t� }| D ]3}t|d��$}|D ]}|�|�}|r-|�tt�||�d���� nqW d � n1 s8w Y q
|S )Nz^interface-name=(.*)$�r� ) �re�compile�set�open�match�update�fnmatch�filter�group)�paths�devices�pat�
interfaces�path�f�line�mr r r �
nm_interfacesB s
�� ��r0 Fc C s" t � r
t| tg|�S t| tg|�S �N)r r r �NM_SERVICE_NAME)�action�syncr r r �systemctl_network_managerP s r5 c C s> t |�dkr d| g}|s|�d� |�|� t�|� d S d S )Nr r z
--no-block)�len�append�extendr r )r3 �servicesr4 �commandr r r r X s
�r c C s^ t � } tjg d�dd�}|�� D ] }|�� �d�}|d �� r,|d dvr,| �|d � q| S )N)�
networkctlz
--no-pagerz --no-legendT)�universal_newlines� r ���)� unmanaged�linger)r! r � check_output�
splitlines�strip�split� isnumeric�add)r+ �outr. �sr r r �networkd_interfacesd s �rI c C s t �ddg� d S )Nr; �reload�r r r r r r �networkctl_reloadn s rL c C s* t | �dkrt�ddgt| � � d S d S )Nr r; � reconfigure)r6 r r �list)r+ r r r �networkctl_reconfigurer s �rO c C s t �ddd| g�dkr
dS dS )z4Return True if at least one matching unit is runningr r z is-activer TF)r r )� unit_patternr r r �systemctl_is_activew s rQ c C s t �ddg� dS )zEReload systemd unit files from disk and re-calculate its dependenciesr z
daemon-reloadNrK r r r r �systemctl_daemon_reload~ s rR c C s t jddd| gt jt jd� dS )z8Flush all IP addresses of a given interface via iproute2�ip�addr�flushr N)r r r )�ifacer r r �
ip_addr_flush� s rW c
C s t j�d| �}|r\z6tt j�|d���#}|�� �� }|dkr/t�d| |� W d � W d S W d � n1 s9w Y W n ty[ } zt� d| t
|�� W Y d }~d S d }~ww zt j� t j�|dd��}t j� |�}W |S ty� } zt�d| |t
|�� W Y d }~d S d }~ww ) Nz/sys/class/net� operstate�downz'device %s operstate is %s, not changingz$Cannot determine operstate of %s: %s�device�driverz0Cannot replug %s: cannot read link %s/device: %s)
r r, �joinr" �readrC �logging�debug�IOError�error�str�realpath�basename)� interface� only_down�devdirr- �state�er[ � driver_namer r r �get_interface_driver_name� s4 � �������rk c C s& t �| ��t ji g�d }|�dd�S )Nr rT � )� netifaces� ifaddressesr �AF_LINK)re �linkr r r �get_interface_macaddress� s rq r+ c sR t � tj�sJ �� js
J �tt� fdd�| ��}t|�dkr%t�|� d S |d S )Nc s � j | t| �t| �d�S )N)�itf_name�
itf_driver�itf_mac)�match_interfacerk rq )�itf��netdefr r �<lambda>� s
�z%find_matching_iface.<locals>.<lambda>r r ) �
isinstance�np�
NetDefinition� has_matchrN r& r6 r^ �info)r+ rx �matchesr rw r �find_matching_iface� s
�
r� c @ sF e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dS )�NetplanCommandTFc C s� || _ || _|| _|| _d | _d| _d| _d | _i | _d | _ d | _
t j dt
jd |f |dd�| _| jjdddd � | jjd
dt jd � |sb| jjd d d
d�| _| jjdddd�}|j| jd� d S d S )NFz%s %sr T)�prog� description�add_helpz--debug�
store_truezEnable debug messages)r3 �helpz --breakpointzAvailable commandsrl �
subcommand)�title�metavar�destr� zShow this help message)r� r� )�func)�
command_idr� � leaf_command�testing�_argsr_ �
breakpoint� commandclass� subcommandsr� r� �argparse�ArgumentParser�sys�argv�parser� add_argument�SUPPRESS�add_subparsers�
subparsers�
add_parser� set_defaults� print_usage)�selfr� r� �leafr� �p_helpr r r �__init__� s> �
�
����zNetplanCommand.__init__c C s
|| _ d S r1 )r� )r� r r r r r$ � s
zNetplanCommand.updatec C sF | j j| j| d�\}| _| js | js!tdtjd� | �� d S d S d S )N)r � namespacez You need to specify a command��file) r� �parse_known_argsr� r� r� �printr� r
r� )r� �nsr r r �
parse_args� s
�zNetplanCommand.parse_argsc C sD | j r
| j �| j� | jrd| jv r| �� | jr t� | �� d S )Nr� )r� 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� �
print_helpr� r
�exitr �EX_USAGEr� r r r r� � s zNetplanCommand.print_usagec C s� |� }i | j |< || j | d< || j | d< |jr"tj�dd �s"d S | jj|j|j|jdd�}|j |j
|d� || j | d< d S )N�class�instance�ENABLE_TEST_COMMANDSF)r� r� r� )r� r� r� ) r� r� r r r r� r� r� r� r� �run)r� �namer� r� �pr r r �_add_subparser_from_class� s
�z(NetplanCommand._add_subparser_from_classc C s@ dd l }|�|�D ]\}}|�|�r t|t�r | �||� q d S )Nr )�inspect�
getmembers�isclass�
issubclassr� r� )r� �
submodulesr� r� �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 r� � s
r� )F)%r� r r^ r� r rm r% r �netplan.libnetplan�
libnetplanr{ �netplan.configmanagerr r r2 r � RuntimeError�
config_errorsr
r r r r0 r5 r rI rL rO rQ rR rW rk rq rN r� � Namespacer� r r r r �<module> s>