brynq-sdk-nmbrs 2.3.1.dev0__py3-none-any.whl → 2.3.2.dev0__py3-none-any.whl

brynq_sdk_nmbrs/schedules.py

@@ -1,10 +1,11 @@
 import pandas as pd
 import requests
-import math
 from brynq_sdk_functions import Functions
-from .schemas.schedules import ScheduleGet, ScheduleCreate
+from .schemas.schedules import ScheduleGet, ScheduleCreate, ScheduleUpdate
 from datetime import datetime
-from typing import Dict, Any, Optional, Tuple
+from typing import Dict, Any, Tuple, Union
+from zeep.exceptions import Fault
+from zeep.helpers import serialize_object
 
 
 class Schedule:
@@ -82,3 +83,76 @@ class Schedule:
 
         except Exception as e:
             raise ValueError(f"Schedule validation failed: {str(e)}")
+
+    def update(self, data: Dict[str, Any]):
+        """
+        Update a schedule for an employee using SOAP API.
+
+        Args:
+            data: Dictionary containing schedule data with fields matching ScheduleUpdate schema:
+                - employeeId: Employee ID
+                - startDate: Start date of the schedule
+                - parttimePercentage: Part-time percentage (0-100)
+                - hoursMonday, hoursTuesday, etc.: Hours for each day of week 1
+                - hoursMonday2, hoursTuesday2, etc.: Hours for each day of week 2
+
+        Returns:
+            Response from the API
+        """
+        try:
+            schedule_model = ScheduleUpdate(**data)
+
+            if self.nmbrs.mock_mode:
+                return schedule_model
+
+            # Convert to SOAP object
+            schedule_soap = schedule_model.to_soap_schedule(self.nmbrs.soap_client_employees)
+
+            # Make SOAP request - CompanyRoosterNr is separate parameter (0 = no template)
+            response = self.nmbrs.soap_client_employees.service.Schedule_UpdateCurrent(
+                EmployeeId=schedule_model.employee_id,
+                Schedule=schedule_soap,
+                CompanyRoosterNr=schedule_model.company_rooster_nr or 0,
+                _soapheaders={'AuthHeaderWithDomain': self.nmbrs.soap_auth_header_employees}
+            )
+
+            return response
+
+        except Fault as e:
+            raise Exception(f"SOAP request failed: {str(e)}")
+        except Exception as e:
+            raise Exception(f"Failed to update Schedule: {str(e)}")
+
+    def get_current(self, employee_id: Union[int, str]) -> pd.DataFrame:
+        """
+        Get current schedule for an employee via SOAP.
+
+        Args:
+            employee_id: The ID of the employee
+
+        Returns:
+            DataFrame with current schedule
+        """
+        if self.nmbrs.mock_mode:
+            return pd.DataFrame()
+
+        try:
+            response = self.nmbrs.soap_client_employees.service.Schedule_GetCurrent(
+                EmployeeId=int(employee_id),
+                _soapheaders={'AuthHeaderWithDomain': self.nmbrs.soap_auth_header_employees}
+            )
+
+            if response:
+                serialized = serialize_object(response)
+                if not isinstance(serialized, list):
+                    serialized = [serialized]
+                df = pd.DataFrame(serialized)
+                df['employee_id'] = str(employee_id)
+                return df
+            else:
+                return pd.DataFrame()
+
+        except Fault as e:
+            raise Exception(f"SOAP request failed: {str(e)}")
+        except Exception as e:
+            raise Exception(f"Failed to get Schedule: {str(e)}")

brynq_sdk_nmbrs/schemas/address.py

@@ -73,3 +73,29 @@ class AddressCreate(BaseModel):
         )
     ] = Field(..., example="NL", description="Country ISO Code", alias="countryISOCode")
     period: Period
+
+
+class AddressDelete(BaseModel):
+    """Schema for deleting an address via SOAP API."""
+    employee_id: int = Field(..., example=12345, description="Employee ID", alias="employeeId")
+    address_id: int = Field(..., example=67890, description="Address ID to delete", alias="addressId")
+
+    class Config:
+        populate_by_name = True
+
+
+class AddressUpdate(BaseModel):
+    """Schema for updating an address via SOAP API."""
+    id: int = Field(..., example=67890, description="Address ID to update", alias="id")
+    default: bool = Field(False, example=False, description="Is default address", alias="default")
+    street: str = Field(..., min_length=1, max_length=200, example="Naritaweg", description="Street", alias="street")
+    house_number: Optional[str] = Field(None, max_length=20, example="70", description="House Number", alias="houseNumber")
+    house_number_addition: Optional[str] = Field(None, max_length=20, example="A", description="House Number Addition", alias="houseNumberAddition")
+    postal_code: Optional[str] = Field(None, max_length=15, example="1043BZ", description="Postal Code", alias="postalCode")
+    city: str = Field(..., min_length=1, max_length=100, example="Amsterdam", description="City", alias="city")
+    state_province: Optional[str] = Field(None, max_length=100, example="Noord-Holland", description="State or Province", alias="stateProvince")
+    country_iso_code: str = Field(..., min_length=2, max_length=3, example="NL", description="Country ISO Code", alias="countryISOCode")
+    address_type: str = Field("HomeAddress", example="HomeAddress", description="Address type", alias="type")
+
+    class Config:
+        populate_by_name = True

brynq_sdk_nmbrs/schemas/children.py

@@ -0,0 +1,67 @@
+import pandas as pd
+import pandera as pa
+from pandera import Bool
+from pandera.typing import Series, String, DateTime
+from brynq_sdk_functions import BrynQPanderaDataFrameModel
+from typing import Optional, Literal
+from pydantic import BaseModel, Field
+from datetime import datetime
+
+# ---------------------------
+# Get Schemas
+# ---------------------------
+class ChildrenGet(BrynQPanderaDataFrameModel):
+    employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
+    child_id: Series[String] = pa.Field(coerce=True, description="Child ID", alias="childId")
+    name: Series[String] = pa.Field(coerce=True, description="Last Name", alias="name")
+    first_name: Series[String] = pa.Field(coerce=True, description="First Name", alias="firstName")
+    initials: Series[String] = pa.Field(coerce=True, nullable=True, description="Initials", alias="initials")
+    gender: Series[String] = pa.Field(coerce=True, description="Gender", alias="gender")
+    birthday: Series[DateTime] = pa.Field(coerce=True, description="Birthday", alias="birthday")
+
+    class _Annotation:
+        primary_key = "child_id"
+        foreign_keys = {
+            "employee_id": {
+                "parent_schema": "EmployeeSchema",
+                "parent_column": "employee_id",
+                "cardinality": "N:1"
+            }
+        }
+
+# ---------------------------
+# Upload Schemas
+# ---------------------------
+class ChildCreate(BaseModel):
+    """Schema for creating a child via SOAP API."""
+    name: str = Field(..., min_length=1, max_length=100, example="Doe", description="Last Name", alias="name")
+    first_name: str = Field(..., min_length=1, max_length=100, example="John", description="First Name", alias="firstName")
+    initials: Optional[str] = Field(None, max_length=20, example="J.D.", description="Initials", alias="initials")
+    gender: Literal["male", "female", "unknown", "undefined"] = Field(..., example="male", description="Gender", alias="gender")
+    birthday: datetime = Field(..., example="2020-01-01T00:00:00", description="Birthday", alias="birthday")
+
+    class Config:
+        populate_by_name = True
+
+
+class ChildDelete(BaseModel):
+    """Schema for deleting a child via SOAP API."""
+    employee_id: int = Field(..., example=12345, description="Employee ID", alias="employeeId")
+    child_id: int = Field(..., example=67890, description="Child ID to delete", alias="childId")
+
+    class Config:
+        populate_by_name = True
+
+
+class ChildUpdate(BaseModel):
+    """Schema for updating a child via SOAP API."""
+    id: int = Field(..., example=67890, description="Child ID to update", alias="id")
+    name: str = Field(..., min_length=1, max_length=100, example="Doe", description="Last Name", alias="name")
+    first_name: str = Field(..., min_length=1, max_length=100, example="John", description="First Name", alias="firstName")
+    initials: Optional[str] = Field(None, max_length=20, example="J.D.", description="Initials", alias="initials")
+    gender: Literal["male", "female", "unknown", "undefined"] = Field(..., example="male", description="Gender", alias="gender")
+    birthday: datetime = Field(..., example="2020-01-01T00:00:00", description="Birthday", alias="birthday")
+
+    class Config:
+        populate_by_name = True
+

brynq_sdk_nmbrs/schemas/company.py

@@ -0,0 +1,16 @@
+from pydantic import BaseModel, Field
+from typing import Optional
+
+
+class CompanyCreate(BaseModel):
+    """Schema for creating a company via SOAP API."""
+    debtor_id: int = Field(..., description="Debtor ID", alias="debtorId", example=34548)
+    company_name: str = Field(..., min_length=1, max_length=200, description="Company name", alias="companyName", example="New Company BV")
+    period_type: int = Field(1, ge=1, le=4, description="Period type (1=Monthly, 2=4-Weekly, 3=Weekly, 4=Quarterly)", alias="periodType", example=1)
+    default_company_id: int = Field(0, description="Default company ID to copy settings from (0 for none)", alias="defaultCompanyId", example=0)
+    labour_agreement_settings_group_guid: Optional[str] = Field(None, description="Labour agreement settings group GUID", alias="labourAgreementSettingsGroupGuid", example="00000000-0000-0000-0000-000000000000")
+    pay_in_advance: bool = Field(False, description="Pay in advance", alias="payInAdvance", example=False)
+
+    class Config:
+        populate_by_name = True
+

brynq_sdk_nmbrs/schemas/debtor.py

@@ -13,4 +13,26 @@ from pydantic import BaseModel, Field, StringConstraints
 class DebtorsGet(BrynQPanderaDataFrameModel):
     debtor_id: Series[String] = pa.Field(coerce=True, description="Debtor ID", alias="debtorId")
     number: Series[String] = pa.Field(coerce=True, description="Debtor number", alias="number")
-    name: Series[Bool] = pa.Field(coerce=True, description="Debtor name", alias="name")
+    name: Series[Bool] = pa.Field(coerce=True, description="Debtor name", alias="name")
+
+
+# ---------------------------
+# SOAP Schemas
+# ---------------------------
+class DebtorCreate(BaseModel):
+    """Schema for creating a debtor via SOAP API."""
+    number: int = Field(..., ge=1, description="Debtor number", alias="number", example=1001)
+    name: str = Field(..., min_length=1, max_length=200, description="Debtor name", alias="name", example="New Debtor Company")
+
+    class Config:
+        populate_by_name = True
+
+
+class DebtorUpdate(BaseModel):
+    """Schema for updating a debtor via SOAP API."""
+    debtor_id: int = Field(..., description="Debtor ID", alias="debtorId", example=34548)
+    number: int = Field(..., ge=1, description="Debtor number", alias="number", example=1001)
+    name: str = Field(..., min_length=1, max_length=200, description="Debtor name", alias="name", example="Updated Debtor Company")
+
+    class Config:
+        populate_by_name = True

brynq_sdk_nmbrs/schemas/department.py

@@ -56,3 +56,36 @@ class EmployeeDepartmentUpdate(BaseModel):
 
     class Config:
         primary_key = "departmentId"
+
+
+# ---------------------------
+# SOAP Schemas (DebtorService)
+# ---------------------------
+class DepartmentMasterCreate(BaseModel):
+    """Schema for creating a master department via SOAP API."""
+    debtor_id: int = Field(..., description="Debtor ID", alias="debtorId", example=34548)
+    code: int = Field(..., ge=1, description="Department Code", alias="code", example=101)
+    description: str = Field(..., min_length=1, max_length=200, description="Department Description", alias="description", example="Engineering")
+
+    class Config:
+        populate_by_name = True
+
+
+class DepartmentMasterUpdate(BaseModel):
+    """Schema for updating a master department via SOAP API."""
+    debtor_id: int = Field(..., description="Debtor ID", alias="debtorId", example=34548)
+    department_id: int = Field(..., description="Department ID", alias="departmentId", example=12345)
+    code: int = Field(..., ge=1, description="Department Code", alias="code", example=101)
+    description: str = Field(..., min_length=1, max_length=200, description="Department Description", alias="description", example="Engineering Updated")
+
+    class Config:
+        populate_by_name = True
+
+
+class DepartmentMasterDelete(BaseModel):
+    """Schema for deleting a master department via SOAP API."""
+    debtor_id: int = Field(..., description="Debtor ID", alias="debtorId", example=34548)
+    department_id: int = Field(..., description="Department ID to delete", alias="departmentId", example=12345)
+
+    class Config:
+        populate_by_name = True

brynq_sdk_nmbrs/schemas/document.py

@@ -0,0 +1,13 @@
+from pydantic import BaseModel, Field
+from typing import Optional
+
+
+class DocumentUpload(BaseModel):
+    """Schema for uploading a document to an employee via SOAP API."""
+    employee_id: int = Field(..., description="Employee ID", alias="employeeId", example=276967)
+    document_name: str = Field(..., description="Document name (with extension)", alias="documentName", example="contract.pdf")
+    document_type_guid: str = Field(..., description="Document type GUID", alias="documentTypeGuid", example="00000000-0000-0000-0000-000000000000")
+
+    class Config:
+        populate_by_name = True
+

brynq_sdk_nmbrs/schemas/employees.py

@@ -25,7 +25,9 @@ class EmployeeGet(BrynQPanderaDataFrameModel):
     last_name: Series[String] = pa.Field(coerce=True, description="Last Name", alias="basicInfo.lastName")
     employee_type: Series[String] = pa.Field(coerce=True, description="Employee Type", alias="basicInfo.employeeType")
     birth_date: Series[DateTime] = pa.Field(coerce=True, description="Birth Date", alias="birthInfo.birthDate")
-    birth_country_code_iso: Series[String] = pa.Field(coerce=True, nullable=True, description="Birth Country Code ISO", alias="birthInfo.birthCountry")
+    birth_country_code_iso: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Birth Country Code ISO", alias="birthInfo.birthCountry")
+    birth_country: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Birth Country Code ISO", alias="birthInfo.birthCountry.codeISO")
+    birth_country_description: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Birth Country Code ISO", alias="birthInfo.birthCountry.description")
     nationality_code_iso: Series[String] = pa.Field(coerce=True, nullable=True, description="Nationality Code ISO", alias="birthInfo.nationality.codeISO")
     gender: Series[String] = pa.Field(coerce=True, nullable=True, description="Gender", alias="birthInfo.gender")
     private_email: Series[String] = pa.Field(coerce=True, nullable=True, description="Private Email", alias="contactInfo.privateEmail")

brynq_sdk_nmbrs/schemas/function.py

@@ -47,3 +47,31 @@ class FunctionGet(BrynQPanderaDataFrameModel):
 class FunctionUpdate(BaseModel):
     function_id: str = Field(..., example="5981", description="Function ID", alias="functionId")
     period_details: Period = Field(..., alias="periodDetails")
+
+
+# SOAP Schemas for DebtorService Functions
+class FunctionCreate(BaseModel):
+    """Schema for creating a master function via SOAP API."""
+    code: int = Field(..., example=101, description="Function Code", alias="code")
+    description: str = Field(..., example="Software Developer", description="Function Description", alias="description")
+
+    class Config:
+        populate_by_name = True
+
+
+class FunctionDelete(BaseModel):
+    """Schema for deleting a master function via SOAP API."""
+    function_id: int = Field(..., example=12345, description="Function ID to delete", alias="functionId")
+
+    class Config:
+        populate_by_name = True
+
+
+class FunctionMasterUpdate(BaseModel):
+    """Schema for updating a master function via SOAP API."""
+    function_id: int = Field(..., example=12345, description="Function ID", alias="functionId")
+    code: int = Field(..., example=101, description="Function Code", alias="code")
+    description: str = Field(..., example="Senior Software Developer", description="Function Description", alias="description")
+
+    class Config:
+        populate_by_name = True

brynq_sdk_nmbrs/schemas/leave.py

@@ -46,6 +46,18 @@ class LeaveDelete(BaseModel):
     leave_request_id: str = Field(..., example="49a69eda-252e-4ccb-a220-38ea90511d4f", description="Leave Request ID", alias="leaveRequestId")
 
 
+class LeaveUpdate(BaseModel):
+    """Schema for updating leave via SOAP API."""
+    employee_id: int = Field(..., example=12345, description="Employee ID", alias="employeeId")
+    leave_id: int = Field(..., example=67890, description="Leave ID", alias="leaveId")
+    start_date: datetime = Field(..., example="2025-01-01T00:00:00", description="Leave Start Date", alias="startDate")
+    end_date: datetime = Field(..., example="2025-01-02T00:00:00", description="Leave End Date", alias="endDate")
+    description: Optional[str] = Field(None, example="Vacation leave", description="Description", alias="description")
+
+    class Config:
+        populate_by_name = True
+
+
 class LeaveBalanceGet(BrynQPanderaDataFrameModel):
     employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
     leave_group_id: Series[String] = pa.Field(coerce=True, nullable=True, description="Leave Group ID", alias="leaveGroupId")

brynq_sdk_nmbrs/schemas/salary.py

@@ -76,6 +76,24 @@ class SalaryCreate(BaseModel):
                 data.pop('salaryTable')
         return data
 
+    class Config:
+        populate_by_name = True
+
+
+class SalaryUpdate(BaseModel):
+    """Schema for updating salary via SOAP API (Salary_UpdateCurrent)."""
+    employee_id: int = Field(..., example=12345, description="Employee ID", alias="employeeId")
+    salary_value: float = Field(..., ge=0, example=5000.00, description="Salary value (gross amount)", alias="salaryValue")
+    salary_type: str = Field(
+        "Bruto_Salaris_Fulltime",
+        example="Bruto_Salaris_Fulltime",
+        description="Salary type enum: Bruto_Salaris_Fulltime, Bruto_Salaris_Parttime, Bruto_Uurloon, etc.",
+        alias="salaryType"
+    )
+
+    class Config:
+        populate_by_name = True
+
 class SalaryTableGet(BrynQPanderaDataFrameModel):
     code: Series[String] = pa.Field(coerce=True, description="Salary Code", alias="Code")
     description: Series[String] = pa.Field(coerce=True, description="Salary Description", alias="Description")

brynq_sdk_nmbrs/schemas/schedules.py

@@ -93,3 +93,52 @@ class ScheduleCreate(BaseModel):
                 }
             }
         }
+
+
+class ScheduleUpdate(BaseModel):
+    """
+    Pydantic model for updating a schedule via SOAP API
+    """
+    employee_id: int = Field(..., example=12345, description="Employee ID", alias="employeeId")
+    start_date: datetime = Field(..., example="2025-01-01T00:00:00", description="Start date of the schedule", alias="startDate")
+    parttime_percentage: float = Field(..., ge=0, le=100, example=100.0, description="Part-time percentage", alias="parttimePercentage")
+    company_rooster_nr: Optional[int] = Field(0, example=0, description="Company Rooster Number (schedule template)", alias="companyRoosterNr")
+    hours_monday: Optional[float] = Field(0, ge=0, le=24, example=8.0, description="Hours Monday", alias="hoursMonday")
+    hours_tuesday: Optional[float] = Field(0, ge=0, le=24, example=8.0, description="Hours Tuesday", alias="hoursTuesday")
+    hours_wednesday: Optional[float] = Field(0, ge=0, le=24, example=8.0, description="Hours Wednesday", alias="hoursWednesday")
+    hours_thursday: Optional[float] = Field(0, ge=0, le=24, example=8.0, description="Hours Thursday", alias="hoursThursday")
+    hours_friday: Optional[float] = Field(0, ge=0, le=24, example=8.0, description="Hours Friday", alias="hoursFriday")
+    hours_saturday: Optional[float] = Field(0, ge=0, le=24, example=0.0, description="Hours Saturday", alias="hoursSaturday")
+    hours_sunday: Optional[float] = Field(0, ge=0, le=24, example=0.0, description="Hours Sunday", alias="hoursSunday")
+    hours_monday2: Optional[float] = Field(0, ge=0, le=24, example=8.0, description="Hours Monday Week 2", alias="hoursMonday2")
+    hours_tuesday2: Optional[float] = Field(0, ge=0, le=24, example=8.0, description="Hours Tuesday Week 2", alias="hoursTuesday2")
+    hours_wednesday2: Optional[float] = Field(0, ge=0, le=24, example=8.0, description="Hours Wednesday Week 2", alias="hoursWednesday2")
+    hours_thursday2: Optional[float] = Field(0, ge=0, le=24, example=8.0, description="Hours Thursday Week 2", alias="hoursThursday2")
+    hours_friday2: Optional[float] = Field(0, ge=0, le=24, example=8.0, description="Hours Friday Week 2", alias="hoursFriday2")
+    hours_saturday2: Optional[float] = Field(0, ge=0, le=24, example=0.0, description="Hours Saturday Week 2", alias="hoursSaturday2")
+    hours_sunday2: Optional[float] = Field(0, ge=0, le=24, example=0.0, description="Hours Sunday Week 2", alias="hoursSunday2")
+
+    def to_soap_schedule(self, soap_client):
+        """Convert to SOAP Schedule object"""
+        ScheduleType = soap_client.get_type('ns0:Schedule')
+        return ScheduleType(
+            StartDate=self.start_date,
+            ParttimePercentage=self.parttime_percentage,
+            HoursMonday=self.hours_monday or 0,
+            HoursTuesday=self.hours_tuesday or 0,
+            HoursWednesday=self.hours_wednesday or 0,
+            HoursThursday=self.hours_thursday or 0,
+            HoursFriday=self.hours_friday or 0,
+            HoursSaturday=self.hours_saturday or 0,
+            HoursSunday=self.hours_sunday or 0,
+            HoursMonday2=self.hours_monday2 or 0,
+            HoursTuesday2=self.hours_tuesday2 or 0,
+            HoursWednesday2=self.hours_wednesday2 or 0,
+            HoursThursday2=self.hours_thursday2 or 0,
+            HoursFriday2=self.hours_friday2 or 0,
+            HoursSaturday2=self.hours_saturday2 or 0,
+            HoursSunday2=self.hours_sunday2 or 0
+        )
+
+    class Config:
+        populate_by_name = True

brynq_sdk_nmbrs/schemas/social_insurance.py

@@ -1,14 +1,47 @@
-import math
 import pandas as pd
 import pandera as pa
 from pandera import Bool
-from pandera.typing import Series, String, Float, DateTime
-import pandera.extensions as extensions
+from pandera.typing import Series, String, DateTime
 from brynq_sdk_functions import BrynQPanderaDataFrameModel
-from typing import Optional, Annotated
-from pydantic import BaseModel, Field, StringConstraints
-from datetime import datetime
+from typing import Optional
+from pydantic import BaseModel, Field
 
+# ---------------------------
+# Get Schemas
+# ---------------------------
+class SocialInsuranceGet(BrynQPanderaDataFrameModel):
+    employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employee_id")
+    id: Series[String] = pa.Field(coerce=True, description="SVW Settings ID", alias="Id")
+    creation_date: Series[DateTime] = pa.Field(coerce=True, nullable=True, description="Creation Date", alias="CreationDate")
+    start_period: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Start Period", alias="StartPeriod")
+    start_year: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Start Year", alias="StartYear")
+    influence_obliged_insurance: Series[Bool] = pa.Field(coerce=True, nullable=True, description="Whether employee influences obligatory insurance status", alias="InfluenceObligedInsurance")
+    wage_cost_benefit: Series[Bool] = pa.Field(coerce=True, nullable=True, description="Eligible for wage cost subsidy (LKV - Loonkostenvoordeel)", alias="WageCostBenefit")
+    cao: Series[String] = pa.Field(coerce=True, nullable=True, description="Collectieve Arbeidsovereenkomst (Collective Labor Agreement)", alias="CAO")
+    wao_wia: Series[Bool] = pa.Field(coerce=True, nullable=True, description="Disability Insurance (WAO/WIA - Wet Arbeidsongeschiktheid)", alias="Wao_Wia")
+    ww: Series[Bool] = pa.Field(coerce=True, nullable=True, description="Unemployment Insurance (WW - Werkloosheidswet)", alias="Ww")
+    zw: Series[Bool] = pa.Field(coerce=True, nullable=True, description="Sickness Benefits (ZW - Ziektewet)", alias="Zw")
+    income_related_contribution_zvw: Series[Bool] = pa.Field(coerce=True, nullable=True, description="Income Related Health Insurance Contribution (ZVW - Zorgverzekeringswet)", alias="IncomeRelatedContributionZvw")
+    code_zvw: Series[String] = pa.Field(coerce=True, nullable=True, description="Health Insurance Code (ZVW)", alias="CodeZvw")
+    risk_group: Series[String] = pa.Field(coerce=True, nullable=True, description="Risk Group for premium calculation", alias="RiskGroup")
+    sector: Series[String] = pa.Field(coerce=True, nullable=True, description="Industry Sector code", alias="Sector")
+    employment_type: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Employment Type", alias="EmploymentType")
+    phase_classification: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Phase Classification", alias="PhaseClassification")
+    employment_sequence_tax_id: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Employment Sequence Tax ID", alias="EmploymentSequenceTaxId")
+
+    class _Annotation:
+        primary_key = "id"
+        foreign_keys = {
+            "employee_id": {
+                "parent_schema": "EmployeeSchema",
+                "parent_column": "employee_id",
+                "cardinality": "N:1"
+            }
+        }
+
+# ---------------------------
+# Upload Schemas
+# ---------------------------
 class SocialInsuranceUpdate(BaseModel):
     employee_id: int = Field(None, example="1234567890", description="Employee ID", alias="EmployeeId")
     influence_obliged_insurance: Optional[bool] = Field(None, example="1234567890", description="Influence Obliged Insurance", alias="InfluenceObligedInsurance")

brynq_sdk_nmbrs/schemas/wage_tax.py

@@ -1,16 +1,76 @@
 import math
+from datetime import datetime
+from typing import Annotated, Optional, Union
+
 import pandas as pd
 import pandera as pa
-from pandera import Bool
-from pandera.typing import Series, String, Float, DateTime
 import pandera.extensions as extensions
-from brynq_sdk_functions import BrynQPanderaDataFrameModel
-from typing import Optional, Annotated, Union
+from pandera import Bool
+from pandera.typing import DateTime, Float, Series, String
 from pydantic import BaseModel, Field, StringConstraints
-from datetime import datetime
+
+from brynq_sdk_functions import BrynQPanderaDataFrameModel
 
 
+# ---------------------------
+# REST API Get Schema (Wage Tax Settings History)
+# ---------------------------
 class WageTaxSettingsGet(BrynQPanderaDataFrameModel):
+    """Schema for REST API: GET /companies/{companyId}/employees/wagetaxsettings"""
+    employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
+    wage_tax_setting_id: Series[String] = pa.Field(coerce=True, description="Wage Tax Setting ID", alias="wageTaxSettingId")
+    start_date: Series[DateTime] = pa.Field(coerce=True, description="Start Date", alias="startDate")
+    created_at: Series[DateTime] = pa.Field(coerce=True, description="Created At", alias="createdAt")
+    payroll_tax_reduction: Series[Bool] = pa.Field(coerce=True, nullable=True, description="Payroll Tax Reduction", alias="payrollTaxReduction")
+    type_of_income: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Type of Income", alias="typeOfIncome")
+    color_table: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Color Table", alias="colorTable")
+    period_table: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Period Table", alias="periodTable")
+    special_table: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Special Table", alias="specialTable")
+    payroll_tax: Series[Bool] = pa.Field(coerce=True, nullable=True, description="Payroll Tax", alias="payrollTax")
+    benefit_scheme: Series[Bool] = pa.Field(coerce=True, nullable=True, description="Benefit Scheme", alias="benefitScheme")
+    auto_small_jobs: Series[Bool] = pa.Field(coerce=True, nullable=True, description="Auto Small Jobs Regulation", alias="autoSmallJobs")
+    yearly_salary: Series[pd.Float64Dtype] = pa.Field(coerce=True, nullable=True, description="Yearly Salary", alias="yearlySalary")
+    deviation_special_rate: Series[pd.Float64Dtype] = pa.Field(coerce=True, nullable=True, description="Deviation Special Rate Payroll Tax Deduction", alias="deviationSpecialRatePayrollTaxDeduction")
+    code_tax_reduction: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Code Tax Reduction", alias="codeTaxReduction")
+    holiday_vouchers: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Holiday Vouchers", alias="holidayVouchers")
+    code_30_percent_rule: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Code Calculate 30 Percent Rule", alias="codeCalculate30PercentRule")
+
+    class _Annotation:
+        primary_key = "wage_tax_setting_id"
+        foreign_keys = {
+            "employee_id": {
+                "parent_schema": "EmployeeSchema",
+                "parent_column": "employee_id",
+                "cardinality": "N:1"
+            }
+        }
+
+
+class WageTaxSettingsCreate(BaseModel):
+    """Schema for REST API: POST /employees/{employeeId}/wagetaxsetting"""
+    start_date: datetime = Field(..., example="2024-01-01T00:00:00Z", description="Start Date", alias="startDate")
+    payroll_tax_reduction: Optional[bool] = Field(None, example=True, description="Payroll Tax Reduction", alias="payrollTaxReduction")
+    type_of_income: Optional[int] = Field(None, example=15, description="Type of Income", alias="typeOfIncome")
+    color_table: Optional[int] = Field(None, example=1, description="Color Table", alias="colorTable")
+    period_table: Optional[int] = Field(None, example=2, description="Period Table", alias="periodTable")
+    special_table: Optional[int] = Field(None, example=0, description="Special Table", alias="specialTable")
+    payroll_tax: Optional[bool] = Field(None, example=True, description="Payroll Tax", alias="payrollTax")
+    benefit_scheme: Optional[bool] = Field(None, example=True, description="Benefit Scheme", alias="benefitScheme")
+    auto_small_jobs: Optional[bool] = Field(None, example=False, description="Auto Small Jobs Regulation", alias="autoSmallJobs")
+    yearly_salary: Optional[float] = Field(None, example=45000.00, description="Yearly Salary", alias="yearlySalary")
+    deviation_special_rate: Optional[float] = Field(None, example=0.0, description="Deviation Special Rate Payroll Tax Deduction", alias="deviationSpecialRatePayrollTaxDeduction")
+    code_tax_reduction: Optional[int] = Field(None, example=0, description="Code Tax Reduction", alias="codeTaxReduction")
+    holiday_vouchers: Optional[int] = Field(None, example=0, description="Holiday Vouchers", alias="holidayVouchers")
+    code_30_percent_rule: Optional[int] = Field(None, example=0, description="Code Calculate 30 Percent Rule", alias="codeCalculate30PercentRule")
+
+    class Config:
+        populate_by_name = True
+
+
+# ---------------------------
+# SOAP API Get Schema (Company Wage Tax List)
+# ---------------------------
+class CompanyWageTaxGet(BrynQPanderaDataFrameModel):
     wagetax_id: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Loonaangifte ID", alias="LoonaangifteID")
     serial_number: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Serial Number", alias="SerialNumber")
     payment_reference: Series[String] = pa.Field(coerce=True, description="Payment Reference", alias="PaymentReference")