diff options
Diffstat (limited to 'python/mtm_upload.py')
| -rw-r--r-- | python/mtm_upload.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/python/mtm_upload.py b/python/mtm_upload.py index d56a60d2..f97bd908 100644 --- a/python/mtm_upload.py +++ b/python/mtm_upload.py @@ -13,19 +13,21 @@ deal_registry = { } -def mtm_process_upload(tradeids, asset_type): +def mtm_process_upload(tradeids, product_type): objs = [ - deal_registry["asset_type"].from_tradeid(tradeid).to_markit() + deal_registry[product_type].from_tradeid(tradeid).to_markit() for tradeid in tradeids ] buf = StringIO() csvwriter = csv.writer(buf) - csvwriter.writerow(MTM_HEADERS[asset_type]) + csvwriter.writerow(MTM_HEADERS[product_type]) csvwriter.writerows( - [obj.get(h, None) for h in MTM_HEADERS[asset_type]] for obj in objs + [obj.get(h, None) for h in MTM_HEADERS[product_type]] for obj in objs ) buf = buf.getvalue().encode() - fname = f"MTM.{datetime.datetime.now():%Y%m%d.%H%M%S}.{asset_type.capitalize()}.csv" + fname = ( + f"MTM.{datetime.datetime.now():%Y%m%d.%H%M%S}.{product_type.capitalize()}.csv" + ) sftp = SftpClient.from_creds("mtm") sftp.put(buf, fname) dest = DAILY_DIR / str(datetime.date.today()) / fname |