File: //usr/share/netplan/netplan_cli/cli/__pycache__/utils.cpython-310.pyc
o
t��f�) � @ sd d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZddlm Z d dl
m Z m Z dZ
dZe e efZdd� Zd d
� Zd d � Zd
edefdd�Zdd� Zdd� Zdedefdd�Zdeddfdd�Zd<d d �Zd<d ed ed e fd!d"�Z d#d$� Z d%d&� Z d'd(� Z d)d*� Z!d+d,� Z"d-d.� Z#d/d0� Z$d1d2� Z%d<d3d4�Z&d5d6� Z'd7efd8d9�Z(G d:d;� d;ej)�Z*dS )=� N� )�ConfigurationError)�
NetDefinition�NetplanExceptionzNetworkManager.servicez+snap.network-manager.networkmanager.servicec C s t j�dd�S )N�NETPLAN_GENERATE_PATHz /usr/libexec/netplan/generate)�os�environ�get� r
r
�+/usr/share/netplan/netplan_cli/cli/utils.py�get_generator_path% s r c C s t jdddtgt jd�dkS )N� systemctl�--quiet�
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��text)r � check_outputr r
r
r � nmcli_out3 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_running9 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_interfacesC s
�� ��r9 � interfacec C s< t dddddd| g�}|�� �d�}|d }|d kr |S d
S ) Nz-m�tabularz-fzGENERAL.CONNECTION�device�show�
r' z--� )r! �strip�split)r: �output�lines�
connectionr
r
r � nm_get_connection_for_interfaceQ s rE rD c C s, z
t dd| g� W d S tjy Y d S w )NrD �up)r r �CalledProcessError)rD r
r
r �nm_bring_interface_upX s
�rH Fc C s" t � r
t| tg|�S t| tg|�S �N)r r
r �NM_SERVICE_NAME)�action�syncr
r
r �systemctl_network_manager_ s rM rK �servicesrL c C s> t |�dkr d| g}|s|�d� |�|� t�|� d S d S )Nr' r
z
--no-block)�len�append�extendr r )rK rN rL �commandr
r
r r
g s
�r
c C s^ t � } tjg d�dd�}|�� D ] }|�� �d�}|d �� r,|d dvr,| �|d � q| S )N)�
networkctlz
--no-pagerz --no-legendTr � r ���)� unmanaged�linger)r* r r �
splitlinesr@ rA � isnumeric�add)r4 �outr7 �sr
r
r �networkd_interfacess s �r] c C s t �ddg� d S )NrS �reload�r r r
r
r
r �networkctl_reload} s r` c C s* t | �dkrt�ddgt| � � d S d S )Nr' rS � reconfigure)rO r r �list)r4 r
r
r �networkctl_reconfigure� s �rc 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_active� s re c C s8 t jdd| gt jt jdd�}|jdkrd|jv rdS dS )z5Return True if output is "masked" or "masked-runtime"r
r T�r r r r �maskedF)r �run�PIPE�
returncoder �rd �resr
r
r �systemctl_is_masked� s �rm c C s. t jdd| gt jt jdd�}|jdkrdS dS )z5Return True if returncode is other than a failure (1)r
r Trf r' F)r rh ri rj rk r
r
r �systemctl_is_installed� s �
rn c C s t �ddg� dS )zEReload systemd unit files from disk and re-calculate its dependenciesr
z
daemon-reloadNr_ r
r
r
r �systemctl_daemon_reload� s ro c C s t jddd| gt jt jd� dS )z8Flush all IP addresses of a given interface via iproute2�ip�addr�flushr Nr )�ifacer
r
r �
ip_addr_flush� s rt 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: %sr< �driverz0Cannot replug %s: cannot read link %s/device: %s)
r r5 �joinr+ �readr@ �logging�debug�IOError�error�str�realpath�basename)r: � only_down�devdirr6 �state�erw � driver_namer
r
r �get_interface_driver_name� s4 � �������r� c C s& t �| ��t ji g�d }|�dd�S )Nr rq r? )� netifaces� ifaddressesr �AF_LINK)r: �linkr
r
r �get_interface_macaddress� s r� r4 c sP t � t�sJ �� js J �tt� fdd�| ��}t|�dkr$t�|� d S |d S )Nc s � j | t| �t| �d�S )N)�
iface_name� iface_driver� iface_mac)�_match_interfacer� r� )�itf��netdefr
r �<lambda>� s
�z%find_matching_iface.<locals>.<lambda>r' r )�
isinstancer �
_has_matchrb r/ rO rz �info)r4 r� �matchesr
r� 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)rK �helpz --breakpointzAvailable commandsr? �
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 rI )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� rh )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 rz r� r r� r. r( �
configmanagerr �netplanr r rJ r � RuntimeError�
config_errorsr r r rb r~ r! r% r9 rE rH rM �boolr
r] r` rc re rm rn ro rt r� r� r� � Namespacer� r
r
r
r �<module> sF