diff options
Diffstat (limited to 'python/position_file_bowdst.py')
| -rw-r--r-- | python/position_file_bowdst.py | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/python/position_file_bowdst.py b/python/position_file_bowdst.py index 905adf03..738d653d 100644 --- a/python/position_file_bowdst.py +++ b/python/position_file_bowdst.py @@ -4,17 +4,22 @@ import csv from io import StringIO from pathlib import Path from process_queue import rename_keys +from serenitas.utils.remote import SftpClient +from serenitas.utils.env import DAILY_DIR -def process_upload(trades, asset_type): +def process_upload(trades, asset_type, upload): buf = StringIO() csvwriter = csv.writer(buf) csvwriter.writerow(HEADERS[asset_type]) csvwriter.writerows(build_line(trade, asset_type) for trade in trades) buf = buf.getvalue().encode() - dest = Path( - f"/home/serenitas/flint/HEDGEMARK.POSITION.BOS_PAT_BOWDOIN.20211116.095655.{asset_type.upper()}Deal.csv" - ) + fname = f"HEDGEMARK.POSITION.BOS_PAT_BOWDOIN.{datetime.datetime.now():%Y%m%d.%H%M%S}.{asset_type.capitalize()}Deal.csv" + if upload: + sftp = SftpClient.from_creds("hm_globeop") + sftp.client.chdir("incoming") + sftp.put(buf, fname) + dest = DAILY_DIR / str(datetime.date.today()) / fname dest.write_bytes(buf) @@ -335,4 +340,4 @@ if __name__ == "__main__": globals()[f"positions_{p}"](positions, conn, args.date) for asset_type, trades in positions.items(): - process_upload(trades, asset_type) + process_upload(trades, asset_type, args.u) |