diff options
| -rw-r--r-- | python/trade_dataclasses.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/python/trade_dataclasses.py b/python/trade_dataclasses.py index 23ccc385..fd62c1d7 100644 --- a/python/trade_dataclasses.py +++ b/python/trade_dataclasses.py @@ -430,7 +430,6 @@ class MTMDeal: class Citco: - _citco_queue: ClassVar[list] = [] _citco_headers = [] _sftp = SftpClient.from_creds("citco") @@ -445,8 +444,13 @@ class Citco: [row.get(h, None) for h in cls._citco_headers] for row in cls._citco_queue ) buf = buf.getvalue().encode() - fname = f"i.innocap_serenitas.{datetime.datetime.now():%Y%m%d%H%M%S}.csv" - cls._sftp.client.chdir("incoming") + if hasattr(cls, "product_key"): + fname = f"i.innocap_serenitas.{datetime.datetime.now():%Y%m%d%H%M%S}.csv" + else: + fname = ( + f"innocap_serenitas_trades_{datetime.datetime.now():%Y%m%d%H%M%S}.csv" + ) + cls._sftp.client.chdir("/incoming") cls._sftp.put(buf, fname) dest = DAILY_DIR / str(datetime.date.today()) / fname dest.write_bytes(buf) @@ -457,6 +461,7 @@ class Citco: class CitcoProduct(Citco): + _citco_queue: ClassVar[list] = [] _citco_headers = GIL product_key = () @@ -476,6 +481,7 @@ class CitcoProduct(Citco): class CitcoTrade(Citco): + _citco_queue: ClassVar[list] = [] _citco_headers = GTL |