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

brynq_sdk_nmbrs/salaries.py

@@ -1,12 +1,13 @@
-import math
-from typing import Any, Dict
+from typing import Any, Dict, Union
 
 import pandas as pd
 import requests
+from zeep.exceptions import Fault
+from zeep.helpers import serialize_object
 
 from brynq_sdk_functions import Functions
 
-from .schemas.salary import SalaryCreate, SalaryGet
+from .schemas.salary import SalaryCreate, SalaryGet, SalaryUpdate
 
 
 class Salaries:
@@ -86,3 +87,74 @@ class Salaries:
             timeout=self.nmbrs.timeout
         )
         return resp
+
+    def get_current(self, employee_id: Union[int, str]) -> pd.DataFrame:
+        """
+        Get current salary for an employee via SOAP.
+
+        Args:
+            employee_id: The ID of the employee
+
+        Returns:
+            DataFrame with current salary
+        """
+        if self.nmbrs.mock_mode:
+            return pd.DataFrame()
+
+        try:
+            response = self.nmbrs.soap_client_employees.service.Salary_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 Salary: {str(e)}")
+
+    def update(self, data: Dict[str, Any]):
+        """
+        Update salary for an employee using SOAP API.
+
+        Args:
+            data: Dictionary containing salary data with fields matching SalaryUpdate schema:
+                - employee_id: Employee ID
+                - salary_value: Salary value (gross amount)
+                - salary_type: Salary type (1=FulltimeSalary, 2=ParttimeSalary, etc.)
+                - start_date: Start date (optional)
+
+        Returns:
+            Response from the API
+        """
+        try:
+            salary_model = SalaryUpdate(**data)
+
+            if self.nmbrs.mock_mode:
+                return salary_model
+
+            # Make SOAP request with dict - SalaryInput only has Value, Type, SalaryTable
+            response = self.nmbrs.soap_client_employees.service.Salary_UpdateCurrent(
+                EmployeeId=salary_model.employee_id,
+                Salary={
+                    'Value': salary_model.salary_value,
+                    'Type': salary_model.salary_type
+                },
+                _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 Salary: {str(e)}")

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

@@ -1,12 +1,15 @@
+from typing import Annotated, Optional
+
 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
+from pandera import Bool
+from pandera.typing import DateTime, Float, Series, String
 from pydantic import BaseModel, Field, StringConstraints
 
+from brynq_sdk_functions import BrynQPanderaDataFrameModel
+
+
 # ---------------------------
 # Get Schemas
 # ---------------------------
@@ -73,3 +76,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

@@ -1,20 +1,26 @@
 from datetime import datetime
-from typing import Dict, Any, Optional
+from typing import Any, Dict, Optional
+
 import pandas as pd
 import pandera as pa
-from pandera import Bool, Int
-from pandera.typing import Series, String, Float, DateTime
 import pandera.extensions as extensions
-from brynq_sdk_functions import BrynQPanderaDataFrameModel
+from pandera import Bool, Int
+from pandera.typing import DateTime, Float, Series, String
 from pydantic import BaseModel, Field
 
+from brynq_sdk_functions import BrynQPanderaDataFrameModel
+
+
 # ---------------------------
 # Get Schemas
 # ---------------------------
 class ScheduleGet(BrynQPanderaDataFrameModel):
     schedule_id: Series[String] = pa.Field(coerce=True, description="Schedule ID", alias="scheduleId")
-    start_date: Series[datetime] = pa.Field(coerce=True, description="Start Date", alias="startDate")
+    start_date_schedule: Series[datetime] = pa.Field(coerce=True, description="Start Date", alias="startDate")
     parttime_percentage: Series[Float] = pa.Field(coerce=True, description="Part-Time Percentage", alias="parttimePercentage")
+    hours_per_week: Series[Float] = pa.Field(coerce=True, nullable=True, description="Hours per week", alias="hoursPerWeek")
+    days_per_week: Series[Float] = pa.Field(coerce=True, nullable=True, description="Days per week", alias="daysPerWeek")
+
     week1_hours_monday: Series[Float] = pa.Field(coerce=True, description="Week 1 Hours Monday", alias="week1.hoursMonday")
     week1_hours_tuesday: Series[Float] = pa.Field(coerce=True, description="Week 1 Hours Tuesday", alias="week1.hoursTuesday")
     week1_hours_wednesday: Series[Float] = pa.Field(coerce=True, description="Week 1 Hours Wednesday", alias="week1.hoursWednesday")
@@ -42,19 +48,28 @@ class ScheduleGet(BrynQPanderaDataFrameModel):
             }
         }
 
-
 # ---------------------------
 # Upload Schemas
 # ---------------------------
-class ScheduleHours(BaseModel):
+class ScheduleHoursWeek1(BaseModel):
     """Schedule hours for each day of the week"""
-    hours_monday: Optional[float] = Field(None, description="Monday hours", alias="hoursMonday")
-    hours_tuesday: Optional[float] = Field(None, description="Tuesday hours", alias="hoursTuesday")
-    hours_wednesday: Optional[float] = Field(None, description="Wednesday hours", alias="hoursWednesday")
-    hours_thursday: Optional[float] = Field(None, description="Thursday hours", alias="hoursThursday")
-    hours_friday: Optional[float] = Field(None, description="Friday hours", alias="hoursFriday")
-    hours_saturday: Optional[float] = Field(None, description="Saturday hours", alias="hoursSaturday")
-    hours_sunday: Optional[float] = Field(None, description="Sunday hours", alias="hoursSunday")
+    week1_hours_monday: Optional[float] = Field(None, description="Monday hours", alias="hoursMonday")
+    week1_hours_tuesday: Optional[float] = Field(None, description="Tuesday hours", alias="hoursTuesday")
+    week1_hours_wednesday: Optional[float] = Field(None, description="Wednesday hours", alias="hoursWednesday")
+    week1_hours_thursday: Optional[float] = Field(None, description="Thursday hours", alias="hoursThursday")
+    week1_hours_friday: Optional[float] = Field(None, description="Friday hours", alias="hoursFriday")
+    week1_hours_saturday: Optional[float] = Field(None, description="Saturday hours", alias="hoursSaturday")
+    week1_hours_sunday: Optional[float] = Field(None, description="Sunday hours", alias="hoursSunday")
+
+class ScheduleHoursWeek2(BaseModel):
+    """Schedule hours for each day of the week"""
+    week2_hours_monday: Optional[float] = Field(None, description="Monday hours", alias="hoursMonday")
+    week2_hours_tuesday: Optional[float] = Field(None, description="Tuesday hours", alias="hoursTuesday")
+    week2_hours_wednesday: Optional[float] = Field(None, description="Wednesday hours", alias="hoursWednesday")
+    week2_hours_thursday: Optional[float] = Field(None, description="Thursday hours", alias="hoursThursday")
+    week2_hours_friday: Optional[float] = Field(None, description="Friday hours", alias="hoursFriday")
+    week2_hours_saturday: Optional[float] = Field(None, description="Saturday hours", alias="hoursSaturday")
+    week2_hours_sunday: Optional[float] = Field(None, description="Sunday hours", alias="hoursSunday")
 
 class ScheduleCreate(BaseModel):
     """
@@ -62,8 +77,9 @@ class ScheduleCreate(BaseModel):
     """
     start_date_schedule: datetime = Field(..., description="Start date of the schedule", example="2021-01-01T09:29:18Z", alias="startDate")
     hours_per_week: Optional[float] = Field(None, description="Hours per week", example=40, alias="hoursPerWeek")
-    week1: Optional[ScheduleHours] = Field(None, description="Week 1 schedule hours", alias="week1")
-    week2: Optional[ScheduleHours] = Field(None, description="Week 2 schedule hours", alias="week2")
+    # split per week schema so we can better sync with get fields.
+    week1: Optional[ScheduleHoursWeek1] = Field(None, description="Week 1 schedule hours", alias="week1")
+    week2: Optional[ScheduleHoursWeek2] = Field(None, description="Week 2 schedule hours", alias="week2")
 
     class Config:
         json_encoders = {
@@ -93,3 +109,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