diff options
Diffstat (limited to 'python/headers.py')
| -rw-r--r-- | python/headers.py | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/python/headers.py b/python/headers.py index 13c9270a..904b5f3e 100644 --- a/python/headers.py +++ b/python/headers.py @@ -1,5 +1,14 @@ +from enum import Enum from typing import Literal + +class DealType(Enum): + Bond = "BOND" + CDS = "CDX" + Swaption = "SWAPTION" + Termination = "TERM" + + HEADERS_PRE = [ "Deal Type", "Deal Id", @@ -380,7 +389,7 @@ HEADERS = { } MTM_HEADERS = { - "cds": [ + DealType.CDS: [ "Swap ID", "Allocation ID", "Description", @@ -450,7 +459,7 @@ MTM_HEADERS = { "Alternate Trade ID", "Definitions Type", ], - "swaption": [ + DealType.Swaption: [ "Swap ID", "Broker Id", "Trade ID", @@ -483,7 +492,7 @@ MTM_HEADERS = { "Swaption Quotation Rate Type", "Effective Date", ], - "termination": [ + DealType.Termination: [ "Swap ID", "Allocation ID", "Description", |