brynq-sdk-bob 2.9.4__tar.gz → 2.9.6__tar.gz

{brynq_sdk_bob-2.9.4 → brynq_sdk_bob-2.9.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: brynq_sdk_bob
-Version: 2.9.4
+Version: 2.9.6
 Summary: Bob wrapper from BrynQ
 Author: BrynQ
 Author-email: support@brynq.com

{brynq_sdk_bob-2.9.4 → brynq_sdk_bob-2.9.6}/brynq_sdk_bob/__init__.py

@@ -1,21 +1,27 @@
 import base64
-from typing import List, Optional, Literal
-import requests
 import os
-from .reports import Reports
+from typing import List, Literal, Optional
+
+import requests
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
+
 from brynq_sdk_brynq import BrynQ
+
 from .bank import Bank
 from .company import Company
+from .custom_tables import CustomTables
 from .documents import CustomDocuments
 from .employment import Employment
 from .named_lists import NamedLists
 from .payments import Payments
+from .payroll_history import History
 from .people import People
+from .reports import Reports
 from .salaries import Salaries
 from .timeoff import TimeOff
 from .work import Work
-from .custom_tables import CustomTables
-from .payroll_history import History
+
 
 class Bob(BrynQ):
     def __init__(self, system_type: Optional[Literal['source', 'target']] = None, test_environment: bool = True, debug: bool = False, target_system: str = None):
@@ -28,6 +34,15 @@ class Bob(BrynQ):
         else:
             self.base_url = "https://api.hibob.com/v1/"
         self.session = requests.Session()
+        retry_strategy = Retry(
+            total=5,
+            backoff_factor=1.0,
+            status_forcelist=[429, 500, 502, 503, 504],
+            respect_retry_after_header=True
+        )
+        adapter = HTTPAdapter(max_retries=retry_strategy)
+        self.session.mount("http://", adapter)
+        self.session.mount("https://", adapter)
         self.session.headers.update(self.headers)
         self.people = People(self)
         self.salaries = Salaries(self)

{brynq_sdk_bob-2.9.4 → brynq_sdk_bob-2.9.6}/brynq_sdk_bob/custom_tables.py

@@ -1,6 +1,10 @@
+from typing import List, Optional
+
 import pandas as pd
-from brynq_sdk_functions import Functions
-from .schemas.custom_tables import CustomTableSchema, CustomTableMetadataSchema
+
+from brynq_sdk_functions import BrynQPanderaDataFrameModel, Functions
+
+from .schemas.custom_tables import CustomTableMetadataSchema, CustomTableSchema
 
 
 class CustomTables:
@@ -8,29 +12,37 @@ class CustomTables:
         self.bob = bob
         self.schema = CustomTableSchema
 
-    def get(self, employee_id: str, custom_table_id: str) -> tuple[pd.DataFrame, pd.DataFrame]:
+    def get(self, employee_ids: List[str], custom_table_id: str, schema_custom_fields: Optional[BrynQPanderaDataFrameModel] = None) -> tuple[pd.DataFrame, pd.DataFrame]:
         """
-        Get custom table data for an employee
+        Get custom table data for employees
 
         Args:
-            employee_id: The employee ID
+            employee_ids: List of employee IDs
             custom_table_id: The custom table ID
+            schema_custom_fields: Optional custom schema for validation
 
         Returns:
             A tuple of (valid_data, invalid_data) as pandas DataFrames
         """
-        resp = self.bob.session.get(url=f"{self.bob.base_url}people/custom-tables/{employee_id}/{custom_table_id}")
-        resp.raise_for_status()
-        data = resp.json()
+        df = pd.DataFrame()
+        for employee_id in employee_ids:
+            resp = self.bob.session.get(url=f"{self.bob.base_url}people/custom-tables/{employee_id}/{custom_table_id}", timeout=self.bob.timeout)
+            resp.raise_for_status()
+            data = resp.json()
 
-        # Normalize the nested JSON response
-        df = pd.json_normalize(
-            data,
-            record_path=['values']
-        )
+            # Normalize the nested JSON response
+            df_temp = pd.json_normalize(
+                data,
+                record_path=['values']
+            )
+            df_temp['employee_id'] = employee_id
+            df = pd.concat([df, df_temp])
+        df = df.reset_index(drop=True)
 
-        df['employee_id'] = employee_id
-        valid_data, invalid_data = Functions.validate_data(df=df, schema=self.schema, debug=True)
+        if schema_custom_fields is not None:
+            valid_data, invalid_data = Functions.validate_data(df=df, schema=schema_custom_fields, debug=True)
+        else:
+            valid_data, invalid_data = Functions.validate_data(df=df, schema=self.schema, debug=True)
 
         return valid_data, invalid_data
 

{brynq_sdk_bob-2.9.4 → brynq_sdk_bob-2.9.6}/brynq_sdk_bob/people.py

@@ -1,14 +1,16 @@
-import pandas as pd
 import re
 from typing import Optional
-from brynq_sdk_functions import Functions
-from brynq_sdk_functions import BrynQPanderaDataFrameModel
+
+import pandas as pd
+
+from brynq_sdk_functions import BrynQPanderaDataFrameModel, Functions
+
 from .bank import Bank
+from .custom_tables import CustomTables
 from .employment import Employment
 from .salaries import Salaries
 from .schemas.people import PeopleSchema
 from .work import Work
-from .custom_tables import CustomTables
 
 
 class People:
@@ -87,8 +89,12 @@ class People:
 
 
         if schema_custom_fields is not None:
+            # Make all custom schema columns nullable to avoid validation errors for missing values
+            custom_schema = schema_custom_fields.to_schema()
+            for col_name in custom_schema.columns:
+                custom_schema.columns[col_name].nullable = True
 
-            valid_people, invalid_people_custom = Functions.validate_data(df=df, schema=schema_custom_fields, debug=True)
+            valid_people, invalid_people_custom = Functions.validate_data(df=df, schema=custom_schema, debug=True)
 
 
         else:

{brynq_sdk_bob-2.9.4 → brynq_sdk_bob-2.9.6}/brynq_sdk_bob/schemas/payments.py

@@ -1,9 +1,12 @@
-import pandera as pa
-from pandera.typing import Series, String, Float, DateTime
 from typing import Optional
+
 import pandas as pd
+import pandera as pa
+from pandera.typing import DateTime, Float, Series, String
+
 from brynq_sdk_functions import BrynQPanderaDataFrameModel
 
+
 class VariablePaymentSchema(BrynQPanderaDataFrameModel):
     can_be_deleted: Series[bool] = pa.Field(nullable=True, coerce=True, description="Can Be Deleted", alias="canBeDeleted")
     department_percent: Series[Float] = pa.Field(nullable=True, coerce=True, description="Department Percent", alias="departmentPercent")

{brynq_sdk_bob-2.9.4 → brynq_sdk_bob-2.9.6}/brynq_sdk_bob/schemas/people.py

@@ -1,13 +1,14 @@
 from datetime import datetime
-from typing import Optional, List, Dict
+from typing import Dict, List, Optional
 
 import pandas as pd
 import pandera as pa
-from pandera import Bool
-from pandera.typing import Series, String, Float
 import pandera.extensions as extensions
-from brynq_sdk_functions import BrynQPanderaDataFrameModel
+from pandera import Bool
 from pandera.engines.pandas_engine import DateTime
+from pandera.typing import Float, Series, String
+
+from brynq_sdk_functions import BrynQPanderaDataFrameModel
 
 
 @extensions.register_check_method()
@@ -194,15 +195,16 @@ class PeopleSchema(BrynQPanderaDataFrameModel):
     payroll_employment_hours_in_day_not_worked: Optional[Series[Float]] = pa.Field(coerce=True, nullable=True, description="Hours in Day Not Worked", alias="payroll.employment.hoursInDayNotWorked")
     payroll_employment_site_working_pattern: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Site Working Pattern", alias="payroll.employment.siteWorkingPattern")
     payroll_employment_actual_working_pattern: Optional[Series[dict]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern", alias="payroll.employment.actualWorkingPattern")
-    payroll_employment_actual_working_pattern_days_sunday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Sunday", alias="payroll.employment.actualWorkingPattern.days.sunday")
+    payroll_employment_actual_working_pattern_days_monday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Monday", alias="payroll.employment.actualWorkingPattern.days.monday")
     payroll_employment_actual_working_pattern_days_tuesday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Tuesday", alias="payroll.employment.actualWorkingPattern.days.tuesday")
     payroll_employment_actual_working_pattern_days_wednesday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Wednesday", alias="payroll.employment.actualWorkingPattern.days.wednesday")
-    payroll_employment_actual_working_pattern_days_monday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Monday", alias="payroll.employment.actualWorkingPattern.days.monday")
-    payroll_employment_actual_working_pattern_days_friday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Friday", alias="payroll.employment.actualWorkingPattern.days.friday")
     payroll_employment_actual_working_pattern_days_thursday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Thursday", alias="payroll.employment.actualWorkingPattern.days.thursday")
+    payroll_employment_actual_working_pattern_days_friday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Friday", alias="payroll.employment.actualWorkingPattern.days.friday")
     payroll_employment_actual_working_pattern_days_saturday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Saturday", alias="payroll.employment.actualWorkingPattern.days.saturday")
+    payroll_employment_actual_working_pattern_days_sunday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Sunday", alias="payroll.employment.actualWorkingPattern.days.sunday")
     payroll_employment_actual_working_pattern_hours_per_day: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Hours Per Day", alias="payroll.employment.actualWorkingPattern.hoursPerDay")
     payroll_employment_actual_working_pattern_working_pattern_id: Optional[Series[pd.Int64Dtype]] = pa.Field(coerce=True, nullable=True, description="Actual Working Pattern - Working Pattern ID", alias="payroll.employment.actualWorkingPattern.workingPatternId")
+    payroll_employment_actual_working_pattern_weekly_work_percentage: Optional[Series[pd.Int64Dtype]] = pa.Field(coerce=True, nullable=True, description="Actual Working Patttern - working pattern weekly work percentage", alias="payroll.employment.actualWorkingPattern.weeklyWorkPercentage")
     payroll_employment_site_working_pattern_working_pattern_type: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Site Working Pattern - Working Pattern Type", alias="payroll.employment.siteWorkingPattern.workingPatternType")
     payroll_employment_site_working_pattern_days_sunday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Site Working Pattern - Sunday", alias="payroll.employment.siteWorkingPattern.days.sunday")
     payroll_employment_site_working_pattern_days_tuesday: Optional[Series[float]] = pa.Field(coerce=True, nullable=True, description="Site Working Pattern - Tuesday", alias="payroll.employment.siteWorkingPattern.days.tuesday")

{brynq_sdk_bob-2.9.4 → brynq_sdk_bob-2.9.6}/brynq_sdk_bob/schemas/salary.py

@@ -1,12 +1,14 @@
+from datetime import date, datetime
+from typing import Any, Dict, 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 pandera import Bool
+from pandera.typing import DateTime, Float, Series, String
 from pydantic import BaseModel, Field
-from datetime import datetime, date
-from typing import Optional, Dict, Any
+
+from brynq_sdk_functions import BrynQPanderaDataFrameModel
 
 
 class SalarySchema(BrynQPanderaDataFrameModel):
@@ -25,6 +27,9 @@ class SalarySchema(BrynQPanderaDataFrameModel):
 
 
 class SalaryCreateSchema(BaseModel):
+    # Function parameters
+    salary_data: Optional[Dict[str, Any]] = Field(None, description="Salary data", alias="salaryData")
+
     can_be_deleted: Optional[bool] = Field(None, description="internal flag", alias="canBeDeleted")
     work_change_type: Optional[str] = Field(None, description="The type of the change that was performed for this work entry. This will contain the ID of the value from the Change Type list.", alias="workChangeType")
     salary_change_reason: Optional[str] = Field(None, description="Reason for the change", alias="change.reason")

{brynq_sdk_bob-2.9.4 → brynq_sdk_bob-2.9.6}/brynq_sdk_bob.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: brynq_sdk_bob
-Version: 2.9.4
+Version: 2.9.6
 Summary: Bob wrapper from BrynQ
 Author: BrynQ
 Author-email: support@brynq.com

{brynq_sdk_bob-2.9.4 → brynq_sdk_bob-2.9.6}/setup.py

@@ -2,7 +2,7 @@ from setuptools import find_namespace_packages, setup
 
 setup(
     name='brynq_sdk_bob',
-    version='2.9.4',
+    version='2.9.6',
     description='Bob wrapper from BrynQ',
     long_description='Bob wrapper from BrynQ',
     author='BrynQ',