o
K��^sE � @ s4 d Z ddlZddlmZmZ ddlmZmZ ddlm Z ddl
m Z m Z m
Z
ddlmZmZ ddlmZ dd lmZ dd
lmZmZmZ dd lmZmZm Z m Z m 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-e.d�e.d�e.d�e.d�d�Z/G dd � d e)�Z0dS ) zS Network Model
Provides network device listings and extended network information
� N)�connect_signal�Text)� DHCPState� NetDevAction)�
ActionMenu)�back_btn�done_btn�menu_btn)�Pile�
WidgetWrap)�Spinner)�StretchyOverlay)�ColSpec� TablePile�TableRow)� button_pile�Color�make_action_menu_row�rewrap�screen)� widget_width� )�AddVlanStretchy� BondStretchy�EditNetworkStretchy�ViewInterfaceInfo)� NetworkConfigureWLANStretchy)�BaseViewzsubiquitycore.views.networkc s � �fdd�}d|_ |S )Nc s4 �| |g� �R � }|� | 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�implD s z _stretchy_shower.<locals>.implT)� opens_dialog)r* r) r- r+ r( r, �_stretchy_showerC s r/ c s< e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Z� ZS ) �NetworkDeviceTablec s || _ || _t� �| �� � d S r )�parent�dev_info�super�__init__�_create)r$ r1 r2 �� __class__r+ r, r4 M s zNetworkDeviceTable.__init__c C st g }t D ]$}t| jd|j �}t|dd�}|| jjv r(|�|�� d||f|f� qt|�}t |d| jj
| � t t d�t | jj�t | jj
�t | �� dd�|t d �g|�g| �� }t|| jjd
d �| _| j�| jj� | jj
d kr�td
�j| jjj| jjjd�}n#| jj
dkr�td�jd�| jjj �d�}nd�| jj | jj | jj g�}t d| jfdt!�"t d| ��fdt d�fg�S )N�_action_r. FT�action�[�clip)�wrap�]� )�colspecs�spacing�vlanz VLAN {id} on interface {link})�id�link�bondz bond master for {interfaces}�, )�
interfacesz / �packz � )#r �getattrr1 r% r2 �enabled_actions�append�strr r �_actionr r �type�_notes�
_address_rowsr �device_colspecs�table�bind�
heading_table�_�formatrA rB rC �joinrD rF �hwaddr�vendor�modelr
r �
info_minor)r$ �actionsr9 �methr. �menu�trows�infor+ r+ r, r5 R sZ ��
���
� � ��
�zNetworkDeviceTable._createc C s� g }| j jdkr&| j jj}|jd ur |�tdj|jd��� n|�td�� | j js1|�td�� | j j rB|�td�j| j j d�� |rKd�
|�}|S d}|S ) N�wlanz ssid: {ssid})�ssidz
not connectedzenslaved to {device})r& rE �-) r2 rN ra �configrb rK rU rV � is_connected� bond_masterrW )r$ �notesrd r+ r+ r, rO � s&
��
�zNetworkDeviceTable._notesc
s� g }d| j j| j jfd| j j| j jffD ]w\}}}|jrwtdj|d��� |j}|r6|� � fdd�|D �� q|j
t j krUt
| jjjjdd�}d |_|�� |�� |f� q|j
t jkrg|�� ttd
��f� q|j
t jkrv|�� td �f� q|jr�|�ttd ��td
�|j��f� qt|�dkr�| j js�| j j}|d u r�d}|�ttd��t|�f� g }|D ]\� } |�t td�� d| fg�� q�|S )N� � zDHCPv{v})�vc s g | ]}� t |�f�qS r+ )r )�.0�addr��labelr+ r, �
<listcomp>� s z4NetworkDeviceTable._address_rows.<locals>.<listcomp>�left��aligng333333�?z timed outrc �staticrE r rH �disabledr> ) r2 �dhcp4�static4�dhcp6�static6�enabledr rV � addresses�extend�stater �PENDINGr r1 r �app�aio_loop�rate�startrK � TIMED_OUTrU � RECONFIGURErW �len�is_used�disabled_reasonr )
r$ � address_inforj � dhcp_status�
static_config�addrs�s�reason�rows�valuer+ rm r, rP � sL �� � �
�� z NetworkDeviceTable._address_rowsc C s� || _ | jjd j}|jd d �|j� |jd d �|j� |jd d �| �� � | j� dt
| jj�� | j� d| � � � d S )Nr r r> � )
r2 rR �
table_rows� base_widget�cells�set_textr% rN rO � remove_rowsr� � insert_rowsrP )r$ r2 � first_rowr+ r+ r, �update� s zNetworkDeviceTable.update) �__name__�
__module__� __qualname__r4 r5 rO rP r� �
__classcell__r+ r+ r6 r, r0 L s 6%r0 zRA wifi device was detected but the necessary support packages were not
available.
z+Wifi support packages are being installed.
z@Wifi support packages failed to install. Please check the logs.
z>Wifi support packages will be installed in the target system.
)�
NOT_AVAILABLE�
INSTALLING�FAILED�DONEc s� e Zd Zed�Zed�Z d.� fdd� Zdd� Zdd � Zd
e_ e
e �Z e
e
d �Ze
e
d �Ze
e�Zd
d� Zd
e_ dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Z d/d$d%�Z d&d'� Z d/d(d)�Z d/d*d+�Z d/d,d-�Z!� Z"S )0� NetworkViewzNetwork connectionsz�Configure at least one interface this server can use to talk to other machines, and which preferably provides sufficient access for updates.�
NOT_NEEDEDc s( || _ i | _g | _tddd�| _tdd�tdd�tddd �d
�| _ttd d � d
D ��gd| jd�| _ t
| j g�| _ |D ]}| � |� q;t
td�| jd�| _t| jg�}d|_| j |g}ttd| jd�ttd�| jd�g�| _t
d| jfg�| _d| _d| _| �|� t � � t || jdt| j �d�� d S )NrH �centerrq r )�rpad� )� min_widthT)�
can_shrinkr� )r r� rh c S s g | ] }t �t|���qS r+ )r r[ r )rk �headerr+ r+ r, ro � s �z(NetworkView.__init__.<locals>.<listcomp>)rH �NAME�TYPE�NOTESrH r> )r@ r? z Create bond)�on_pressrp �TBD�BackrG F)r� �buttons�
focus_buttons�excerpt) r �dev_name_to_table�cur_netdev_namesr �errorr rQ r r rT r
� device_pile�new_linkr rU � _create_bond�_create_bond_btnr rr r �doner �cancelr� �bottom�"wlan_support_install_state_showing�
error_showing�%update_for_wlan_support_install_stater3 r4 r r� )r$ r � netdev_infos�wlan_support_install_stater2 �bpr� r6 r+ r, r4 � sT
����
� � ��
�zNetworkView.__init__c � sR �| j jj| j �|�ddd�I d H }t| ||�}|�| j j�d�� | �|� d S )Nz Loading infoT)�
can_cancel�INFO) r r~ �wait_with_text_dialog�get_info_for_netdevr r r! r" r# )r$ r% r` r' r+ r+ r, �
_show_INFO" s � � zNetworkView._show_INFOc C s | j jj�| �|j�� d S r )r r~ r � create_taskr� r% �r$ r% r2 r+ r+ r, � _action_INFO* s
�zNetworkView._action_INFOTrh ri c C s2 t | || �� �}|�| jj�|�� | �|� d S r �r � get_candidate_bond_member_namesr r r! r" r# )r$ r% r2 r' r+ r+ r, �_action_EDIT_BOND4 �
�z NetworkView._action_EDIT_BONDc C sL | j j�|�� | j �|j� | �|� W d � d S 1 s w Y d S r )r r! r" � delete_linkr% �del_linkr� r+ r+ r, �_action_DELETE< s "�zNetworkView._action_DELETEc C s( |\}}|j }|d�|j|j�|� d S )Nz{}/{})r2 rV r% )r$ �senderr9 � netdev_tabler] r2 r+ r+ r, rM A s zNetworkView._actionc C s` t �d|� |r
td�}ntd�}| jjd �|� tdt| jjd �t| jjd ��| j_d S )Nzview route_watcher %s�DonezContinue without networkr � r ) �log�debugrU r� r� � set_label�maxr �width)r$ �routesrn r+ r+ r, �update_default_routesF s
�z!NetworkView.update_default_routesc C sd t | jjj�}|�� ttttd��|g�gdd�}|| j �
� ftd�| j �
� fg| j j dd�< d S )NzApplying changesr� rq rH r ) r r r~ r r� r r r rU r� �options�contents)r$ r� �cr+ r+ r, �show_apply_spinnerS s
��� �z NetworkView.show_apply_spinnerc C s( t | jj�dkrg | jjdd�< d S d S )Nr> r )r� r� r� )r$ r+ r+ r, �hide_apply_spinnera s �z NetworkView.hide_apply_spinnerc C s� t �d|j|j| jv � |j| jv r| �|� d S | j�|j� | j�� | j�|j�}t | |�}|| j|j< || j
� d�fg| j
j |d |d �< d S )Nznew_link %s %srG r )
r� r� r% r� r� � update_linkrK �sort�indexr0 r� r� r� )r$ � new_dev_info�netdev_i� device_tabler+ r+ r, r� e s �
�zNetworkView.new_linkc C sf t | jt�rt| jjd�r| jj�|� t�d|j|j| j v � |j| j vr(d S | j
|j � |� d S )Nr� zupdate_link %s %s) �
isinstance�_wr
�hasattrr' r� r� r� r% r� r� r� )r$ r2 r+ r+ r, r� t s � zNetworkView.update_linkc C sp t | jj�d }| jj|k}| jj|= |r | j�� d S | jj�� s0| j jd7 _| jj�� r"| jj�� d S )Nr )r� r� r� �focus_position�_select_last_selectable�focus�
selectable�_select_first_selectable)r$ r� �
new_length�refocusr+ r+ r, � _remove_row s
�zNetworkView._remove_rowc C s� t �d|j|j| jv � |jr|jrd S |j| jv r1| j�|j�}| �|d � | j|= | j|j= t | j
t �rI| j
j }t
|dd �|u rK| �� d S d S d S )Nzdel_link %s %sr r& )r� r� r% r� �
is_virtual�
has_configr� r� r� r� r� r
r' rI �remove_overlay)r$ r2 r� r' r+ r+ r, r� � s" �
�zNetworkView.del_linkc C sB g }| j �� D ]}|j}|jdv rq|jd urq|�|j� q|S )N)rA rD )r� �valuesr2 rN rf rK r% )r$ �namesrR r2 r+ r+ r, r� � s
z+NetworkView.get_candidate_bond_member_namesNc C s2 t | d | �� �}|�| jj�d�� | �|� d S )N�add_bondr� )r$ r� r' r+ r+ r, r� � r� zNetworkView._create_bondc C sz |dkrd S | j r d}nd}| jr|d }n|}d| _t| }tt|�dd�| j�� ftd�| j�� fg| jj||�< d S )Nr� r> r Tr� rq rH )r� r� � wlan_support_install_state_textsr r r� r� r� )r$ r| �start_i�end_i�textr+ r+ r, r� � s
�z1NetworkView.update_for_wlan_support_install_statec C s | j s d }}d| _ nd}d}t�| j�| j�� ftd�| j�� fg| jj||�< |dkr=|d }| j�d|j f � d S |dkrI| j�d� d S |d krU| j�d
� d S |d kra| j�d � d S |d
krm| j�d� d S |dkry| j�d� d S | j�d� d S )Nr Tr> rH z
stop-networkdz$Stopping systemd-networkd-failed: %r�applyzHNetwork configuration could not be applied; please verify your settings.�timeoutz=Network configuration timed out; please verify your settings.�downz"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.)
r� r �
info_errorr� r� r� r r� r� �stderr)r$ r9 r` r� r� �excr+ r+ r, �show_network_error� s0 �
�z NetworkView.show_network_errorc C s$ | j r g | jjdd�< | j�� d S )Nr r> )r� r� r� r r� )r$ �resultr+ r+ r, r� � s zNetworkView.donec C s | j �� d S r )r r� )r$ �buttonr+ r+ r, r� � s zNetworkView.cancel)r� r )#r� r� r� rU �titler� r4 r� r� r. r/ r �_action_EDIT_WLANr �_action_EDIT_IPV4�_action_EDIT_IPV6r �_action_ADD_VLANr� r� rM r� r� r� r� r� r� r� r� r� r� r r� r� r� r+ r+ r6 r, r� � s: �8
r� )1�__doc__�logging�urwidr r � subiquitycore.models.networkr r �subiquitycore.ui.actionmenur �subiquitycore.ui.buttonsr r r �subiquitycore.ui.containerr
r �subiquitycore.ui.spinnerr �subiquitycore.ui.stretchyr
�subiquitycore.ui.tabler r r �subiquitycore.ui.utilsr r r r r �subiquitycore.ui.widthr �"network_configure_manual_interfacer r r r � network_configure_wlan_interfacer �subiquitycore.viewr � getLoggerr� r/ r0 rU r� r� r+ r+ r+ r, �<module> s2
�