2015-04-06 22:21:28 +05:30

315 lines
12 KiB
Python

#
# This file is part of Plinth.
#
# 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/>.
#
from dbus.exceptions import DBusException
from django.contrib import messages
from django.contrib.auth.decorators import login_required
from django.core.urlresolvers import reverse_lazy
from django.shortcuts import redirect
from django.template.response import TemplateResponse
from gettext import gettext as _
import NetworkManager
import uuid
import urllib
from .forms import CONNECTION_TYPE_NAMES, ConnectionTypeSelectForm, \
AddEthernetForm, AddWifiForm
from plinth import cfg
subsubmenu = [{'url': reverse_lazy('network:index'),
'text': _('Network Connections')},
{'url': reverse_lazy('network:add'),
'text': _('Add Connection')}]
def init():
"""Initialize the Network module."""
menu = cfg.main_menu.get('system:index')
menu.add_urlname(_('Network'), 'glyphicon-signal', 'network:index', 18)
@login_required
def index(request):
"""Show connection list."""
connections = []
active = []
for conn in NetworkManager.NetworkManager.ActiveConnections:
try:
settings = conn.Connection.GetSettings()['connection']
except DBusException:
# DBusException can be thrown here if the index is quickly loaded
# after a connection is deactivated.
continue
active.append(settings['id'])
for conn in NetworkManager.Settings.ListConnections():
settings = conn.GetSettings()['connection']
# Display a friendly type name if known.
conn_type = CONNECTION_TYPE_NAMES.get(settings['type'],
settings['type'])
connections.append({
'name': settings['id'],
'id': urllib.parse.quote_plus(settings['id']),
'type': conn_type,
'is_active': settings['id'] in active,
})
connections.sort(key=lambda x: x['is_active'], reverse=True)
return TemplateResponse(request, 'connections_list.html',
{'title': _('Network Connections'),
'subsubmenu': subsubmenu,
'connections': connections})
@login_required
def edit(request, conn_id):
"""Serve connection editing form."""
form = None
name = urllib.parse.unquote_plus(conn_id)
form_data = {'name': name}
conn_found = False
for conn in NetworkManager.Settings.ListConnections():
settings = conn.GetSettings()
if settings['connection']['id'] == name:
conn_found = True
break
if not conn_found:
return redirect(reverse_lazy('network:index'))
if request.method == 'POST':
if settings['connection']['type'] == '802-11-wireless':
form = AddWifiForm(request.POST)
else:
form = AddEthernetForm(request.POST)
if form.is_valid():
new_settings = {
'connection': {
'id': form.cleaned_data['name'],
'type': settings['connection']['type'],
'uuid': settings['connection']['uuid'],
},
'ipv4': {'method': form.cleaned_data['ipv4_method']},
}
if form.cleaned_data['ipv4_method'] == 'manual':
new_settings['ipv4']['addresses'] = [
(form.cleaned_data['ipv4_address'],
24, # CIDR prefix length
'0.0.0.0')] # gateway
if settings['connection']['type'] == '802-3-ethernet':
new_settings['802-3-ethernet'] = {}
elif settings['connection']['type'] == '802-11-wireless':
new_settings['802-11-wireless'] = {
'ssid': form.cleaned_data['ssid'],
}
conn.Update(new_settings)
return redirect(reverse_lazy('network:index'))
else:
form_data['ipv4_method'] = settings['ipv4']['method']
if settings['ipv4']['addresses']:
form_data['ipv4_address'] = settings['ipv4']['addresses'][0][0]
if settings['connection']['type'] == '802-11-wireless':
form_data['ssid'] = settings['802-11-wireless']['ssid']
form = AddWifiForm(form_data)
else:
form = AddEthernetForm(form_data)
return TemplateResponse(request, 'connections_edit.html',
{'title': _('Edit Connection'),
'subsubmenu': subsubmenu,
'form': form})
@login_required
def activate(request, conn_id):
"""Activate the connection."""
name = urllib.parse.unquote_plus(conn_id)
# Find the connection
connections = NetworkManager.Settings.ListConnections()
connections = dict([(x.GetSettings()['connection']['id'], x)
for x in connections])
conn = connections[name]
# Find a suitable device
ctype = conn.GetSettings()['connection']['type']
if ctype == 'vpn':
for dev in NetworkManager.NetworkManager.GetDevices():
if (dev.State == NetworkManager.NM_DEVICE_STATE_ACTIVATED
and dev.Managed):
break
else:
messages.error(
request,
_('Failed to activate connection: '
'No active, managed device found'))
return redirect(reverse_lazy('network:index'))
else:
dtype = {
'802-11-wireless': NetworkManager.NM_DEVICE_TYPE_WIFI,
'802-3-ethernet': NetworkManager.NM_DEVICE_TYPE_ETHERNET,
'gsm': NetworkManager.NM_DEVICE_TYPE_MODEM,
}.get(ctype, ctype)
for dev in NetworkManager.NetworkManager.GetDevices():
if (dev.DeviceType == dtype
and dev.State == NetworkManager.NM_DEVICE_STATE_DISCONNECTED):
break
else:
messages.error(
request,
_('Failed to activate connection: '
'No suitable and available %s device found' % ctype))
return redirect(reverse_lazy('network:index'))
NetworkManager.NetworkManager.ActivateConnection(conn, dev, "/")
messages.success(request, _('Activated connection %s.') % name)
return redirect(reverse_lazy('network:index'))
@login_required
def deactivate(request, conn_id):
"""Deactivate the connection."""
name = urllib.parse.unquote_plus(conn_id)
active = NetworkManager.NetworkManager.ActiveConnections
active = dict([(x.Connection.GetSettings()['connection']['id'], x)
for x in active])
NetworkManager.NetworkManager.DeactivateConnection(active[name])
messages.success(request, _('Deactivated connection %s.') % name)
return redirect(reverse_lazy('network:index'))
@login_required
def add(request):
"""Serve the connection type selection form."""
form = None
if request.method == 'POST':
form = ConnectionTypeSelectForm(request.POST)
if form.is_valid():
conn_type = form.cleaned_data['conn_type']
if conn_type == '802-3-ethernet':
return redirect(reverse_lazy('network:add_ethernet'))
elif conn_type == '802-11-wireless':
return redirect(reverse_lazy('network:add_wifi'))
else:
form = ConnectionTypeSelectForm()
return TemplateResponse(request, 'connections_type_select.html',
{'title': _('Add Connection'),
'subsubmenu': subsubmenu,
'form': form})
@login_required
def add_ethernet(request):
"""Serve ethernet connection create form."""
form = None
if request.method == 'POST':
form = AddEthernetForm(request.POST)
if form.is_valid():
conn = {
'connection': {
'id': form.cleaned_data['name'],
'type': '802-3-ethernet',
'uuid': str(uuid.uuid4()),
},
'802-3-ethernet': {},
'ipv4': {'method': form.cleaned_data['ipv4_method']},
}
if form.cleaned_data['ipv4_method'] == 'manual':
conn['ipv4']['addresses'] = [
(form.cleaned_data['ipv4_address'],
24, # CIDR prefix length
'0.0.0.0')] # gateway
NetworkManager.Settings.AddConnection(conn)
return redirect(reverse_lazy('network:index'))
else:
form = AddEthernetForm()
return TemplateResponse(request, 'connections_create.html',
{'title': _('Editing New Ethernet Connection'),
'subsubmenu': subsubmenu,
'form': form})
@login_required
def add_wifi(request):
"""Serve wifi connection create form."""
form = None
if request.method == 'POST':
form = AddWifiForm(request.POST)
if form.is_valid():
conn = {
'connection': {
'id': form.cleaned_data['name'],
'type': '802-11-wireless',
'uuid': str(uuid.uuid4()),
},
'802-11-wireless': {
'ssid': form.cleaned_data['ssid'],
},
'ipv4': {'method': form.cleaned_data['ipv4_method']},
}
if form.cleaned_data['ipv4_method'] == 'manual':
conn['ipv4']['addresses'] = [
(form.cleaned_data['ipv4_address'],
24, # CIDR prefix length
'0.0.0.0')] # gateway
NetworkManager.Settings.AddConnection(conn)
return redirect(reverse_lazy('network:index'))
else:
form = AddWifiForm()
return TemplateResponse(request, 'connections_create.html',
{'title': _('Editing New Wi-Fi Connection'),
'subsubmenu': subsubmenu,
'form': form})
@login_required
def delete(request, conn_id):
"""Handle deleting connections, showing a confirmation dialog first.
On GET, display a confirmation page.
On POST, delete the connection.
"""
name = urllib.parse.unquote_plus(conn_id)
if request.method == 'POST':
for conn in NetworkManager.Settings.ListConnections():
settings = conn.GetSettings()['connection']
if settings['id'] == name:
conn.Delete()
messages.success(request, _('Connection %s deleted.') % name)
return redirect(reverse_lazy('network:index'))
messages.error(
request,
_('Failed to delete connection %s: not found.') % name)
return redirect(reverse_lazy('network:index'))
return TemplateResponse(request, 'connections_delete.html',
{'title': _('Delete Connection'),
'subsubmenu': subsubmenu,
'name': name})