openstack
/
freezer
Code Issues Proposed changes

Fix ssh connection while restore 

ssh connection gets lost while restoring, so we check if the
connection is not  alive we reconnect again.

Change-Id: I2f6bce2555d89ac103dd77248013ddbdfa862bc4
This commit is contained in:
Saad Zaher 2017-03-01 19:23:52 +00:00
parent 9bddd2927c
commit cca78b6ba0
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
freezer/storage/ssh.py
View File

@ -46,11 +46,26 @@ class SshStorage(fslike.FsLikeStorage):
self.port = port
self.ssh = None
self.ftp = None
self._validate()
self.init()
super(SshStorage, self).__init__(
storage_path=storage_path,
max_segment_size=max_segment_size)
def _validate(self):
"""
Validates if all parameters required to ssh are available.
:return: True or raises ValueError
"""
if not self.remote_ip:
raise ValueError('Please provide --ssh-host value.')
elif not self.remote_username:
raise ValueError('Please provide --ssh-username value.')
elif not self.ssh_key_path:
raise ValueError('Please provide path to ssh key using '
'--ssh-key argument.')
return True
def init(self):
ssh = paramiko.SSHClient()
ssh.load_system_host_keys()
@ -95,6 +110,8 @@ class SshStorage(fslike.FsLikeStorage):
return True
def get_file(self, from_path, to_path):
if not self.ssh.get_transport().is_alive():
self.init()
self.ftp.get(from_path, to_path)
def put_file(self, from_path, to_path):