nik/FreedomBox
1
0
Fork 0
mirror of https://github.com/freedombox/FreedomBox.git synced 2026-01-21 07:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

backups: Implement hostname property on SSH repository

Browse Source 
Use repository class instances instead of custom ad-hoc data structures such as
repo_data.

Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org>
Reviewed-by: James Valleroy <jvalleroy@mailbox.org>
This commit is contained in:
Sunil Mohan Adapa 2019-08-28 15:27:43 -07:00 committed by James Valleroy
parent 0eb2db3f9a
commit 6c14e34875
No known key found for this signature in database
GPG Key ID: 77C0C75E7B650808
2 changed files with 20 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
plinth/modules/backups/repository.py
View File

@ -32,7 +32,7 @@ from plinth.errors import ActionError
from . import (ROOT_REPOSITORY, ROOT_REPOSITORY_NAME, ROOT_REPOSITORY_UUID,
_backup_handler, api, get_known_hosts_path,
restore_archive_handler, store)
restore_archive_handler, split_path, store)
from .errors import BorgError, BorgRepositoryDoesNotExistError, SshfsError
logger = logging.getLogger(__name__)
@ -369,6 +369,12 @@ class SshBorgRepository(BaseBorgRepository):
"""
return self._mountpoint
@property
def hostname(self):
"""Return hostname from the remote path."""
_, hostname, _ = split_path(self._path)
return hostname.split('%')[0] # XXX: Likely incorrect to split
@property
def _mountpoint(self):
"""Return the local mount point where repository is to be mounted."""

39
plinth/modules/backups/views.py
View File

@ -311,37 +311,27 @@ class VerifySshHostkeyView(SuccessMessageMixin, FormView):
form_class = forms.VerifySshHostkeyForm
template_name = 'verify_ssh_hostkey.html'
success_url = reverse_lazy('backups:index')
repo_data = {}
repository = None
def get_form_kwargs(self):
"""Pass additional keyword args for instantiating the form."""
kwargs = super().get_form_kwargs()
hostname = self._get_hostname()
kwargs['hostname'] = hostname
kwargs['hostname'] = self._get_repository().hostname
return kwargs
def get_context_data(self, **kwargs):
"""Return additional context for rendering the template."""
context = super().get_context_data(**kwargs)
context['title'] = _('Verify SSH hostkey')
context['hostname'] = self._get_hostname()
context['hostname'] = self._get_repository().hostname
return context
def _get_repo_data(self):
def _get_repository(self):
"""Fetch the repository data from DB only once."""
if not self.repo_data:
uuid = self.kwargs['uuid']
self.repo_data = store.get(uuid)
if not self.repository:
self.repository = create_repository(self.kwargs['uuid'])
return self.repo_data
def _get_hostname(self):
"""Get the hostname of the repository.
Network interface information is stripped out.
"""
_, hostname, _ = split_path(self._get_repo_data()['path'])
return hostname.split('%')[0] # XXX: Likely incorrect to split
return self.repository
@staticmethod
def _add_ssh_hostkey(ssh_public_key):
@ -355,7 +345,7 @@ class VerifySshHostkeyView(SuccessMessageMixin, FormView):
def get(self, *args, **kwargs):
"""Skip this view if host is already verified."""
if is_ssh_hostkey_verified(self._get_hostname()):
if is_ssh_hostkey_verified(self._get_repository().hostname):
messages.success(self.request, _('SSH host already verified.'))
return self._add_remote_repository()
@ -370,19 +360,16 @@ class VerifySshHostkeyView(SuccessMessageMixin, FormView):
def _add_remote_repository(self):
"""On successful verification of host, add repository."""
repo_data = self._get_repo_data()
path = repo_data['path']
credentials = repo_data['credentials']
repository = self._get_repository()
uuid = self.kwargs['uuid']
encryption = 'none'
if 'encryption_passphrase' in credentials and \
credentials['encryption_passphrase']:
if 'encryption_passphrase' in repository.credentials and \
repository.credentials['encryption_passphrase']:
encryption = 'repokey'
try:
dir_contents = _list_remote_directory(path, credentials)
repository = SshBorgRepository(uuid=uuid, path=path,
credentials=credentials)
dir_contents = _list_remote_directory(repository.path,
repository.credentials)
repository.mount()
repository = _create_remote_repository(repository, encryption,
dir_contents)