Added separate transaction types for Stamp Duty & STT.

Handled Segregation of Folio transactions

Author: Abhishek Jain

casparser/enums.py

@@ -27,6 +27,8 @@ class TransactionType(Enum):
     SWITCH_IN_MERGER = auto()
     SWITCH_OUT = auto()
     SWITCH_OUT_MERGER = auto()
-    TAX = auto()
+    STT_TAX = auto()
+    STAMP_DUTY_TAX = auto()
+    SEGREGATION = auto()
     MISC = auto()
     UNKNOWN = auto()

casparser/process/cas_detailed.py

@@ -36,7 +36,11 @@ def get_transaction_type(
         )
     elif units is None:
         if "stt" in description:
-            txn_type = TransactionType.TAX
+            txn_type = TransactionType.STT_TAX
+        elif "stamp" in description:
+            txn_type = TransactionType.STAMP_DUTY_TAX
+        elif ("segregate" in description.lower()) or ("segregation" in description.lower()):
+            txn_type = TransactionType.SEGREGATION
         else:
             txn_type = TransactionType.MISC
     elif units > 0:
@@ -167,6 +171,11 @@ def process_detailed_text(text):
                 nav = Decimal(m.group(5).replace(",", "_"))
                 balance = Decimal(m.group(6).replace(",", "_").replace("(", "-"))
             txn_type, dividend_rate = get_transaction_type(desc, units)
+            if txn_type == TransactionType.SEGREGATION:
+                units = amt
+                balance = amt
+                amt = 0
+                nav = 0
             if units is not None:
                 curr_scheme_data["close_calculated"] += units
             curr_scheme_data["transactions"].append(