diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/trade_dataclasses.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/python/trade_dataclasses.py b/python/trade_dataclasses.py index 7b80a103..279c50e9 100644 --- a/python/trade_dataclasses.py +++ b/python/trade_dataclasses.py @@ -195,7 +195,12 @@ class DealKind: def get_admin_headers(fund, trade_type): if fund in ("SERCGMAST", "BOWDST", "BRINKER"): - return HEADERS[trade_type] + try: + return HEADERS[trade_type] + except: + from headers.globeop_upload import globeop_IRS, globeop_TRS + + return globeop_TRS def get_fname( @@ -213,6 +218,7 @@ def get_fname( "capfloor": "CapFloor", "repo": "RepoDeal", "termination": "Termination", + "trs": "TRS", } trade_tag: str if isinstance(trade_type, tuple): @@ -948,7 +954,7 @@ class TRSDeal( default=None, metadata={"insert": False, "mtm": "Swap ID", "globeop": "DealId"} ) - def to_globeop(self): + def to_admin(self): obj = self.serialize("globeop") if obj["buysell"]: key1, key2 = "Pay", "Receive" @@ -1032,7 +1038,7 @@ class IRSDeal( default=None, metadata={"insert": False, "mtm": "Swap ID", "globeop": "DealId"} ) - def to_globeop(self): + def to_admin(self): obj = self.serialize("globeop") if obj["payreceive"]: key1, key2 = "Receive", "Pay" |