diff options
| -rw-r--r-- | python/process_queue.py | 3 | ||||
| -rw-r--r-- | python/trade_dataclasses.py | 2 |
2 files changed, 4 insertions, 1 deletions
diff --git a/python/process_queue.py b/python/process_queue.py index 738f7cb0..4fcecf9f 100644 --- a/python/process_queue.py +++ b/python/process_queue.py @@ -125,12 +125,13 @@ def terminate_list( term_dict = {"T": [], "A": []} for term in p.lrange(key, 0, -1): termination = loads(term) + DealKind["termination"].from_dict(**termination).mtm_stage() try: build_termination(term_dict, conn, fund=fund, **termination) except TypeError as e: logging.error(e) return - + DealKind["termination"].mtm_upload() for k, v in term_dict.items(): if v and upload: dest = get_filepath(base_dir, (trade_type, k), fund) diff --git a/python/trade_dataclasses.py b/python/trade_dataclasses.py index c035d8d7..0d36f338 100644 --- a/python/trade_dataclasses.py +++ b/python/trade_dataclasses.py @@ -213,6 +213,8 @@ class DealKind: return CDSDeal case "swaption": return SwaptionDeal + case "termination": + return TerminationDeal case _: return None |