brynq-sdk-alight 1.0.0__py3-none-any.whl

brynq_sdk_alight/schemas/job.py

@@ -0,0 +1,129 @@
+"""
+Flat, user-friendly job/deployment model for Alight SDK.
+"""
+
+import datetime
+from typing import List, Optional, Dict, Any
+from pydantic import Field, BaseModel
+
+from .utils import add_to_nested_path  # not used here; kept for parity if needed
+
+
+class Job(BaseModel):
+    """
+    Simplified job/deployment model.
+    Uses PURE schema-driven conversion - NO hardcoded structure mappings.
+    Uses aliases to match expected field names in Employee model.
+    """
+    model_config = {
+        "populate_by_name": True  # Allow populating by field name in addition to alias
+    }
+
+    title: str = Field(
+        description="Job title",
+        alias="indicative_person_dossier.indicative_deployment.job.job_title"
+    )
+    department: Optional[str] = Field(
+        default=None,
+        description="Department name",
+        alias="indicative_person_dossier.indicative_deployment.department_name"
+    )
+    position_id: Optional[str] = Field(
+        default=None,
+        description="Position identifier",
+        alias="indicative_person_dossier.indicative_deployment.position_id"
+    )
+    position_title: Optional[str] = Field(
+        default=None,
+        description="Position title",
+        alias="indicative_person_dossier.indicative_deployment.position_title"
+    )
+    location_code: Optional[str] = Field(
+        default=None,
+        description="Work location code",
+        alias="indicative_person_dossier.indicative_deployment.work_location.location_id"
+    )
+    work_location_city_name: Optional[str] = Field(
+        default=None,
+        description="Work location city name",
+        alias="indicative_person_dossier.indicative_deployment.work_location.address.city_name"
+    )
+    work_location_country_sub_division_code: Optional[str] = Field(
+        default=None,
+        description="Work location country subdivision/state code",
+        alias="indicative_person_dossier.indicative_deployment.work_location.address.country_sub_division_code"
+    )
+    work_location_postal_code: Optional[str] = Field(
+        default=None,
+        description="Work location postal code",
+        alias="indicative_person_dossier.indicative_deployment.work_location.address.postal_code"
+    )
+    manager_id: Optional[str] = Field(
+        default=None,
+        description="Manager employee ID",
+        alias="indicative_person_dossier.indicative_deployment.manager_id.id"
+    )
+
+    # Schedule
+    weekly_hours: Optional[float] = Field(
+        default=None,
+        description="Weekly working hours",
+        alias="indicative_person_dossier.indicative_deployment.schedule.scheduled_hours[0].value"
+    )
+    weekly_hours_basis: Optional[str] = Field(
+        default="Week",
+        description="Schedule basis for weekly hours",
+        alias="indicative_person_dossier.indicative_deployment.schedule.scheduled_hours[0].schedule_basis"
+    )
+    pay_cycle_hours: Optional[float] = Field(
+        default=None,
+        description="Hours per pay cycle",
+        alias="indicative_person_dossier.indicative_deployment.schedule.scheduled_hours[1].value"
+    )
+    pay_cycle_hours_basis: Optional[str] = Field(
+        default=None,
+        description="Schedule basis for pay cycle hours",
+        alias="indicative_person_dossier.indicative_deployment.schedule.scheduled_hours[1].schedule_basis"
+    )
+    scheduled_days_per_week: Optional[float] = Field(
+        default=None,
+        description="Scheduled days per week",
+        alias="indicative_person_dossier.indicative_deployment.schedule.scheduled_days[0].value"
+    )
+    scheduled_days_basis: Optional[str] = Field(
+        default=None,
+        description="Schedule basis for scheduled days",
+        alias="indicative_person_dossier.indicative_deployment.schedule.scheduled_days[0].schedule_basis"
+    )
+    day_schedule_id: Optional[str] = Field(
+        default=None,
+        description="Day schedule ID",
+        alias="indicative_person_dossier.indicative_deployment.schedule.day_schedule[0].id"
+    )
+    fte_ratio: Optional[str] = Field(
+        default=None,
+        description="Full-time equivalent ratio",
+        alias="indicative_person_dossier.indicative_deployment.full_time_equivalent_ratio"
+    )
+    work_level_code: Optional[str] = Field(
+        default="FullTime",
+        description="FullTime/PartTime",
+        alias="indicative_person_dossier.indicative_deployment.work_level_code"
+    )
+
+    # Organization fields removed - these belong in Employee model
+
+    # Dates - employment lifecycle dates moved to Employee model
+    valid_from: Optional[datetime.date] = Field(
+        default=None,
+        description="Valid from date",
+        alias="indicative_person_dossier.indicative_deployment.valid_from"
+    )
+
+    def to_nested_dict(self) -> Dict[str, Any]:
+        """
+        Convert to nested structure compatible with employee job fields.
+        Rely on index-aware aliases.
+        """
+        data = self.model_dump(exclude_none=True, by_alias=True)
+        return data

brynq_sdk_alight/schemas/leave.py

@@ -0,0 +1,58 @@
+"""
+Flat, user-friendly leave model for Alight SDK.
+"""
+
+import datetime
+from typing import Optional, Dict, Any
+from pydantic import Field, BaseModel
+
+from .utils import add_to_nested_path  # not used here
+
+
+class Leave(BaseModel):
+    """
+    Simplified leave model.
+    Uses PURE schema-driven conversion - NO hardcoded structure mappings.
+    Uses aliases to match expected field names in Employee model.
+    Maps to IndicativeEmployment.employment_lifecycle.leave[0].
+    """
+    model_config = {
+        "populate_by_name": True
+    }
+
+    leave_reason_code: Optional[str] = Field(
+        default=None,
+        description="Reason for leave",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.leave[0].leave_reason_code"
+    )
+    leave_status_code: Optional[str] = Field(
+        default=None,
+        description="Current leave status",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.leave[0].leave_status_code"
+    )
+    leave_start_date: Optional[datetime.date] = Field(
+        default=None,
+        description="Leave start date",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.leave[0].leave_start_date"
+    )
+    last_paid_date: Optional[datetime.date] = Field(
+        default=None,
+        description="Last paid date during leave",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.leave[0].last_paid_date"
+    )
+    scheduled_work_return_date: Optional[datetime.date] = Field(
+        default=None,
+        description="Scheduled work return date",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.leave[0].scheduled_work_return_date"
+    )
+    work_return_date: Optional[datetime.date] = Field(
+        default=None,
+        description="Actual work return date",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.leave[0].work_return_date"
+    )
+
+    def to_nested_dict(self) -> Dict[str, Any]:
+        """
+        Convert to nested structure compatible with employee model.
+        """
+        return self.model_dump(exclude_none=True, by_alias=True)

brynq_sdk_alight/schemas/payments.py

@@ -0,0 +1,207 @@
+from typing import Optional, Dict, Any
+import datetime
+from pydantic import Field, BaseModel
+
+from .utils import add_to_nested_path, post_process_nested_data, convert_datetime_to_xml, construct_model
+from .absence import Absences
+from .timequota import TimeQuotas
+from ..schemas.generated_envelope_xsd_schema.process_pay_serv_emp import PayServEmpExtension as XsdPayServEmpExtension
+from ..schemas.generated_envelope_xsd_schema.process_pay_serv_emp import PayServEmpPayElements as XsdPayServEmpPayElements
+
+
+class PayServEmpExtensionCreate(BaseModel):
+    """
+    Alias-driven flat model for NGA PayServEmpExtension (index-aware paths, no custom logic).
+    """
+    model_config = {
+        "populate_by_name": True
+    }
+    # Payment Instructions [0]
+    payment_valid_from: Optional[datetime.date] = Field(default=None, alias="payment_instructions[0].valid_from")
+    payment_type: Optional[str] = Field(default=None, alias="payment_instructions[0].payment_type")
+    payment_method: Optional[str] = Field(default=None, alias="payment_instructions[0].payment_method")
+    payment_type_code: Optional[str] = Field(default=None, alias="payment_instructions[0].payment_type_code")
+    local_payment_method: Optional[str] = Field(default=None, alias="payment_instructions[0].local_payment_method")
+    amount: Optional[list[str]] = Field(default=None, alias="payment_instructions[0].amount")
+    currency_code: Optional[str] = Field(default=None, alias="payment_instructions[0].currency_code")
+    payment_percentage: Optional[list[str]] = Field(default=None, alias="payment_instructions[0].payment_percentage")
+    name_on_account: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.name_on_account")
+    account_type_code: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.type_code[0]")
+    bank_routing_id: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.bank_routing_id[0]")
+    bank_routing_id_swift: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.bank_routing_id[1]")
+    bank_routing_id_swift_scheme_name: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.bank_routing_id[1].scheme_name")
+    account_id: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.account_id[0]")
+    iban: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.iban")
+    account_country_code: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.country_code[0]")
+    account_currency_code: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.currency_code")
+    additional_account_id: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.additional_account_id")
+    account_postal_code: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.postal_code")
+    account_city_name: Optional[str] = Field(default=None, alias="payment_instructions[0].direct_deposit_account.city_name")
+
+    # Payment Instructions [1] (second instruction, e.g., OTHER)
+    payment2_valid_from: Optional[datetime.date] = Field(default=None, alias="payment_instructions[1].valid_from")
+    payment2_type: Optional[str] = Field(default=None, alias="payment_instructions[1].payment_type")
+    payment2_method: Optional[str] = Field(default=None, alias="payment_instructions[1].payment_method")
+    payment2_type_code: Optional[str] = Field(default=None, alias="payment_instructions[1].payment_type_code")
+    payment2_amount: Optional[str] = Field(default=None, alias="payment_instructions[1].amount")
+    name_on_account2: Optional[str] = Field(default=None, alias="payment_instructions[1].direct_deposit_account.name_on_account")
+    account_type_code_2: Optional[str] = Field(default=None, alias="payment_instructions[1].direct_deposit_account.type_code[0]")
+    bank_routing_id2: Optional[str] = Field(default=None, alias="payment_instructions[1].direct_deposit_account.bank_routing_id[0]")
+    bank_routing_id2_swift: Optional[str] = Field(default=None, alias="payment_instructions[1].direct_deposit_account.bank_routing_id[1]")
+    bank_routing_id2_swift_scheme_name: Optional[str] = Field(default=None, alias="payment_instructions[1].direct_deposit_account.bank_routing_id[1].scheme_name")
+    account_id2: Optional[str] = Field(default=None, alias="payment_instructions[1].direct_deposit_account.account_id[0]")
+    additional_account_id2: Optional[str] = Field(default=None, alias="payment_instructions[1].direct_deposit_account.additional_account_id")
+    iban2: Optional[str] = Field(default=None, alias="payment_instructions[1].direct_deposit_account.iban")
+    account_country_code2: Optional[str] = Field(default=None, alias="payment_instructions[1].direct_deposit_account.country_code[0]")
+    account_currency_code2: Optional[str] = Field(default=None, alias="payment_instructions[1].direct_deposit_account.currency_code")
+
+    # Cost Assignment [0]
+    cost_valid_from: Optional[datetime.date] = Field(default=None, alias="cost_assignment[0].valid_from")
+    cost_center_code: Optional[str] = Field(default=None, alias="cost_assignment[0].cost_center_code")
+    cost_center_name: Optional[str] = Field(default=None, alias="cost_assignment[0].cost_center_name")
+    percentage: Optional[int] = Field(default=None, alias="cost_assignment[0].percentage")
+
+    # Pay Scales (single)
+    pay_scales_valid_from: Optional[datetime.date] = Field(default=None, alias="pay_scales.valid_from")
+    pay_scale_type: Optional[str] = Field(default=None, alias="pay_scales.pay_scale_type")
+    pay_scale_group: Optional[str] = Field(default=None, alias="pay_scales.pay_scale_group")
+    pay_scale_level: Optional[str] = Field(default=None, alias="pay_scales.pay_scale_level")
+
+    # Date Specifications (indices)
+    date_active_status_type: Optional[str] = Field(default=None, alias="date_specifications.date[0].date_type")
+    date_active_status: Optional[datetime.date] = Field(default=None, alias="date_specifications.date[0].value")
+    date_hire_type: Optional[str] = Field(default=None, alias="date_specifications.date[1].date_type")
+    date_hire: Optional[datetime.date] = Field(default=None, alias="date_specifications.date[1].value")
+    date_continuous_service_type: Optional[str] = Field(default=None, alias="date_specifications.date[2].date_type")
+    date_continuous_service: Optional[datetime.date] = Field(default=None, alias="date_specifications.date[2].value")
+    date_seniority_type: Optional[str] = Field(default=None, alias="date_specifications.date[3].date_type")
+    date_seniority: Optional[datetime.date] = Field(default=None, alias="date_specifications.date[3].value")
+    date_benefits_service_type: Optional[str] = Field(default=None, alias="date_specifications.date[4].date_type")
+    date_benefits_service: Optional[datetime.date] = Field(default=None, alias="date_specifications.date[4].value")
+    date_company_service_type: Optional[str] = Field(default=None, alias="date_specifications.date[5].date_type")
+    date_company_service: Optional[datetime.date] = Field(default=None, alias="date_specifications.date[5].value")
+    date_first_day_type: Optional[str] = Field(default=None, alias="date_specifications.date[6].date_type")
+    date_first_day_of_work: Optional[datetime.date] = Field(default=None, alias="date_specifications.date[6].value")
+
+    # Approvers [0]
+    approver_valid_from: Optional[datetime.date] = Field(default=None, alias="approvers.approver[0].valid_from")
+    approver_value: Optional[str] = Field(default=None, alias="approvers.approver[0].value")
+    approver_type: Optional[str] = Field(default=None, alias="approvers.approver[0].type_value")
+
+    # Contract Elements [0]
+    contract_valid_from: Optional[datetime.date] = Field(default=None, alias="contract_elements[0].valid_from")
+    contract_type: Optional[str] = Field(default=None, alias="contract_elements[0].contract_type")
+    contract_start_date: Optional[datetime.date] = Field(default=None, alias="contract_elements[0].contract_start_date")
+    contract_end_date: Optional[datetime.date] = Field(default=None, alias="contract_elements[0].contract_end_date")
+
+    # Payroll Specific Groupings (single object)
+    psg_valid_from: Optional[datetime.date] = Field(default=None, alias="payroll_specific_groupings.valid_from")
+    psg1: Optional[str] = Field(default=None, alias="payroll_specific_groupings.payroll_specific_grouping1")
+    psg2: Optional[str] = Field(default=None, alias="payroll_specific_groupings.payroll_specific_grouping2")
+    psg3: Optional[str] = Field(default=None, alias="payroll_specific_groupings.payroll_specific_grouping3")
+    psg4: Optional[str] = Field(default=None, alias="payroll_specific_groupings.payroll_specific_grouping4")
+    psg5: Optional[str] = Field(default=None, alias="payroll_specific_groupings.payroll_specific_grouping5")
+    psg6: Optional[str] = Field(default=None, alias="payroll_specific_groupings.payroll_specific_grouping6")
+    psg7: Optional[str] = Field(default=None, alias="payroll_specific_groupings.payroll_specific_grouping7")
+    psg8: Optional[str] = Field(default=None, alias="payroll_specific_groupings.payroll_specific_grouping8")
+    psg9: Optional[str] = Field(default=None, alias="payroll_specific_groupings.payroll_specific_grouping9")
+    psg10: Optional[str] = Field(default=None, alias="payroll_specific_groupings.payroll_specific_grouping10")
+
+    # Alternate Identifiers
+    payroll_exchange_id: Optional[str] = Field(default=None, alias="alternate_identifiers.payroll_exchange_id")
+    alternate_id: Optional[str] = Field(default=None, alias="alternate_identifiers.alternate_id[0]")
+    alternate_id_type: Optional[str] = Field(default=None, alias="alternate_identifiers.alternate_id[0].type_value")
+    pay_serv_id: Optional[str] = Field(default=None, alias="alternate_identifiers.pay_serv_id")
+    prior_incorrect_pay_serv_id: Optional[str] = Field(default=None, alias="alternate_identifiers.prior_incorrect_pay_serv_id")
+
+    # Alternate Descriptions
+    alt_desc_valid_from: Optional[datetime.date] = Field(default=None, alias="alternate_descriptions.valid_from")
+    alt_desc_1: Optional[str] = Field(default=None, alias="alternate_descriptions.description[0]")
+    alt_desc_1_type: Optional[str] = Field(default=None, alias="alternate_descriptions.description[0].type_value")
+    alt_desc_2: Optional[str] = Field(default=None, alias="alternate_descriptions.description[1]")
+    alt_desc_2_type: Optional[str] = Field(default=None, alias="alternate_descriptions.description[1].type_value")
+    alt_desc_3: Optional[str] = Field(default=None, alias="alternate_descriptions.description[2]")
+    alt_desc_3_type: Optional[str] = Field(default=None, alias="alternate_descriptions.description[2].type_value")
+
+    # Time Elements
+    time_elements: Optional[Absences] = None
+    time_quotas: Optional[TimeQuotas] = None
+
+    def to_nested_dict(self) -> Dict[str, Any]:
+        flat = self.model_dump(exclude_none=True, by_alias=True)
+        nested: Dict[str, Any] = {}
+        for k, v in flat.items():
+            add_to_nested_path(nested, k, v)
+        # Merge time elements
+        if self.time_elements is not None:
+            te = self.time_elements.to_nested_dict()
+            if te:
+                # Wrap into container
+                nested.setdefault("pay_serv_emp_time_elements", {})
+                nested["pay_serv_emp_time_elements"].update({"time_element": te.get("time_element", [])})
+        # Merge time quotas
+        if self.time_quotas is not None:
+            tq = self.time_quotas.to_nested_dict()
+            if tq:
+                nested.setdefault("pay_serv_emp_time_quotas", {})
+                nested["pay_serv_emp_time_quotas"].update({"time_quota": tq.get("time_quota", [])})
+        post_process_nested_data(nested, XsdPayServEmpExtension)
+        return nested
+
+    def to_model(self) -> XsdPayServEmpExtension:
+        nested = self.to_nested_dict()
+        try:
+            return construct_model(XsdPayServEmpExtension, nested)
+        except Exception:
+            return XsdPayServEmpExtension.model_validate(nested)
+
+
+class PayElementCreate(BaseModel):
+    """
+    Alias-driven flat model for a single PayElement item.
+    """
+    model_config = {
+        "populate_by_name": True
+    }
+    valid_from: Optional[datetime.date] = Field(default=None, alias="valid_from")
+    id: Optional[str] = Field(default=None, alias="id[0].value")
+    pay_element_type: Optional[str] = Field(default=None, alias="pay_element_type.value")
+    amount: Optional[str] = Field(default=None, alias="amount.value")
+    currency_code: Optional[str] = Field(default=None, alias="currency_code.value")
+    rate: Optional[str] = Field(default=None, alias="rate.value")
+    units: Optional[str] = Field(default=None, alias="units.value")
+    unit_type: Optional[str] = Field(default=None, alias="unit_type.value")
+    reference_number: Optional[str] = Field(default=None, alias="reference_number.value")
+    cost_center_code: Optional[str] = Field(default=None, alias="cost_center_code.value")
+    end_date: Optional[datetime.date] = Field(default=None, alias="end_date.value")
+    premium_id: Optional[str] = Field(default=None, alias="premium_id.value")
+    different_valuation: Optional[str] = Field(default=None, alias="different_valuation.value")
+    off_cycle_indicator: Optional[bool] = Field(default=None, alias="off_cycle_indicator.value")
+
+    def to_nested_dict(self) -> Dict[str, Any]:
+        flat = self.model_dump(exclude_none=True, by_alias=True)
+        nested: Dict[str, Any] = {}
+        for k, v in flat.items():
+            add_to_nested_path(nested, k, v)
+        # Ensure required wrappers and types explicitly (schema nuances)
+        # id must be a list of wrapper dicts
+        if "id" in nested:
+            if isinstance(nested["id"], list):
+                nested["id"] = [item if isinstance(item, dict) and "value" in item else {"value": item} for item in nested["id"]]
+            else:
+                nested["id"] = [{"value": nested["id"]}]
+        # Wrap simple fields (only those not already aliased as .value)
+        for key in ("pay_element_type", "amount", "currency_code"):
+            if key in nested and not (isinstance(nested[key], dict) and "value" in nested[key]):
+                nested[key] = {"value": nested[key]}
+        # Convert dates
+        if "valid_from" in nested and isinstance(nested["valid_from"], (datetime.date, datetime.datetime)):
+            nested["valid_from"] = convert_datetime_to_xml(nested["valid_from"], None)
+        if "end_date" in nested and isinstance(nested["end_date"], dict) and "value" in nested["end_date"]:
+            inner = nested["end_date"]["value"]
+            if isinstance(inner, (datetime.date, datetime.datetime)):
+                nested["end_date"]["value"] = convert_datetime_to_xml(inner, None)
+        # Final pass through schema container to catch any remaining wrappers
+        container = {"pay_element": [nested]}
+        post_process_nested_data(container, XsdPayServEmpPayElements)
+        return container["pay_element"][0]

brynq_sdk_alight/schemas/salary.py

@@ -0,0 +1,67 @@
+"""
+Flat, user-friendly salary model for Alight SDK.
+"""
+
+import datetime
+from typing import Optional, Dict, Any
+from pydantic import Field, BaseModel
+
+from .utils import add_to_nested_path  # not used directly; kept for parity if needed
+
+
+class Salary(BaseModel):
+    """
+    Simplified salary/compensation model.
+    Uses PURE schema-driven conversion - NO hardcoded structure mappings.
+    Uses aliases to match expected field names in Employee model.
+    """
+    model_config = {
+        "populate_by_name": True  # Allow populating by field name (base_salary) in addition to alias (pay_amount)
+    }
+
+    base_salary: float = Field(
+        description="Base salary amount",
+        alias="indicative_person_dossier.pay_cycle_remuneration.remuneration.amount.value"
+    )
+    currency_code: str = Field(
+        default="USD",
+        description="Currency code (USD/GBP/EUR)",
+        alias="indicative_person_dossier.pay_cycle_remuneration.remuneration.amount.currency_code"
+    )
+    pay_frequency: Optional[str] = Field(
+        default="Monthly",
+        description="Monthly/Weekly/Hourly",
+        alias="indicative_person_dossier.pay_cycle_remuneration.pay_cycle_interval_code"
+    )
+
+    # Pay Elements
+    element_code: Optional[str] = Field(
+        default="0010",
+        description="Pay element code",
+        alias="indicative_person_dossier.pay_cycle_remuneration.remuneration.pay_element.id"
+    )
+    element_type: Optional[str] = Field(
+        default="RECURRING",
+        description="RECURRING/BONUS/etc.",
+        alias="indicative_person_dossier.pay_cycle_remuneration.remuneration.pay_element.type_code"
+    )
+
+    # Dates
+    valid_from: Optional[datetime.date] = Field(
+        default=None,
+        description="Valid from date",
+        alias="indicative_person_dossier.pay_cycle_remuneration.valid_from"
+    )
+
+    def to_nested_dict(self) -> Dict[str, Any]:
+        """
+        Convert salary data to format suitable for pay elements in the HR-XML.
+        Using aliases for direct mapping to employee fields.
+        """
+        data = self.model_dump(exclude_none=True, by_alias=True)
+
+        # Ensure amount is a string for XML compatibility
+        if 'pay_amount' in data and isinstance(data['pay_amount'], (int, float)):
+            data['pay_amount'] = str(data['pay_amount'])
+
+        return data

brynq_sdk_alight/schemas/termination.py

@@ -0,0 +1,48 @@
+"""
+Flat, user-friendly termination model for Alight SDK.
+"""
+
+import datetime
+from typing import Optional, Dict, Any
+from pydantic import Field, BaseModel
+
+from .utils import add_to_nested_path  # not used here
+
+
+class Termination(BaseModel):
+    """
+    Simplified termination model.
+    Maps to IndicativeEmployment.employment_lifecycle.termination.
+    """
+    model_config = {
+        "populate_by_name": True
+    }
+
+    voluntary_termination_indicator: Optional[bool] = Field(
+        default=None,
+        description="Voluntary termination indicator",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.termination.voluntary_termination_indicator"
+    )
+    termination_reason_code: Optional[str] = Field(
+        default=None,
+        description="Reason for termination",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.termination.termination_reason_code"
+    )
+    termination_date: Optional[datetime.date] = Field(
+        default=None,
+        description="Termination date",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.termination.termination_date"
+    )
+    last_worked_date: Optional[datetime.date] = Field(
+        default=None,
+        description="Last worked date",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.termination.last_worked_date"
+    )
+    last_paid_date: Optional[datetime.date] = Field(
+        default=None,
+        description="Last paid date",
+        alias="indicative_person_dossier.indicative_employment.employment_lifecycle.termination.last_paid_date"
+    )
+
+    def to_nested_dict(self) -> Dict[str, Any]:
+        return self.model_dump(exclude_none=True, by_alias=True)

brynq_sdk_alight/schemas/timequota.py

@@ -0,0 +1,66 @@
+"""
+Flat, user-friendly time quota model for Alight SDK.
+"""
+
+import datetime
+from typing import Optional, Dict, Any, List
+from pydantic import Field, BaseModel
+
+from .utils import add_to_nested_path, convert_datetime_to_xml
+
+
+class TimeQuota(BaseModel):
+    """
+    Maps to PayServEmpTimeQuotas.time_quota[0].
+    """
+    model_config = {"populate_by_name": True}
+
+    id: Optional[str] = Field(default=None, alias="time_quota[0].id[0].value")
+    units: Optional[str] = Field(default=None, alias="time_quota[0].units")
+    units_taken: Optional[str] = Field(default=None, alias="time_quota[0].units_taken")
+    units_remaining: Optional[str] = Field(default=None, alias="time_quota[0].units_remaining")
+    unit_type: Optional[str] = Field(default=None, alias="time_quota[0].unit_type")
+    accrued_to_date: Optional[str] = Field(default=None, alias="time_quota[0].accrued_to_date")
+    deduction_start: Optional[datetime.date] = Field(default=None, alias="time_quota[0].deduction_start")
+    deduction_end: Optional[datetime.date] = Field(default=None, alias="time_quota[0].deduction_end")
+    valid_from: Optional[datetime.date] = Field(default=None, alias="time_quota[0].valid_from")
+    valid_to: Optional[datetime.date] = Field(default=None, alias="time_quota[0].valid_to")
+
+    def to_nested_dict(self) -> Dict[str, Any]:
+        flat = self.model_dump(exclude_none=True, by_alias=True)
+        nested: Dict[str, Any] = {}
+        for k, v in flat.items():
+            add_to_nested_path(nested, k, v)
+        # Convert date/period values to XML-compatible
+        for key in ("valid_from", "valid_to", "deduction_start", "deduction_end"):
+            node = nested
+            parts = key.split(".")
+            if key in nested and isinstance(nested[key], (datetime.date, datetime.datetime)):
+                nested[key] = convert_datetime_to_xml(nested[key], None)
+        return nested
+
+
+class TimeQuotas(BaseModel):
+    """
+    Wrapper for multiple TimeQuota items mapping to time_quota[*].
+    """
+    time_quotas: Optional[List[TimeQuota]] = Field(default=None, description="List of time quota items")
+
+    def to_nested_dict(self) -> Dict[str, Any]:
+        nested: Dict[str, Any] = {}
+        if not self.time_quotas:
+            return nested
+        items: List[Dict[str, Any]] = []
+        for idx, item in enumerate(self.time_quotas):
+            data = item.to_nested_dict()
+            remapped: Dict[str, Any] = {}
+            for k, v in data.items():
+                remapped[k.replace("time_quota[0]", f"time_quota[{idx}]")] = v
+            item_nested: Dict[str, Any] = {}
+            for k, v in remapped.items():
+                add_to_nested_path(item_nested, k, v)
+            if "time_quota" in item_nested and isinstance(item_nested["time_quota"], list):
+                items.extend(item_nested["time_quota"])
+        if items:
+            nested["time_quota"] = items
+        return nested