diff options
| -rw-r--r-- | python/process_queue.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/python/process_queue.py b/python/process_queue.py index 38c06f00..78e1f725 100644 --- a/python/process_queue.py +++ b/python/process_queue.py @@ -32,7 +32,8 @@ from typing import Tuple, Union from quantlib.time.api import pydate_from_qldate, UnitedStates, Days, Date from tabulate import tabulate from headers import get_headers, get_termination_headers -from collections import defaultdict + +_client_name = {"SERCGMAST": "Serenitas", "BOWDST": "HEDGEMARK", "BRINKER": "LMCG"} def get_effective_date(d, swaption_type): @@ -145,7 +146,7 @@ def rename_keys(d, mapping): def build_line(obj, trade_type="bond", fund="SERCGMAST"): - obj["Client"] = client_name[fund] + obj["Client"] = _client_name[fund] obj["State"] = "Valid" rename_cols = { "fund": "Fund", @@ -758,7 +759,7 @@ def build_termination( "DealId": dealid if globeopid is None else None, "GoTradeId": globeopid if globeopid else None, "Action": "Update", - "Client": client_name[fund], + "Client": _client_name[fund], "SubAction": "Termination", "PartialTermination": "Y" if partial_termination else "N", "TerminationAmount": termination_amount, @@ -777,9 +778,7 @@ def build_termination( if __name__ == "__main__": - client_name = defaultdict("LMCG") - client_name["SERCGMAST"] = "Serenitas" - client_name["BOWDST"] = "HEDGEMARK" + parser = argparse.ArgumentParser() parser.add_argument( "-n", "--no-upload", action="store_true", help="do not upload to Globeop" |