diff options
| -rw-r--r-- | python/book_bbg.py | 5 | ||||
| -rw-r--r-- | python/manual_trade_upload.py | 28 |
2 files changed, 9 insertions, 24 deletions
diff --git a/python/book_bbg.py b/python/book_bbg.py index d0e83ac9..331a13ad 100644 --- a/python/book_bbg.py +++ b/python/book_bbg.py @@ -1,5 +1,8 @@ from serenitas.utils.remote import SftpClient -from serenitas.ops.trade_dataclasses import Deal, DealType, BbgDeal +import sys + +sys.path.append("/home/flint/projects/ops/serenitas") +from ops.trade_dataclasses import Deal, DealType, BbgDeal from stat import S_ISREG import re import time diff --git a/python/manual_trade_upload.py b/python/manual_trade_upload.py index 7e0600a2..24e7f984 100644 --- a/python/manual_trade_upload.py +++ b/python/manual_trade_upload.py @@ -5,32 +5,14 @@ from ops.funds import Fund from ops.trade_dataclasses import DealKind asset_class = "cds" -trade_ids = [ - 4234, - 4232, - 4231, - 4230, - 4221, - 4220, - 4219, - 4218, - 4213, - 4209, - 4208, - 4207, - 4206, - 4205, - 4204, - 4190, - 4187, - 4184, - 4182, -] +trade_ids = [4234, 4232, 4219, 4218] buffers = [] for tradeid in trade_ids: - buffers.append(DealKind[asset_class].from_tradeid(tradeid).to_citco("NEW")) - + if isinstance(DealKind[asset_class].from_tradeid(tradeid).to_citco("UPDATE"), list): + buffers.extend(DealKind[asset_class].from_tradeid(tradeid).to_citco("UPDATE")) + else: + buffers.append(DealKind[asset_class].from_tradeid(tradeid).to_citco("UPDATE")) Fund["ISOSEL"].staging_queue.extend(buffers) Fund["ISOSEL"].build_buffer(asset_class) |