diff options
Diffstat (limited to 'python/innocap_file_transfer.py')
| -rw-r--r-- | python/innocap_file_transfer.py | 22 |
1 files changed, 18 insertions, 4 deletions
diff --git a/python/innocap_file_transfer.py b/python/innocap_file_transfer.py index db555a66..7e39a264 100644 --- a/python/innocap_file_transfer.py +++ b/python/innocap_file_transfer.py @@ -2,8 +2,9 @@ from io import BytesIO from serenitas.utils.env import DAILY_DIR import datetime from csv_headers.citco import GIL, GTL -from serenitas.utils.remote import SftpClient +from serenitas.utils.remote import Client from serenitas.analytics.dates import prev_business_day +from report_ops.sma import build_position_file def concat_csv(file_type, date): @@ -23,16 +24,28 @@ def concat_csv(file_type, date): return buf.getvalue(), fname -def main(date, upload): +def upload_citco_files(date, upload): for file_type in ("trade", "instrument"): buf, fname = concat_csv(file_type, date) dest = DAILY_DIR / str(date) / fname dest.write_bytes(buf) if upload: - innocap_sftp = SftpClient.from_creds("innocap", folder="Innocap") + innocap_sftp = Client.from_creds("innocap", folder="Innocap") innocap_sftp.put(buf, fname) +def upload_position_files(date, fund, upload): + buf, dest = build_position_file( + date, + fund, + ) + if upload: + client = Client.from_creds("innocap", folder="Innocap") + client.put(buf, dest.name) + client = Client.from_creds("citco") + client.put(buf, dest.name) + + if __name__ == "__main__": import argparse @@ -53,4 +66,5 @@ if __name__ == "__main__": ) args = parser.parse_args() - main(args.cob, not args.no_upload) + upload_citco_files(args.cob, not args.no_upload) + upload_position_files(args.cob, "ISOSEL", not args.no_upload) |