brynq-sdk-nmbrs 2.3.4.dev0__py3-none-any.whl → 2.4.5__py3-none-any.whl

brynq_sdk_nmbrs/manager.py

@@ -1,11 +1,19 @@
+from typing import Any, Dict
+
 import pandas as pd
 import requests
+
 from brynq_sdk_functions import Functions as BrynQFunctions
-import math
-from typing import Dict, Any
+
 from .schemas.manager import (
-    ManagerGet, ManagerBasicGet, EmployeeManagerGet, ManagerHistoricBasicGet,
-    ManagerCreate, ManagerUpdate, ManagerDelete, UpdateEmployeeManager
+    EmployeeManagerGet,
+    ManagerBasicGet,
+    ManagerCreate,
+    ManagerDelete,
+    ManagerGet,
+    ManagerHistoricBasicGet,
+    ManagerUpdate,
+    UpdateEmployeeManager,
 )
 
 
@@ -127,16 +135,17 @@ class EmployeeManager:
         Returns:
             Response from the API
         """
-        # Validate with Pydantic model
         manager_model = UpdateEmployeeManager(**data)
 
         if self.nmbrs.mock_mode:
             return manager_model
 
-        # Convert validated model to dict for API payload
-        payload = manager_model.dict(exclude_none=True)
+        payload = manager_model.model_dump(by_alias=True, exclude_none=True)
+
+        # because of nmbrs bug, it needs to be passed (without year and period) as empty dict if not in the data.
+        if 'periodDetails' not in payload or payload.get('periodDetails') is None:
+            payload['periodDetails'] = {}
 
-        # Send request
         resp = self.nmbrs.session.put(
             url=f"{self.nmbrs.base_url}employees/{employee_id}/manager",
             json=payload,
@@ -149,16 +158,33 @@ class Manager:
     def __init__(self, nmbrs):
         self.nmbrs = nmbrs
 
-    def get(self, debtor_id: str) -> tuple[pd.DataFrame, pd.DataFrame]:
+    def get(self, debtor_id: str = None) -> tuple[pd.DataFrame, pd.DataFrame]:
         """
-        Get all managers for a specific debtor.
+        Get all managers for debtor(s).
 
         Args:
-            debtor_id: The ID of the debtor
+            debtor_id: Optional debtor ID. If not provided, fetches for all debtors.
 
         Returns:
             Tuple of (valid_managers, invalid_managers) DataFrames
         """
+        if debtor_id:
+            df = self._get(debtor_id)
+        else:
+            df = pd.DataFrame()
+            for debtor in self.nmbrs.debtor_ids:
+                df = pd.concat([df, self._get(debtor)])
+
+        if df.empty:
+            return df, pd.DataFrame()
+
+        valid_managers, invalid_managers = BrynQFunctions.validate_data(df=df, schema=ManagerGet, debug=self.nmbrs.debug)
+        return valid_managers, invalid_managers
+
+    def _get(self, debtor_id: str) -> pd.DataFrame:
+        """
+        Internal method to get managers for a single debtor.
+        """
         try:
             request = requests.Request(
                 method='GET',
@@ -169,42 +195,32 @@ class Manager:
             df = pd.DataFrame(data)
             df['debtorId'] = debtor_id
 
-            valid_managers, invalid_managers = BrynQFunctions.validate_data(df=df, schema=ManagerGet, debug=True)
-
-        except requests.HTTPError as e:
+        except requests.HTTPError:
             df = pd.DataFrame()
-            valid_managers = df
-            invalid_managers = df
 
-        return valid_managers, invalid_managers
+        return df
 
-    def get_basic(self, debtor_id: str) -> tuple[pd.DataFrame, pd.DataFrame]:
+    def get_basic(self, debtor_id: str = None) -> tuple[pd.DataFrame, pd.DataFrame]:
         """
-        Get basic manager information for a specific debtor.
+        Get basic manager information for debtor(s).
 
         Args:
-            debtor_id: The ID of the debtor
+            debtor_id: Optional debtor ID. If not provided, fetches for all debtors.
 
         Returns:
             Tuple of (valid_managers, invalid_managers) DataFrames
         """
-        try:
-            request = requests.Request(
-                method='GET',
-                url=f"{self.nmbrs.base_url}debtors/{debtor_id}/managers"
-            )
-
-            data = self.nmbrs.get_paginated_result(request)
-            df = pd.DataFrame(data)
-            df['debtorId'] = debtor_id
-
-            valid_managers, invalid_managers = BrynQFunctions.validate_data(df=df, schema=ManagerBasicGet, debug=True)
-
-        except requests.HTTPError as e:
+        if debtor_id:
+            df = self._get(debtor_id)
+        else:
             df = pd.DataFrame()
-            valid_managers = df
-            invalid_managers = df
+            for debtor in self.nmbrs.debtor_ids:
+                df = pd.concat([df, self._get(debtor)])
+
+        if df.empty:
+            return df, pd.DataFrame()
 
+        valid_managers, invalid_managers = BrynQFunctions.validate_data(df=df, schema=ManagerBasicGet, debug=self.nmbrs.debug)
         return valid_managers, invalid_managers
 
     def create(self, debtor_id: str, data: Dict[str, Any]):

brynq_sdk_nmbrs/salaries.py

@@ -1,13 +1,11 @@
-from typing import Any, Dict, Union
+from typing import Any, Dict, Optional
 
 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, SalaryUpdate
+from .schemas.salary import SalaryCreate, SalaryGet
 
 
 class Salaries:
@@ -88,73 +86,58 @@ class Salaries:
         )
         return resp
 
-    def get_current(self, employee_id: Union[int, str]) -> pd.DataFrame:
+    def get_cost_per_hour(self,
+                          company_id: Optional[str] = None,
+                          employee_id: Optional[str] = None,
+                          year: Optional[int] = None,
+                          period: Optional[int] = None) -> pd.DataFrame:
         """
-        Get current salary for an employee via SOAP.
+        Get the estimated cost per hour of employees.
 
         Args:
-            employee_id: The ID of the employee
+            company_id: Optional company ID. If not provided, fetches for all companies.
+            employee_id: Optional employee ID filter
+            year: Optional year filter
+            period: Optional period filter
 
         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]):
+            DataFrame containing cost per hour data
         """
-        Update salary for an employee using SOAP API.
+        if company_id:
+            return self._get_cost_per_hour(company_id, employee_id, year, period)
 
-        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)
+        df = pd.DataFrame()
+        for company in self.nmbrs.company_ids:
+            df = pd.concat([df, self._get_cost_per_hour(company, employee_id, year, period)])
 
-        Returns:
-            Response from the API
+        return df
+
+    def _get_cost_per_hour(self,
+                           company_id: str,
+                           employee_id: Optional[str] = None,
+                           year: Optional[int] = None,
+                           period: Optional[int] = None) -> pd.DataFrame:
+        """
+        Internal method to get cost per hour for a single company.
         """
-        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)}")
+        params = {}
+        if employee_id is not None:
+            params['employeeId'] = employee_id
+        if year is not None:
+            params['year'] = year
+        if period is not None:
+            params['period'] = period
+
+        request = requests.Request(
+            method='GET',
+            url=f"{self.nmbrs.base_url}companies/{company_id}/employees/costperhour",
+            params=params
+        )
+
+        data = self.nmbrs.get_paginated_result(request)
+        df = pd.DataFrame(data)
+
+        if not df.empty:
+            df['companyId'] = company_id
+
+        return df

brynq_sdk_nmbrs/salary_tables.py

@@ -1,10 +1,7 @@
-from typing import Dict, List, Union, Tuple
+from typing import Dict, Tuple
 import pandas as pd
 from zeep.exceptions import Fault
-from zeep.ns import WSDL, SOAP_ENV_11
-from zeep.xsd import ComplexType, Element, String
 from zeep.helpers import serialize_object
-# import logging
 from brynq_sdk_functions import Functions
 from .schemas.salary import SalaryTableGet, SalaryScalesGet, SalaryStepsGet
 

brynq_sdk_nmbrs/schedules.py

@@ -1,11 +1,11 @@
+from typing import Any, Dict, Tuple
+
 import pandas as pd
 import requests
+
 from brynq_sdk_functions import Functions
-from .schemas.schedules import ScheduleGet, ScheduleCreate, ScheduleUpdate
-from datetime import datetime
-from typing import Dict, Any, Tuple, Union
-from zeep.exceptions import Fault
-from zeep.helpers import serialize_object
+
+from .schemas.schedules import ScheduleCreate, ScheduleGet
 
 
 class Schedule:
@@ -83,76 +83,3 @@ 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/absence.py

@@ -22,28 +22,38 @@ from datetime import datetime
 class AbsenceGet(BrynQPanderaDataFrameModel):
     employee_id: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Employee ID", alias="EmployeeId")
     absence_id: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Absence ID", alias="AbsenceId")
-    comment: Series[String] = pa.Field(coerce=True, description="Comment", alias="Comment")
+    comment: Series[String] = pa.Field(coerce=True, nullable=True, description="Comment", alias="Comment")
     percentage: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Percentage", alias="Percentage")
     start: Series[DateTime] = pa.Field(coerce=True, description="Start", alias="Start")
     registration_start_date: Series[DateTime] = pa.Field(coerce=True, description="Registration Start Date", alias="RegistrationStartDate")
-    end: Series[DateTime] = pa.Field(coerce=True, description="End", alias="End")
-    registration_end_date: Series[DateTime] = pa.Field(coerce=True, description="Registration End Date", alias="RegistrationEndDate")
-    dossier: Series[String] = pa.Field(coerce=True, description="Dossier", alias="Dossier")
-    dossiernr: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Dossier Number", alias="Dossiernr")
+    end: Series[DateTime] = pa.Field(coerce=True, nullable=True, description="End", alias="End")
+    registration_end_date: Series[DateTime] = pa.Field(coerce=True, nullable=True, description="Registration End Date", alias="RegistrationEndDate")
+    dossier: Series[String] = pa.Field(coerce=True, nullable=True, description="Dossier", alias="Dossier")
+    dossiernr: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Dossier Number", alias="Dossiernr")
+
+    class _Annotation:
+        primary_key = "absence_id"
+        foreign_keys = {
+            "employee_id": {
+                "parent_schema": "EmployeeSchema",
+                "parent_column": "employee_id",
+                "cardinality": "N:1"
+            }
+        }
 
 
 class AbsenceCreate(BaseModel):
-    employee_id: Optional[int] = Field(None, example="1234567890", description="Employee ID", alias="EmployeeId")
-    absence_id: int = Field(coerce=True, description="Absence ID", alias="AbsenceId")
-    comment: str = Field(coerce=True, description="Comment", alias="Comment")
-    percentage: int = Field(coerce=True, description="Percentage", alias="Percentage")
-    start: datetime = Field(coerce=True, description="Start", alias="Start")
-    registration_start_date: datetime = Field(coerce=True, description="Registration Start Date", alias="RegistrationStartDate")
-    end: datetime = Field(coerce=True, description="End", alias="End")
-    registration_end_date: datetime = Field(coerce=True, description="Registration End Date", alias="RegistrationEndDate")
-    dossier: str = Field(coerce=True, description="Dossier", alias="Dossier")
-    dossiernr: int = Field(coerce=True, description="Dossier Number", alias="Dossiernr")
-    new_dossier: bool = pa.Field(coerce=True, description="New Dossier", alias="NewDossier")
+    employee_id: Optional[int] = Field(None, example=1234567890, description="Employee ID", alias="EmployeeId")
+    absence_id: int = Field(..., example=1, description="Absence ID", alias="AbsenceId")
+    comment: Optional[str] = Field(None, example="Sick leave", description="Comment", alias="Comment")
+    percentage: int = Field(..., ge=0, le=100, example=100, description="Percentage", alias="Percentage")
+    start: datetime = Field(..., example="2021-01-01T00:00:00Z", description="Start", alias="Start")
+    registration_start_date: datetime = Field(..., example="2021-01-01T00:00:00Z", description="Registration Start Date", alias="RegistrationStartDate")
+    end: Optional[datetime] = Field(None, example="2021-01-10T00:00:00Z", description="End", alias="End")
+    registration_end_date: Optional[datetime] = Field(None, example="2021-01-10T00:00:00Z", description="Registration End Date", alias="RegistrationEndDate")
+    dossier: Optional[str] = Field(None, example="Sick leave dossier", description="Dossier", alias="Dossier")
+    dossiernr: Optional[int] = Field(None, example=1, description="Dossier Number", alias="Dossiernr")
+    new_dossier: bool = Field(..., example=True, description="New Dossier", alias="NewDossier")
 
 
     def to_soap_settings(self, soap_client):

brynq_sdk_nmbrs/schemas/address.py

@@ -2,9 +2,8 @@ from typing import Annotated, Optional
 
 import pandas as pd
 import pandera as pa
-import pandera.extensions as extensions
 from pandera import Bool
-from pandera.typing import DateTime, Float, Series, String
+from pandera.typing import Series, String
 from pydantic import BaseModel, Field, StringConstraints
 
 from brynq_sdk_functions import BrynQPanderaDataFrameModel
@@ -75,30 +74,4 @@ class AddressCreate(BaseModel):
             max_length=3
         )
     ] = 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
+    period: Period = Field(..., example={"year": 2021, "period": 4}, description="Period details")

brynq_sdk_nmbrs/schemas/children.py

@@ -1,6 +1,4 @@
-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

brynq_sdk_nmbrs/schemas/contracts.py

@@ -1,12 +1,13 @@
-import pandas as pd
+from datetime import datetime
+from typing import Optional
+
 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
+from pandera.typing import DateTime, Float, Series, String
 from pydantic import BaseModel, Field, field_validator
-from datetime import datetime
+
+from brynq_sdk_functions import BrynQPanderaDataFrameModel
+
 
 # ---------------------------
 # Get Schemas

brynq_sdk_nmbrs/schemas/costcenter.py

@@ -98,8 +98,6 @@ class CostcenterTable(BaseModel):
     default: Optional[bool] = Field(True, example=True, description="Default", alias="default")
 
 
-
-
 class EmployeeCostcenterUpdate(BaseModel):
     employee_id: str = Field(..., example="c605f980-1c4c-42c1-8ddb-2d90c58da0b3", description="Employee Cost Center ID", alias="employeeId")
     employee_cost_centers: List[CostcenterTable] = Field(..., description="Employee Cost Centers", alias="employeeCostCenters")
@@ -115,10 +113,12 @@ class EmployeeCostcenterDelete(BaseModel):
 # CostCenter CRUD schemas - These are hypothetical since the API doesn't have create/update/delete endpoints
 # but we add them for consistency with other schema files
 class CostcenterCreate(BaseModel):
+    company_id: str = Field(..., description="Company identifier", alias="companyId")
     code: str = Field(..., example="CC001", description="Code", alias="code")
     description: str = Field(..., example="Sales Department", description="Description", alias="description")
 
 class CostcenterUpdate(BaseModel):
+    company_id: str = Field(..., description="Company identifier", alias="companyId")
     cost_center_id: str = Field(..., example="a405f980-1c4c-42c1-8ddb-2d90c58da0b1", description="Cost Center ID", alias="costCenterId")
     code: str = Field(..., example="CC001", description="Code", alias="code")
     description: str = Field(..., example="Sales Department", description="Description", alias="description")

brynq_sdk_nmbrs/schemas/costunit.py

@@ -23,10 +23,12 @@ class CostunitGet(BrynQPanderaDataFrameModel):
 # Upload Schemas
 # ---------------------------
 class CostunitCreate(BaseModel):
+    company_id: str = Field(..., description="Company identifier", alias="companyId")
     code: str = Field(..., example="CU001", description="Code", alias="code")
     description: str = Field(..., example="Marketing Unit", description="Description", alias="description")
 
 class CostunitUpdate(BaseModel):
+    company_id: str = Field(..., description="Company identifier", alias="companyId")
     cost_unit_id: str = Field(..., example="b505f980-1c4c-42c1-8ddb-2d90c58da0b2", description="Cost Unit ID", alias="costUnitId")
     code: str = Field(..., example="CU001", description="Code", alias="code")
     description: str = Field(..., example="Marketing Unit", description="Description", alias="description")

brynq_sdk_nmbrs/schemas/days.py

@@ -15,22 +15,22 @@ from pydantic import BaseModel, Field
 # ---------------------------
 # PANDERA SCHEMA
 class VariableDaysGet(BrynQPanderaDataFrameModel):
-    days: Series[String] = pa.Field(coerce=True, description="Hour Component ID", alias="days")
-    days_for_wage_components_per_day: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Hour Code", alias="daysForWageComponentsPerDay")
+    days: Series[String] = pa.Field(coerce=True, description="Days", alias="days")
+    days_for_wage_components_per_day: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Days for Wage Components per Day", alias="daysForWageComponentsPerDay")
     employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
 
-    class Config:
-        coerce = True
+    class _Annotation:
+        primary_key = "employee_id"
 
 class FixedDaysGet(BrynQPanderaDataFrameModel):
-    days: Series[String] = pa.Field(coerce=True, description="Hour Component ID", alias="days")
-    days_for_wage_components_per_day: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Hour Code", alias="daysForWageComponentsPerDay")
+    days: Series[String] = pa.Field(coerce=True, description="Days", alias="days")
+    days_for_wage_components_per_day: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True, description="Days for Wage Components per Day", alias="daysForWageComponentsPerDay")
     end_year: Series[pd.Int64Dtype] = pa.Field(nullable=True, coerce=True, description="End Year", alias="endYear")
     end_period: Series[pd.Int64Dtype] = pa.Field(nullable=True, coerce=True, description="End Period", alias="endPeriod")
     employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
 
-    class Config:
-        coerce = True
+    class _Annotation:
+        primary_key = "employee_id"
 
 # ---------------------------
 # Upload Schemas
@@ -70,7 +70,7 @@ class VariableHoursCreate(BaseModel):
     cost_center_id: Optional[str] = Field(None, example="aa506564-d1db-4fa8-83dc-d68db4cfcd82", description="Cost Center ID", alias="costCenterId")
     cost_unit_id: Optional[str] = Field(None, example="d8ac6afb-2ac6-43bf-9880-2d382cdace43", description="Cost Unit ID", alias="costUnitId")
     comment: Optional[str] = Field(None, example="Shift hours", description="Comment", alias="comment")
-    period_details: Optional[PeriodPost] = Field(None, example="Period details", description="Period details", alias="periodDetails")
+    period_details: Optional[PeriodPost] = Field(None, example={"period": {"year": 2025, "period": 1}, "unprotectedMode": False}, description="Period details", alias="periodDetails")
 
 class VariableHoursUpdate(BaseModel):
     hour_component_id: str = Field(..., example="49a69eda-252e-4ccb-a220-38ea90511d4f", description="Hour Component ID", alias="hourComponentId")
@@ -86,11 +86,13 @@ class HoursDelete(BaseModel):
 
 
 class FixedDaysCreate(BaseModel):
+    employee_id: str = Field(..., example="3054d4cf-b449-489d-8d2e-5dd30e5ab994", description="Employee identifier", alias="employeeId")
     number_of_days: int = Field(..., ge=0, le=1000, example=40, description="Days", alias="days")
     days_for_wage_components_per_day: Optional[int] = Field(None, ge=0, le=1000, example=40, description="Days for Wage Components per Day", alias="daysForWageComponentsPerDay")
-    period_details: Optional[PeriodPost] = Field(None, example="Period details", description="Period details", alias="periodDetails")
+    period_details: Optional[PeriodPost] = Field(None, example={"period": {"year": 2025, "period": 1}, "unprotectedMode": False}, description="Period details", alias="periodDetails")
 
 class VariableDaysCreate(BaseModel):
+    employee_id: str = Field(..., example="3054d4cf-b449-489d-8d2e-5dd30e5ab994", description="Employee identifier", alias="employeeId")
     number_of_days: int = Field(..., ge=0, le=1000, example=40, description="Days", alias="days")
     days_for_wage_components_per_day: Optional[int] = Field(None, ge=0, le=1000, example=40, description="Days for Wage Components per Day", alias="daysForWageComponentsPerDay")
-    period_details: Optional[PeriodPost] = Field(None, example="Period details", description="Period details", alias="periodDetails")
+    period_details: Optional[PeriodPost] = Field(None, example={"period": {"year": 2025, "period": 1}, "unprotectedMode": False}, description="Period details", alias="periodDetails")

brynq_sdk_nmbrs/schemas/debtor.py

@@ -1,11 +1,8 @@
-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
+
 from brynq_sdk_functions import BrynQPanderaDataFrameModel
-from typing import Optional, Annotated
-from pydantic import BaseModel, Field, StringConstraints
+
 
 # ---------------------------
 # Get Schemas
@@ -13,26 +10,7 @@ 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")
-
-
-# ---------------------------
-# 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")
+    name: Series[String] = pa.Field(coerce=True, description="Debtor name", alias="name")
 
-    class Config:
-        populate_by_name = True
+    class _Annotation:
+        primary_key = "debtor_id"