brynq-sdk-nmbrs 2.2.2__tar.gz → 2.3.0__tar.gz

{brynq_sdk_nmbrs-2.2.2 → brynq_sdk_nmbrs-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq_sdk_nmbrs
-Version: 2.2.2
+Version: 2.3.0
 Summary: Nmbrs wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_nmbrs-2.2.2 → brynq_sdk_nmbrs-2.3.0}/brynq_sdk_nmbrs/__init__.py

@@ -139,6 +139,28 @@ class Nmbrs(BrynQ):
 
             return auth_header
 
+    def _get_soap_auth_header_employees(self):
+        """
+        Creates the SOAP authentication header using credentials from initial_credentials.
+
+        Returns:
+            AuthHeaderWithDomainType: The authentication header for SOAP requests
+        """
+        initial_credentials = self.get_system_credential(system='nmbrs', label='bob')
+        config = initial_credentials.get("config", {})
+
+        # Get the AuthHeaderWithDomain type from the WSDL
+        AuthHeaderWithDomainType = self.soap_client_employees.get_element('ns0:AuthHeaderWithDomain')
+
+        # Create the auth header using credentials from config
+        auth_header = AuthHeaderWithDomainType(
+            Username=config.get("soap_api_username"),
+            Token=config.get("soap_api_token"),
+            Domain=config.get("soap_api_domain")
+        )
+
+        return auth_header
+
     def get_paginated_result(self, request: requests.Request) -> List:
         has_next_page = True
         result_data = []

{brynq_sdk_nmbrs-2.2.2 → brynq_sdk_nmbrs-2.3.0}/brynq_sdk_nmbrs/employees.py

@@ -22,7 +22,7 @@ from .days import VariableDays
 from .wagecomponents import EmployeeVariableWageComponents, EmployeeFixedWageComponents
 from .leave import Leave, LeaveBalance
 from .schemas.employees import (
-    EmployeeGet, EmployeeCreate, EmployeeUpdate, EmployeeDelete,
+    BsnGet, EmployeeGet, EmployeeCreate, EmployeeUpdate, EmployeeDelete,
     BasicInfo, BirthInfo, ContactInfo, PartnerInfo, Period, AdditionalEmployeeInfo,
     CreateEmployeePersonalInfo
 )
@@ -83,6 +83,24 @@ class Employees:
 
         return df
 
+    def get_private_info(self) -> str:
+        combined = []
+        for company_id in self.nmbrs.company_ids:
+            request = requests.Request(method='GET',
+                                    url=f"{self.nmbrs.base_url}companies/{company_id}/employees/privateInfos")
+            data = self.nmbrs.get_paginated_result(request)
+            df = pd.json_normalize(
+                data,
+                record_path='privateInfos',
+                meta=['employeeId']
+            )
+            df['companyId'] = company_id
+            combined.append(df)
+        df = pd.concat(combined, ignore_index=True)
+        valid_bsn, invalid_bsn = Functions.validate_data(df, BsnGet)
+
+        return valid_bsn, invalid_bsn
+
     def get_default_templates(self) -> pd.DataFrame:
         default_templates = pd.DataFrame()
         for company in self.nmbrs.company_ids:

{brynq_sdk_nmbrs-2.2.2 → brynq_sdk_nmbrs-2.3.0}/brynq_sdk_nmbrs/schemas/employees.py

@@ -84,7 +84,7 @@ class BirthInfo(BaseModel):
         StringConstraints(
             pattern=r'^[A-Za-z]+$',
             strip_whitespace=True,
-            min_length=2, 
+            min_length=2,
             max_length=3
         )
     ]] = Field(None, example="PT", description="Nationality Code ISO", alias="nationalityCodeISO")
@@ -140,3 +140,12 @@ class EmployeeUpdate(BaseModel):
 
 class EmployeeDelete(BaseModel):
     employee_id: str = Field(..., example="3054d4cf-b449-489d-8d2e-5dd30e5ab994", description="Employee ID", alias="employeeId")
+
+class BsnGet(BrynQPanderaDataFrameModel):
+    employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
+    social_security_number: Series[String] = pa.Field(coerce=True, description="Social Security Number", alias="BSN")
+    company_id: Series[String] = pa.Field(coerce=True, description="Company ID", alias="companyId")
+    created_at: Series[DateTime] = pa.Field(coerce=True, description="Created At", alias="createdAt")
+
+    class _Annotation:
+        primary_key = "employee_id"

{brynq_sdk_nmbrs-2.2.2 → brynq_sdk_nmbrs-2.3.0}/brynq_sdk_nmbrs/schemas/wage_tax.py

@@ -10,7 +10,7 @@ from pydantic import BaseModel, Field, StringConstraints
 from datetime import datetime
 
 
-class WageTaxGet(BrynQPanderaDataFrameModel):
+class WageTaxSettingsGet(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")
@@ -24,6 +24,47 @@ class WageTaxGet(BrynQPanderaDataFrameModel):
     correction_period_start: Series[DateTime] = pa.Field(nullable=True, coerce=True, description="Correction Tijdvak Start", alias="CorrectionTijdvakStart")
     correction_period_end: Series[DateTime] = pa.Field(nullable=True, coerce=True, description="Correction Tijdvak End", alias="CorrectionTijdvakEnd")
 
+# <soap:Envelope xmlns:soap="http://www.w3.org/2003/05/soap-envelope" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:xsd="http://www.w3.org/2001/XMLSchema">
+#    <soap:Body>
+#       <WageTax_GetListResponse xmlns="https://api.nmbrs.nl/soap/v3/EmployeeService">
+#          <WageTax_GetListResult>
+#             <WageTaxSettings>
+#                <Id>2998190</Id>
+#                <JaarloonBT>12144.00</JaarloonBT>
+#                <AfwBijzTariefLH>0</AfwBijzTariefLH>
+#                <AutoKleineBanenRegeling>true</AutoKleineBanenRegeling>
+#                <Loonheffingkorting>true</Loonheffingkorting>
+#                <Voordeelreg>true</Voordeelreg>
+#                <Loonheffing>true</Loonheffing>
+#                <CodeAfdrachtvermindering>0</CodeAfdrachtvermindering>
+#                <KleurTabel>1</KleurTabel>
+#                <SoortInkomen>15</SoortInkomen>
+#                <SpecialeTabel>0</SpecialeTabel>
+#                <TijdvakTabel>2</TijdvakTabel>
+#                <VakantieBonnen>0</VakantieBonnen>
+#                <CodeCalc30PercRule>0</CodeCalc30PercRule>
+#             </WageTaxSettings>
+#          </WageTax_GetListResult>
+#       </WageTax_GetListResponse>
+#    </soap:Body>
+# </soap:Envelope>
+
+class WageTaxGet(BrynQPanderaDataFrameModel):
+    wage_tax_id: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Wage Tax ID", alias="Id")
+    employee_id: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Employee ID", alias="EmployeeId")
+    yearly_salary: Series[pd.Float64Dtype] = pa.Field(coerce=True, description="Yearly Salary", alias="JaarloonBT")
+    deviation_special_rate_payroll_tax_deduction: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Afw Bijz Tarief LH", alias="AfwBijzTariefLH")
+    auto_small_jobs: Series[Bool] = pa.Field(coerce=True, description="Auto Kleine Banen Regeling", alias="AutoKleineBanenRegeling")
+    payroll_tax_deduction: Series[Bool] = pa.Field(coerce=True, description="Loonheffingkorting", alias="Loonheffingkorting")
+    benefit_scheme: Series[Bool] = pa.Field(coerce=True, description="Voordeelreg", alias="Voordeelreg")
+    payroll_tax: Series[Bool] = pa.Field(coerce=True, description="Loonheffing", alias="Loonheffing")
+    code_tax_reduction: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Code Afdrachtvermindering", alias="CodeAfdrachtvermindering")
+    color_table: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Kleur Tabel", alias="KleurTabel")
+    type_of_income: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Soort Inkomen", alias="SoortInkomen")
+    special_table: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Speciale Tabel", alias="SpecialeTabel")
+    period_table: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Tijdvak Tabel", alias="TijdvakTabel")
+    holiday_vouchers: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Vakantie Bonnen", alias="VakantieBonnen")
+    code_calculate_30_percent_rule: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Code Calc 30% Rule", alias="CodeCalc30PercRule")
 
 class WageTaxUpdate(BaseModel):
     employee_id: Optional[int] = Field(None, example="1234567890", description="Employee ID", alias="EmployeeId")

{brynq_sdk_nmbrs-2.2.2 → brynq_sdk_nmbrs-2.3.0}/brynq_sdk_nmbrs/wage_tax.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, List, Union, Tuple
 import pandas as pd
-from .schemas.wage_tax import WageTaxGet, WageTaxUpdate
+from .schemas.wage_tax import WageTaxGet, WageTaxUpdate, WageTaxSettingsGet
 from zeep.exceptions import Fault
 from zeep.ns import WSDL, SOAP_ENV_11
 from zeep.xsd import ComplexType, Element, String
@@ -33,7 +33,7 @@ class WageTax:
                 wagetax_settings_temp['companyId'] = company['number']
                 wagetax_settings = pd.concat([wagetax_settings, wagetax_settings_temp])
 
-        valid_wagetax_settings, invalid_wagetax_settings = Functions.validate_data(df=wagetax_settings, schema=WageTaxGet, debug=True)
+        valid_wagetax_settings, invalid_wagetax_settings = Functions.validate_data(df=wagetax_settings, schema=WageTaxSettingsGet, debug=True)
 
         # No validation schema for now, but could be added later
         return valid_wagetax_settings, invalid_wagetax_settings
@@ -79,6 +79,49 @@ class WageTax:
         except Exception as e:
             raise Exception(f"Failed to get salary tables: {str(e)}")
 
+    def get(self, employee_id: int) -> Tuple[pd.DataFrame, pd.DataFrame]:
+        """
+        Get wage tax settings for a specific employee.
+
+        Args:
+            employee_id (int): The ID of the employee
+
+        Returns:
+            pd.DataFrame: DataFrame containing the wage tax settings
+        """
+        try:
+            # Get the auth header using the centralized method
+            auth_header = self.nmbrs._get_soap_auth_header_employees()
+
+            # Make SOAP request with the proper header structure
+            response = self.soap_client_employees.service.WageTax_GetCurrent(
+                EmployeeId=employee_id,
+                _soapheaders=[auth_header]
+            )
+
+            # Convert response to DataFrame
+            if response:
+                # Convert Zeep objects to Python dictionaries
+                serialized_response = serialize_object(response)
+
+                # Convert to list if it's not already
+                if not isinstance(serialized_response, list):
+                    serialized_response = [serialized_response]
+
+                # Convert to DataFrame
+                df = pd.DataFrame(serialized_response)
+                df['EmployeeId'] = employee_id
+                valid_wage_tax, invalid_wage_tax = Functions.validate_data(df=df, schema=WageTaxGet, debug=True)
+
+                return valid_wage_tax, invalid_wage_tax
+            else:
+                return pd.DataFrame(), pd.DataFrame()
+
+        except Fault as e:
+            raise Exception(f"SOAP request failed: {str(e)}")
+        except Exception as e:
+            raise Exception(f"Failed to get wage tax settings: {str(e)}")
+
     def update(self, data: Dict[str, Any]) -> pd.DataFrame:
         try:
             wage_tax_model = WageTaxUpdate(**data)
@@ -86,6 +129,9 @@ class WageTax:
             if self.nmbrs.mock_mode:
                 return wage_tax_model
 
+            # Get the auth header using the centralized method
+            auth_header = self.nmbrs._get_soap_auth_header_employees()
+
             # Use the model's built-in SOAP conversion method
             wage_tax_settings = wage_tax_model.to_soap_settings(self.nmbrs.soap_client_employees)
 

{brynq_sdk_nmbrs-2.2.2 → brynq_sdk_nmbrs-2.3.0}/brynq_sdk_nmbrs.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq-sdk-nmbrs
-Version: 2.2.2
+Version: 2.3.0
 Summary: Nmbrs wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_nmbrs-2.2.2 → brynq_sdk_nmbrs-2.3.0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 
 setup(
     name='brynq_sdk_nmbrs',
-    version='2.2.2',
+    version='2.3.0',
     description='Nmbrs wrapper from BrynQ',
     long_description='Nmbrs wrapper from BrynQ',
     author='BrynQ',