diff options
| -rw-r--r-- | python/citco_submission.py | 10 | ||||
| -rw-r--r-- | python/quantifi_upload_status.py | 10 |
2 files changed, 4 insertions, 16 deletions
diff --git a/python/citco_submission.py b/python/citco_submission.py index ea3b41cd..a6c46aa1 100644 --- a/python/citco_submission.py +++ b/python/citco_submission.py @@ -6,18 +6,12 @@ from paramiko.ssh_exception import SSHException import logging -@contextmanager -def retry_on_exception_sftp(): - yield - - def close_and_reconnect(): retries = 5 for i in range(retries): try: - with retry_on_exception_sftp(): - CitcoSubmission._sftp.client.close() - CitcoSubmission.init_sftp() + CitcoSubmission._sftp.client.close() + CitcoSubmission.init_sftp() except (SSHException, OSError) as e: if i == retries - 1: raise e diff --git a/python/quantifi_upload_status.py b/python/quantifi_upload_status.py index 32aa8267..e32d8b4a 100644 --- a/python/quantifi_upload_status.py +++ b/python/quantifi_upload_status.py @@ -8,18 +8,12 @@ from report_ops.logger import get_logger logger = get_logger(__name__) -@contextmanager -def retry_on_exception_sftp(): - yield - - def close_and_reconnect(): retries = 5 for i in range(retries): try: - with retry_on_exception_sftp(): - QuantifiRemote._client.client.close() - QuantifiRemote.init_client() + QuantifiRemote._client.client.close() + QuantifiRemote.init_client() except (SSHException, OSError) as e: if i == retries - 1: raise e |