diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/citco_ops/utils.py | 4 | ||||
| -rw-r--r-- | python/citco_submission.py | 4 |
2 files changed, 6 insertions, 2 deletions
diff --git a/python/citco_ops/utils.py b/python/citco_ops/utils.py index aedcc931..41450a36 100644 --- a/python/citco_ops/utils.py +++ b/python/citco_ops/utils.py @@ -167,6 +167,10 @@ class CitcoSubmission(Deal, deal_type=None, table_name="citco_submission"): ) return html + @classmethod + def _init_sftp(cls): + return SftpClient.from_creds("citco", folder="/outgoing/notifications") + _recipients = { "ISOSEL": ( diff --git a/python/citco_submission.py b/python/citco_submission.py index e5e437ec..f707e42a 100644 --- a/python/citco_submission.py +++ b/python/citco_submission.py @@ -7,7 +7,7 @@ from paramiko.ssh_exception import SSHException def run(): - sftp = SftpClient.from_creds("citco", folder="/outgoing/notifications") + sftp = CitcoSubmission._init_sftp() while True: try: for f in sftp.client.listdir_iter(): @@ -19,7 +19,7 @@ def run(): CitcoSubmission.commit() except (SSHException, OSError): sftp.client.close() - sftp = SftpClient.from_creds("citco", folder="/outgoing/notifications") + sftp = CitcoSubmission._init_sftp() time.sleep(60) |