U
K��^�= � @ s� d Z ddlZddlmZmZ ddlmZmZ ddlm Z ddl
m Z m Z m
Z
ddlmZ ddlmZ dd lmZ dd
lmZmZmZ dd lmZmZmZm Z dd l m Z d
dl m Z m!Z!m"Z"m#Z# d
dl$m%Z% ddl&m'Z' e�(d�Z)dd� Z*G dd� de'�Z+dS )zS Network Model
Provides network device listings and extended network information
� N)�connect_signal�Text)� addr_version� NetDevAction)�
ActionMenu)�back_btn�done_btn�menu_btn)�Pile)�Spinner)�StretchyOverlay)�ColSpec� TablePile�TableRow)� button_pile�Color�make_action_menu_row�screen)� widget_width� )�AddVlanStretchy� BondStretchy�EditNetworkStretchy�ViewInterfaceInfo)� NetworkConfigureWLANStretchy)�BaseViewzsubiquitycore.views.networkc s � �fdd�}d|_ |S )Nc s2 �| |f� �� }|� | jj�|�� | �|� d S �N)�attach_context�
controller�context�child�show_stretchy_overlay)�self�name�device�stretchy��args�cls� �6/usr/share/subiquity/subiquitycore/ui/views/network.py�implB s z _stretchy_shower.<locals>.implT)� opens_dialog)r( r' r+ r) r& r* �_stretchy_showerA s r- c s� e Zd Zed�Zed�Z� fdd�Zee�Z ee
�Z ee d�Z
ee d�Zee�Zee�Zdd� Zd d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z dd � Z d*d d �Z d d!� Z d+d"d#�Z d,d$d%�Z!d-d&d'�Z"d.d(d)�Z#� Z$S )/� NetworkViewzNetwork connectionsz�Configure at least one interface this server can use to talk to other machines, and which preferably provides sufficient access for updates.c s� || _ || _i | _g | _tddd�| _tdd�tdd�tddd �d
�| _t| � � �| _
t t d �| j
d �| _t| jg�}d
|_| j
|g}ttd| jd �tt d�| jd �g�| _td| jfg�| _| jj�| j� d| _t� � t || jdt | j �d�� d S )N� �center��alignr )�rpad� )Z min_widthT)Z
can_shrinkr3 )r � � z Create bond)Zon_press�leftZTBDZBack�packF)�rows�buttonsZ
focus_buttons�excerpt) �modelr � dev_to_table� cur_netdevsr �errorr
�device_colspecsr
�_build_model_inputs� device_piler �_� _create_bondZ_create_bond_btnr r2 r �doner �cancelr: �bottom�network_event_receiverZadd_default_route_watcher�_route_watcher�
error_showing�super�__init__r r; )r" r< r Zbpr9 �� __class__r) r* rL P sH
� � � ����zNetworkView.__init__r6 � c C s~ | j j�|��f t� }|jdkrB|jd D ]}|�| j�|�� q*d |_| � |� |D ]}| �
|� qV| j � � W 5 Q R X d S )N�bond�
interfaces) r r r �set�type�config�addr< Zget_netdev_by_name�del_link� update_linkZ apply_config)r" r# r$ Z touched_devs�devr) r) r* �_action_DELETE� s
zNetworkView._action_DELETEc C s" |\}}|d� |j|j�|� d S )Nz{}/{})�formatr# )r" �sender�actionr$ �methr) r) r* �_action� s zNetworkView._actionc C s` t �d|� |rtd�}ntd�}| jjd �|� tdt| jjd �t| jjd ��| j_d S )Nzview route_watcher %sZDonezContinue without networkr � r ) �log�debugrC r: � base_widgetZ set_label�maxr �width)r" Zroutes�labelr) r) r* rI � s
�zNetworkView._route_watcherc C sd t | jjj�}|�� ttttd��|g�gdd�}|| j �
� ftd�| j �
� fg| j j dd�< d S )NzApplying changesr0 r1 r/ r ) r r �app�aio_loop�startr r r rC rG �options�contents)r" �s�cr) r) r* �show_apply_spinner� s
��� �z NetworkView.show_apply_spinnerc C s$ t | jj�dkr g | jjdd�< d S )N� r )�lenrG rj )r" r) r) r* �hide_apply_spinner� s z NetworkView.hide_apply_spinnerc C s� g }|j dkr$|jjs$|�td�� | j�� D ]@}|j dkr>q.|j|j� dg �kr.|�td�j
|jd�� qpq.|r�d� |�}nd}|S ) NZethz
not connectedrP rQ zenslaved to {device})r$ �, �-) rS �infoZ is_connected�appendrC r< �get_all_netdevsr# rT �getrZ �join)r" rX Znotes�dev2r) r) r* �_notes_for_device� s
� z NetworkView._notes_for_devicec
s� g }|� � }dD �]N}|�|��rtdj|d��� |�|�}|rZ|�� fdd�|D �� n�|�|�dkr�t| jj j
dd�}d |_ |� � |�
� |f� nl|�|�d
kr�|�
� ttd ��f� nF|�|�d kr�|�
� td
�f� n$|�
� ttdj|�|�d���f� qg }|j�dg �D ]"}t|�|k�r|�
t|�� �q|r|�
ttd��td�|��f� qt|�dk�r�|j�s�|j}|d k�r�d}|�
ttd��t|�f� g } |D ]&\� }
| �
ttd�� d|
fg�� �q�| S )N)r6 rO zDHCPv{v})�vc s g | ]}� t |�f�qS r) )r )�.0Zaddr�re r) r* �
<listcomp>� s z8NetworkView._address_rows_for_device.<locals>.<listcomp>ZPENDINGr7 r1 g333333�?ZTIMEDOUTz timed outZ RECONFIGURErr zunknown state {state})�stateZ addressesZstaticrq r r/ Zdisabledrn )�dhcp_addressesZ dhcp_enabledr rZ rv �extendZ
dhcp_stater r rf rg Zraterh rt rC rT r �strrw ro Zis_usedZdisabled_reasonr ) r" rX Z address_infor rz Zaddrsrk Zip�reasonr9 �valuer) r| r* �_address_rows_for_device� sZ
�����
z$NetworkView._address_rows_for_devicec C s� t �d|j|j|| jk� || jkr2| �|� d S t| j�D ] \}}|j|jkr<|} qdq<t| j�}| � ||�}|| j
� d�fg| j
j |d |d �< d S )Nznew_link %s %s %sr8 r )
r` ra r# �ifindexr> r= rW � enumeratero �_device_widgetrB ri rj )r" Znew_dev�iZcur_dev�netdev_i�wr) r) r* �new_link� s" �
�zNetworkView.new_linkc C s� t �d|j|j|| jk� || jkr(d S | j| }|jd j}|jd d � |j� |jd d � |j
� |jd d � | � |�� |� dt
|j�� |�d| �|�� d S )Nzupdate_link %s %s %sr r rn r5 )r` ra r# r� r> r= Z
table_rowsrb Zcells�set_textrS ry Z remove_rowsro Z insert_rowsr� )r" rX Z old_tableZ first_rowr) r) r* rW s �
zNetworkView.update_linkc C sd t | jj�d }| jj|k}| jj|= |r6| j�� n*| jj�� sT| j jd7 _q6| jj�� d S )Nr )ro rB rj Zfocus_positionZ_select_last_selectableZfocusZ
selectableZ_select_first_selectable)r" r� Z
new_lengthZrefocusr) r) r* � _remove_row s
zNetworkView._remove_rowc C s� t �d|j|j|| jk� |jr.|jd k r.d S || jkrb| j�|�}| �|d � | j|= | j |= t
| j t �r�| j j
}t|dd �|kr�| �� d S )Nzdel_link %s %s %sr r$ )r` ra r# r� r> Z
is_virtualrT �indexr� r= �
isinstanceZ_wr r% �getattrZremove_overlay)r" rX r� r% r) r) r* rV + s" �
zNetworkView.del_linkNc C s� |d krt | j�}|g| j||�< g }tD ]D}t| d|j �}t|dd�}|�|�r*|�|�� d||f|f� q*t|�}t |d| j
|� t t d�t |j�t |j
�t | �|�dd�|t d �g|�g| �|� }t|| jd
d �} | | j|< | �| j� |j
d k�rtd
�jf |j�}
nF|j
dk�r<td�jd�|jd �d�}
n d�|jj|jj|jj g�}
t d| fdt � t d|
��fdt d�fg�S )NZ_action_r, FTr\ �[Zclip)Zwrap�]rn )�colspecs�spacingZvlanz VLAN {id} on interface {link}rP z bond master for {interfaces}rq rQ )rQ z / r8 z r/ ) ro r> r r� r# Zsupports_actionrt r� r r r^ r r rS ry r� r r@ r= Zbind�
heading_tablerC rZ rT rw rs ZhwaddrZvendorr< r
r �
info_minor) r" rX r� Zactionsr\ r] r, ZmenuZtrows�tablers r) r) r* r� = s\
����
� � �
�zNetworkView._device_widgetc C sN t tdd� dD ��gd| jd�| _| jg}| j�� D ]}|�| �|�� q4|S )Nc S s g | ]}t �t|���qS r) )r r� r )r{ �headerr) r) r* r} u s z3NetworkView._build_model_inputs.<locals>.<listcomp>)r/ �NAMEZTYPEZNOTESr/ rn )r� r� )r r r@ r� r< ru rt r� )r" r9 rX r) r) r* rA s s �� �z NetworkView._build_model_inputsc C s* t | �}|�| jj�d�� | �|� d S )NZadd_bond)r r r r r r! )r" r[ r% r) r) r* rD � s zNetworkView._create_bondc C s� d| _ t�| j�| j�� ftd�| j�� fg| jjdd�< |dkr`|d }| j�d|j f � nz|dkrv| j�d� nd|dkr�| j�d � nN|d
kr�| j�d � n8|d kr�| j�d
� n"|dkr�| j�d� n | j�d� d S )NTr/ r z
stop-networkdz$Stopping systemd-networkd-failed: %rZapplyzHNetwork configuration could not be applied; please verify your settings.Ztimeoutz=Network configuration timed out; please verify your settings.Zdownz"Downing network interfaces failed.zadd-vlanzFailed to add a VLAN tag.zrm-devz%Failed to delete a virtual interface.z>An unexpected error has occurred; please verify your settings.)
rJ r Z
info_errorr? rG ri r rj r� �stderr)r" r\ rs �excr) r) r* �show_network_error� s( �
�z NetworkView.show_network_errorc C s4 | j rg | jjdd�< | jj�| j� | j�� d S )Nr rn )rJ rG rj r rH � remove_default_route_watcherrI rE )r" �resultr) r) r* rE � s �zNetworkView.donec C s | j j�| j� | j �� d S r )r rH r� rI rF )r" Zbuttonr) r) r* rF � s �zNetworkView.cancel)N)N)N)N)N)%�__name__�
__module__� __qualname__rC �titler; rL r- r Z _action_INFOr Z_action_EDIT_WLANr Z_action_EDIT_IPV4Z_action_EDIT_IPV6r Z_action_EDIT_BONDr Z_action_ADD_VLANrY r^ rI rm rp ry r� r� rW r� rV r� rA rD r� rE rF �
__classcell__r) r) rM r* r. J s4 ,
1
6
r. ),�__doc__ZloggingZurwidr r Z subiquitycore.models.networkr r Zsubiquitycore.ui.actionmenur Zsubiquitycore.ui.buttonsr r r Zsubiquitycore.ui.containerr
Zsubiquitycore.ui.spinnerr Zsubiquitycore.ui.stretchyr Zsubiquitycore.ui.tabler
r r Zsubiquitycore.ui.utilsr r r r Zsubiquitycore.ui.widthr Z"network_configure_manual_interfacer r r r Z network_configure_wlan_interfacer Zsubiquitycore.viewr Z getLoggerr` r- r. r) r) r) r* �<module> s"
|