ENH: Add additional members to AocType and GroupOfContract

lifelib-dev · Jul 8, 2023 · 8f9cdfb · 8f9cdfb
1 parent c750be4
commit 8f9cdfb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lifelib/libraries/ifrs17a/ifrs17/DataStructure.py b/lifelib/libraries/ifrs17a/ifrs17/DataStructure.py
@@ -105,8 +105,9 @@ class VariableType(KeyedOrderedDimension):
 
 
 ### AoC Variable Type
+@_dataclass
 class AocType(VariableType): 
-    pass
+    PnlType: str
 
 
 AocStep = _namedtuple('AocStep', ['AocType', 'Novelty'])
@@ -318,6 +319,7 @@ class GroupOfContract(DataNode):
     Portfolio: str
     YieldCurveName: str
     Partner: str
+    IsReinsurance: bool
 
 
 @_dataclass(eq=False)

diff --git a/lifelib/libraries/ifrs17a/ifrs17/Importers.py b/lifelib/libraries/ifrs17a/ifrs17/Importers.py
@@ -389,7 +389,8 @@ def _FromDataSetGroupOfContracts(dataset, datarow):
             Profitability=datarow["Profitability"],
             Portfolio=pf,
             YieldCurveName=datarow["YieldCurveName"] if "YieldCurveName" in dataset.Tables["GroupOfInsuranceContract"].columns else '',
-            Partner=''
+            Partner='',
+            IsReinsurance=False
         )
 
         return ExtendGroupOfContract(gic, datarow)
@@ -419,6 +420,7 @@ def _FromDataSetGroupOfReinsuranceContract(dataset, datarow):
             Portfolio=pf,
             Partner=datarow["Partner"],
             YieldCurveName=datarow["YieldCurveName"] if "YieldCurveName" in dataset.Tables["GroupOfInsuranceContract"].columns.values else '',
+            IsReinsurance=True
         )
         return ExtendGroupOfContract(gric, datarow)