Sunil Mohan Adapa 92a5a11659
ssh: Fix flake8 failure by removing unused import
Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org>
2019-10-29 15:28:23 -07:00

77 lines
2.5 KiB
Python

#
# This file is part of FreedomBox.
#
# 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/>.
#
"""
Views for the SSH module
"""
from django.contrib import messages
from django.utils.translation import ugettext_lazy as _
from plinth import actions
from plinth.modules import ssh
from plinth.views import AppView
from . import is_password_authentication_disabled
from .forms import SSHServerForm
class SshAppView(AppView):
app_id = 'ssh'
name = ssh.name
description = ssh.description
port_forwarding_info = ssh.port_forwarding_info
template_name = 'ssh.html'
form_class = SSHServerForm
def get_context_data(self, *args, **kwargs):
context = super().get_context_data(**kwargs)
context['host_keys'] = ssh.get_host_keys()
return context
def get_initial(self):
"""Initial form value"""
initial = super().get_initial()
initial.update({
'password_auth_disabled': is_password_authentication_disabled(),
})
return initial
def form_valid(self, form):
"""Apply changes from the form"""
old_config = self.get_initial()
new_config = form.cleaned_data
def is_field_changed(field):
return old_config[field] != new_config[field]
passwd_auth_changed = is_field_changed('password_auth_disabled')
if passwd_auth_changed:
if new_config['password_auth_disabled']:
passwd_auth = 'no'
message = _('SSH authentication with password disabled.')
else:
passwd_auth = 'yes'
message = _('SSH authentication with password enabled.')
actions.superuser_run(
'ssh', ['set-password-config', '--value', passwd_auth])
actions.superuser_run('service', ['reload', 'ssh'])
messages.success(self.request, message)
return super().form_valid(form)