119 lines
4.2 KiB
Python
119 lines
4.2 KiB
Python
|
# Copyright 2015 Canonical, Ltd.
|
||
|
#
|
||
|
# This program is free software: you can redistribute it and/or modify
|
||
|
# it under the terms of the GNU Affero General Public License as
|
||
|
# published by the Free Software Foundation, either version 3 of the
|
||
|
# License, or (at your option) any later version.
|
||
|
#
|
||
|
# This program is distributed in the hope that it will be useful,
|
||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
# GNU Affero General Public License for more details.
|
||
|
#
|
||
|
# You should have received a copy of the GNU Affero General Public License
|
||
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
||
|
""" Network Model
|
||
|
|
||
|
Provides network device listings and extended network information
|
||
|
|
||
|
"""
|
||
|
|
||
|
import logging
|
||
|
import textwrap
|
||
|
from urwid import (ListBox, Pile, BoxAdapter,
|
||
|
Text, Columns)
|
||
|
import yaml
|
||
|
from netifaces import AF_INET, AF_INET6
|
||
|
from subiquitycore.ui.lists import SimpleList
|
||
|
from subiquitycore.ui.buttons import cancel_btn, menu_btn, done_btn
|
||
|
from subiquitycore.ui.utils import Padding, Color
|
||
|
from subiquitycore.view import BaseView
|
||
|
|
||
|
|
||
|
log = logging.getLogger('subiquitycore.views.network')
|
||
|
|
||
|
|
||
|
class NetworkView(BaseView):
|
||
|
def __init__(self, model, signal):
|
||
|
self.model = model
|
||
|
self.signal = signal
|
||
|
self.items = []
|
||
|
self.body = [
|
||
|
Padding.center_79(self._build_model_inputs()),
|
||
|
Padding.line_break(""),
|
||
|
## Padding.center_79(self._build_additional_options()),
|
||
|
## Padding.line_break(""),
|
||
|
Padding.fixed_10(self._build_buttons()),
|
||
|
]
|
||
|
# FIXME determine which UX widget should have focus
|
||
|
self.lb = ListBox(self.body)
|
||
|
self.lb.set_focus(2) # _build_buttons
|
||
|
super().__init__(self.lb)
|
||
|
|
||
|
def _build_buttons(self):
|
||
|
cancel = Color.button(cancel_btn(on_press=self.cancel),
|
||
|
focus_map='button focus')
|
||
|
done = Color.button(done_btn(on_press=self.done),
|
||
|
focus_map='button focus')
|
||
|
self.default_focus = done
|
||
|
|
||
|
buttons = [done, cancel]
|
||
|
return Pile(buttons, focus_item=done)
|
||
|
|
||
|
def _build_model_inputs(self):
|
||
|
ifaces = [iface for (name, iface) in sorted(self.model.config.ethernets.items())]
|
||
|
ifname_width = 8 # default padding
|
||
|
|
||
|
col_1 = []
|
||
|
for iface in ifaces:
|
||
|
col_1.append(
|
||
|
Color.info_major(
|
||
|
menu_btn(label=iface.name,
|
||
|
on_press=self.on_net_dev_press),
|
||
|
focus_map='button focus'))
|
||
|
for addr in iface.addresses:
|
||
|
col_1.append(Text("")) # space for address
|
||
|
col_1 = BoxAdapter(SimpleList(col_1),
|
||
|
height=len(col_1))
|
||
|
|
||
|
col_2 = []
|
||
|
for iface in ifaces:
|
||
|
col_2.append(Text(iface.vendor))
|
||
|
for addr in iface.addresses:
|
||
|
t = addr.with_prefixlen
|
||
|
if addr.version == 4:
|
||
|
if iface.dhcp4:
|
||
|
t += " (dhcp)"
|
||
|
else:
|
||
|
t += " (static)"
|
||
|
elif addr.version == 6:
|
||
|
if iface.dhcp6:
|
||
|
t += " (dhcp)"
|
||
|
else:
|
||
|
t += " (static)"
|
||
|
col_2.append(Text(t))
|
||
|
|
||
|
if len(col_2):
|
||
|
col_2 = BoxAdapter(SimpleList(col_2, is_selectable=False),
|
||
|
height=len(col_2))
|
||
|
ifname_width += len(max(ifaces, key=len))
|
||
|
if ifname_width > 20:
|
||
|
ifname_width = 20
|
||
|
else:
|
||
|
col_2 = Pile([Text("No network interfaces detected.")])
|
||
|
|
||
|
return Columns([(ifname_width, col_1), col_2], 2)
|
||
|
|
||
|
def done(self, result):
|
||
|
actions = [iface.action.get() for iface in
|
||
|
self.model.get_configured_interfaces()]
|
||
|
actions += self.model.get_default_route()
|
||
|
log.debug('Configured Network Actions:\n{}'.format(
|
||
|
yaml.dump(actions, default_flow_style=False)))
|
||
|
self.signal.emit_signal('network:finish', actions)
|
||
|
|
||
|
def cancel(self, button):
|
||
|
self.model.reset()
|
||
|
self.signal.prev_signal()
|