aboutsummaryrefslogtreecommitdiffstats
path: root/python
diff options
context:
space:
mode:
Diffstat (limited to 'python')
-rw-r--r--python/process_queue.py3
-rw-r--r--python/trade_dataclasses.py2
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