nik/FreedomBox
1
0
Fork 0
mirror of https://github.com/freedombox/FreedomBox.git synced 2026-02-18 08:33:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

backups: Fix issue with repository not being initialized

Browse Source 
Signed-off-by: Joseph Nuthalapati <njoseph@thoughtworks.com>
This commit is contained in:
Joseph Nuthalapati 2019-05-17 18:56:05 +05:30
parent 3a8fbdd799
commit 56babf5cb4
No known key found for this signature in database
GPG Key ID: 5398F00A2FA43C35
1 changed files with 23 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
plinth/modules/backups/forms.py
View File

@ -157,6 +157,7 @@ class AddRepositoryForm(forms.Form):
dir_path = dir_path.replace('~', f'/home/{username}')
password = credentials['ssh_password']
ssh_client = paramiko.SSHClient()
# TODO Prompt to accept fingerprint of the server
ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
ssh_client.connect(hostname, username=username, password=password)
@ -164,23 +165,27 @@ class AddRepositoryForm(forms.Form):
msg = _('Accessing the remote repository failed. Details: %(err)s')
raise forms.ValidationError(msg, params={'err': str(err)})
else:
with ssh_client.open_sftp() as sftp_client:
try:
dir_contents = sftp_client.listdir(dir_path)
except FileNotFoundError:
logger.info(
_(f"Directory {dir_path} doesn't exist. Creating ..."))
sftp_client.mkdir(dir_path)
else:
if dir_contents:
try:
self.repository = SshBorgRepository(
path=path, credentials=credentials)
self.repository.get_info()
except BorgRepositoryDoesNotExistError:
msg = _(f'Directory {path.split(":")[-1]} is '
'neither empty nor is an existing '
'backups repository.')
raise forms.ValidationError(msg)
sftp_client = ssh_client.open_sftp()
try:
dir_contents = sftp_client.listdir(dir_path)
except FileNotFoundError:
logger.info(
_(f"Directory {dir_path} doesn't exist. Creating ..."))
sftp_client.mkdir(dir_path)
self.repository = SshBorgRepository(path=path,
credentials=credentials)
else:
if dir_contents:
try:
self.repository = SshBorgRepository(
path=path, credentials=credentials)
self.repository.get_info()
except BorgRepositoryDoesNotExistError:
msg = _(f'Directory {path.split(":")[-1]} is '
'neither empty nor is an existing '
'backups repository.')
raise forms.ValidationError(msg)
finally:
sftp_client.close()
finally:
ssh_client.close()