U
K��^_1 � @ s� d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZ d dl m
Z
ddddd d
d d gZ d
dgZ e�
d�Zdd� ZG dd� de j�ZG dd� d�ZG dd� de�ZG dd� de�ZdS )� N)�AF_INET�AF_INET6)�pgettext)�netplan�loZbridgeZtunZtapZdummyZsitZcanz???�vlan�bondz subiquitycore.models.networkc C s t �| �jS �N)� ipaddressZ ip_interface�version)�ip� r
�4/usr/share/subiquity/subiquitycore/models/network.py� addr_version" s r c @ sZ e Zd Zed d�Zed d�Zed d�Zed d�Zed d�Zed d�Z ed d�Z
dd � Z d
S ) � NetDevActionZInfoz Edit Wifiz Edit IPv4z Edit IPv6z Edit bondzAdd a VLAN tagZDeletec C s t t| �j| j�S r )r �type�__name__�value��selfr
r
r �str0 s zNetDevAction.strN) r �
__module__� __qualname__r �INFOZ EDIT_WLANZ EDIT_IPV4Z EDIT_IPV6Z EDIT_BONDZADD_VLANZDELETEr r
r
r
r r & s
r c @ sD e Zd ZdddddddgZdddhZdd d
d d gZdhZd
dgZdS )�BondParametersz
balance-rrz
active-backupz balance-xorZ broadcastz802.3adz balance-tlbz balance-albZlayer2zlayer2+3zlayer3+4zencap2+3zencap3+4ZslowZfastN)r r r ZmodesZsupports_xmit_hash_policyZxmit_hash_policiesZsupports_lacp_rateZ
lacp_ratesr
r
r
r r 4 s. � �� ��r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zed d � �Z e j
d
d � �Z dd� Z edd� �Z dd� Z
edd� �Zedd� �Zedd� �Zedd� �Zed d � �Zd Zed d � �Zd Zd Zed!d � �Zed"d � �Zed#d � �Zd$d%� Zd&d'� Zd(S ))�
NetworkDevc C s: || _ || _|| _i | _d | _d | _i | _d d d�| _d S )N�� � )�_model�_namer �config�infoZdisabled_reasonZ dhcp_events� _dhcp_state)r �model�name�typr
r
r �__init__\ s �zNetworkDev.__init__c C sj g g d�}| j d k rf| j j�� D ]D}|jtkr4d}n|jtkr d}nq |jdkr || �t|j �� q |S )Nr r r Zdhcp)
r" � addresses�valuesZfamilyr r �source�appendr �address)r �r�a�vr
r
r �dhcp_addressesi s
zNetworkDev.dhcp_addressesc C s( | j d krdS | j �dj|d�d�S d S )NF�dhcp{v}�r/ )r! �get�format�r r r
r
r � dhcp_enabledw s
zNetworkDev.dhcp_enabledc C s$ | j �dj|d�d�sd S | j| S )Nr1 r2 F)r! r3 r4 r# r5 r
r
r �
dhcp_state} s zNetworkDev.dhcp_statec C s || j |< d S r )r# )r r �stater
r
r �set_dhcp_state� s zNetworkDev.set_dhcp_statec C s | j S r )r r r
r
r r% � s zNetworkDev.namec C s� || j krv| jrv| jd k rv|| jjkr:tdj| j |d���| | jj|< t| j | j � }| jj| j < d |_
| j|_d | _|| _ d S )Nz#renaming {old_name} over {new_name})Zold_name�new_name) r% �
is_virtualr" r$ �devices_by_name� RuntimeErrorr4 r r r r! r )r r: Z dead_devicer
r
r r% � s" �� �c C s t | d|j �S )NZ
_supports_)�getattrr% )r �actionr
r
r �supports_action� s zNetworkDev.supports_actionc C s4 | j �di ��� D ] \}}|�d�}||f S dS )N�
access-points�password)NN)r! r3 �items)r �ssidZsettings�pskr
r
r �configured_ssid� s
zNetworkDev.configured_ssidc C s> | j �di �}|�� |d k r:i ||< |d k r:||| d< d S )NrA rB )r! �
setdefault�clear)r rD rE Zapsr
r
r � set_ssid_psk� s zNetworkDev.set_ssid_pskc C s | j d k r| j jS d S d S r )r" �ifindexr r
r
r rJ � s
zNetworkDev.ifindexc C s
| j tkS r )r �"NETDEV_ALLOWED_VIRTUAL_IFACE_TYPESr r
r
r r; � s zNetworkDev.is_virtualc C s8 | j �� D ](}|jdkr
| j|j�dg �kr
dS q
dS )Nr �
interfacesTF�r �get_all_netdevsr r% r! r3 �r �devr
r
r �
is_bond_slave� s
zNetworkDev.is_bond_slavec C sZ | j �� D ]J}|jdkr2| j|j�dg �kr2 dS |jdkr
| j|j�d�kr
dS q
dS )Nr rL Tr �linkFrM rO r
r
r �is_used� s
zNetworkDev.is_usedc C s dd� t | jj�� �D �S )Nc S s g | ]\}}|j d kr|j�qS )�global)Zscoper )�.0�_Zaddrr
r
r �
<listcomp>� s
�z9NetworkDev.actual_global_ip_addresses.<locals>.<listcomp>)�sortedr" r( rC r r
r
r �actual_global_ip_addresses� s z%NetworkDev.actual_global_ip_addressesTc C s
| j dkS )N�wlan�r r r
r
r �<lambda>� � zNetworkDev.<lambda>c C s
| j dkS )Nr r[ r r
r
r r\ � r] c C s | j dko| j S )Nr )r rQ r r
r
r r\ � r] c C s | j o | j S r )r; rS r r
r
r r\ � r] c C s| | j �dj|d�d � | j �dj|d�d � g }| j �dg �D ]}t|�|kr>|�|� q>|rj|| j d< n| j �dd � d S )Nr1 r2 �
gateway{v}r( )r! �popr4 r3 r r+ )r r Zaddrsr r
r
r � remove_ip_networks_for_version� s z)NetworkDev.remove_ip_networks_for_versionc C s� |d � d�d }|d|d � d�d 7 }| j�dg ��|� dj|d�}|d rd|d | j|< n| j�|d � | j�d
i �}|d
r�|�dg ��|d
� |d r�|�d g ��|d � d S )
Nr, �/r �network� r( r^ r2 ZgatewayZ nameserversZ
searchdomains�search)�splitr! rG r+ r4 r_ �extend)r r rb r, Zgwkey�nsr
r
r � add_network� s zNetworkDev.add_networkN) r r r r' r0 r6 r7 r9 �propertyr% �setterr@ rF rI rJ r; rQ rS rY Z_supports_INFOZ_supports_EDIT_WLANZ_supports_EDIT_IPV4Z_supports_EDIT_IPV6Z_supports_EDIT_BONDZ_supports_ADD_VLANZ_supports_DELETEr` rh r
r
r
r r Z sH
�� r c @ st e Zd ZdZd dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
d dd�Z dd� Z dd� Z
dd� Zdd� Zd S ) � NetworkModel� Tc C s || _ i | _d| _|| _d S )NF)� support_wlanr< Z has_network�project)r rn rm r
r
r r' s zNetworkModel.__init__c C s t �� | _| j�|� d S r )r ZConfigr! Zload_from_root)r Z netplan_rootr
r
r �parse_netplan_configs s
z"NetworkModel.parse_netplan_configsc C s� t �d||j|j� |jtkr"d S | js6|jdkr6d S |jrJ|jtkrJd S | j� |j�}|d k rt|j
d k rlq�||_
n\| j � |�}|jr�|s�d S t
| |j|j�}||_
||_ t �d||jt�|j �� || j|j< |S )Nznew_link %s %s %srZ z new_link %s %s with config %s)�log�debugr% r �NETDEV_IGNORED_IFACE_TYPESrm r; rK r< r3 r" r! Zconfig_for_devicer r Zsanitize_interface_config)r rJ rR rP r! r
r
r �new_link
s6
�
� zNetworkModel.new_linkc C s* | j �� D ]\}}|j|kr
| S q
d S r )r< rC rJ �r rJ r% rP r
r
r � update_link+ s
zNetworkModel.update_linkc C sR | j �� D ]B\}}|j|kr
d |_|jr<|jd krD| j |= n| j |= | S q
d S r )r< rC rJ r" r; r! rt r
r
r �del_link0 s
zNetworkModel.del_linkc C s8 dj |j|d�}t| |d� }| j|< |j|d�|_|S )Nz {name}.{tag})r% �tagr )rR �id)r4 r% r r< r! )r Zdevicerw r% rP r
r
r �new_vlan@ s �zNetworkModel.new_vlanc C s& t | |d� }| j|< ||d�|_|S )Nr )rL Z
parameters)r r< r! )r r% rL ZparamsrP r
r
r �new_bondI s
�zNetworkModel.new_bondFc C s. dd� t | j�� �D �}|s*dd� |D �}|S )Nc S s g | ] \}}|�qS r
r
)rU �kr/ r
r
r rW R s z0NetworkModel.get_all_netdevs.<locals>.<listcomp>c S s g | ]}|j d k r|�qS r )r! )rU r/ r
r
r rW T s
)rX r< rC )r Zinclude_deletedZdevsr
r
r rN Q s z NetworkModel.get_all_netdevsc C s
| j | S r )r< )r r% r
r
r �get_netdev_by_nameW s z NetworkModel.get_netdev_by_namec C s d� d�| j�tj|dd�g�S )N�
z,# This is the network config written by '{}'F)Zdefault_flow_style)�joinr4 rn �yaml�dump)r r! r
r
r �stringify_configZ s � �z NetworkModel.stringify_configc C s^ dddii}ddddd�}| � � D ]6}||j }|d �|i �}|jsL|jr"|j||j< q"|S ) Nrb r � Z ethernetsZbondsZwifisZvlans)Zethr rZ r )rN r rG r! rS r% )r r! Z type_to_keyrP �keyZconfigsr
r
r �
render_configa s ���
zNetworkModel.render_configc C s$ dd| � | �� �d�ddd�d�iS )NZ write_filesz$etc/netplan/00-installer-config.yaml)�pathZcontentz@etc/cloud/cloud.cfg.d/subiquity-disable-cloudinit-networking.cfgz network: {config: disabled}
)Zetc_netplan_installerZnonet)r� r� r r
r
r �renderu s ����zNetworkModel.renderN)T)F)r r r �__doc__r' ro rs ru rv ry rz rN r| r� r� r� r
r
r
r rk � s
!
rk )�enumr
Zloggingr Zsocketr r Zsubiquitycore.gettext38r Z
subiquitycorer rr rK Z getLoggerrp r �Enumr r �objectr rk r
r
r
r �<module> s. �
& $ |