Handle SSHException

7 years ago · 75f53137bc
1 changed files with 3 additions and 1 deletions
--- a/auto_backup/models/db_backup.py
+++ b/auto_backup/models/db_backup.py
@ -126,7 +126,9 @@ class DbBackup(models.Model):
            # Just open and close the connection
            with self.sftp_connection():
                raise exceptions.Warning(_("Connection Test Succeeded!"))
-        except (pysftp.CredentialException, pysftp.ConnectionException):
+        except (pysftp.CredentialException,
+                pysftp.ConnectionException,
+                pysftp.SSHException):
            _logger.info("Connection Test Failed!", exc_info=True)
            raise exceptions.Warning(_("Connection Test Failed!"))