update data models

flux_sdk/benefits_administration/capabilities/process_employees_deductions/data_models.py

@@ -1,11 +1,25 @@
+from typing import Optional
+from decimal import Decimal
+from datetime import datetime
+
+from enum import Enum
+
 class DeductionDetails:
     companyId: str
     employeeId: str
     deductionCode: str
     effectiveFrom: datetime
     endDate: datetime
-    totalContribution: Optional[Decimal]
     employeeContribution: Optional[Decimal]
-    domesticPartnerPostTax: Optional[Decimal]
-    imputedIncome: Optional[Decimal]
     companyContribution: Optional[Decimal]
+
+class DeductionCodeField(Enum):
+    EMPLOYEE_CONTRIBUTION = "EMPLOYEE_CONTRIBUTION"
+    COMPANY_CONTRIBUTION = "COMPANY_CONTRIBUTION"
+
+class ExternalDeductionCodeToRipplingCode:
+    externalCode: str
+    ripplingCode: str
+    ripplingDeductionField: DeductionCodeField
+
+

flux_sdk/benefits_administration/capabilities/process_employees_deductions/interface.py

@@ -1,7 +1,10 @@
 from abc import ABC, abstractmethod
+from io import IOBase
 
-from flux_sdk.flux_core.data_models import CompanyInfo, File
-from flux_sdk.benefits_administration.capabilities.process_employees_deductions.data_models import DeductionDetails
+from flux_sdk.benefits_administration.capabilities.process_employees_deductions.data_models import {
+    DeductionDetails,
+    ExternalDeductionCodeToRipplingCode
+}
 
 
 class ProcessEmployeesDeductions(ABC):
@@ -22,7 +25,7 @@ def get_file_name(company_name: str) -> str:
 
     @staticmethod
     @abstractmethod
-    def format_and_fetch_deduction_info(uri: str, stream: IOBase) -> list[DeductionDetails]:
+    def format_and_fetch_deduction_info(uri: str, stream: IOBase, deduction_code_mapping: list[ExternalDeductionCodeToRipplingCode]) -> list[DeductionDetails]:
         """
         This method receives the file which contains the deductions relevant to the companies employees and returns the
         deductions details for each employee

flux_sdk/benefits_administration/capabilities/report_employees_personal_and_employment_data/data_models.py

@@ -100,6 +100,8 @@ class EmployeePersonalAndEmploymentData:
     This contains the core data about an employee which is relevant to a benefits administration provider
     '''
     id: str
+    employee_number: int
+    company_tax_id: str
     personal: Employee
     employment: Employment
     pay: Pay