diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/citco_ops/utils.py | 11 | ||||
| -rw-r--r-- | python/citco_submission.py | 10 |
2 files changed, 8 insertions, 13 deletions
diff --git a/python/citco_ops/utils.py b/python/citco_ops/utils.py index c13bf3ae..57f8ef84 100644 --- a/python/citco_ops/utils.py +++ b/python/citco_ops/utils.py @@ -74,16 +74,13 @@ def round_up(n, decimals=0): @dataclass class CitcoSubmission(Deal, deal_type=None, table_name="citco_submission"): - fname: str = field() + fname: str identifier_type: Literal["trade", "instrument"] identifier: str serenitas_id: str submit_date: datetime.datetime = field(default=datetime.datetime.now()) processed: bool = field(default=False) - _sftp: ClassVar = field( - default=SftpClient.from_creds("citco", folder="/outgoing/notifications"), - init=False, - ) + _sftp: ClassVar = field(metadata={"insert": False}) @classmethod def from_citco_line(cls, line, fname): @@ -165,7 +162,7 @@ class CitcoSubmission(Deal, deal_type=None, table_name="citco_submission"): tablefmt="unsafehtml", ) html = HTMLBody( - f""" + f""" <html> <head> <style> @@ -183,7 +180,7 @@ class CitcoSubmission(Deal, deal_type=None, table_name="citco_submission"): @classmethod def init_sftp(cls): - return SftpClient.from_creds("citco", folder="/outgoing/notifications") + cls._sftp = SftpClient.from_creds("citco", folder="/outgoing/notifications") @classmethod def check_cache(cls): diff --git a/python/citco_submission.py b/python/citco_submission.py index 2495c746..fcef1ac4 100644 --- a/python/citco_submission.py +++ b/python/citco_submission.py @@ -1,24 +1,22 @@ from stat import S_ISREG import time -import pickle -from serenitas.utils.remote import SftpClient from citco_ops.utils import CitcoSubmission from paramiko.ssh_exception import SSHException def run(): - sftp = CitcoSubmission.init_sftp() + CitcoSubmission.init_sftp() while True: try: - for f in sftp.client.listdir_iter(): + for f in CitcoSubmission._sftp.client.listdir_iter(): if S_ISREG(f.st_mode): CitcoSubmission.process(f.filename) CitcoSubmission.check_cache() CitcoSubmission.commit() except (SSHException, OSError): - sftp.client.close() - sftp = CitcoSubmission.init_sftp() + CitcoSubmission._sftp.client.close() + CitcoSubmission.init_sftp() time.sleep(60) |