brynq-sdk-bob 2.8.5__tar.gz → 2.9.1__tar.gz

{brynq_sdk_bob-2.8.5 → brynq_sdk_bob-2.9.1}/PKG-INFO

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

{brynq_sdk_bob-2.8.5 → brynq_sdk_bob-2.9.1}/brynq_sdk_bob/__init__.py

@@ -21,8 +21,9 @@ class Bob(BrynQ):
     def __init__(self, system_type: Optional[Literal['source', 'target']] = None, test_environment: bool = True, debug: bool = False, target_system: str = None):
         super().__init__()
         self.timeout = 3600
+        self.test_environment = test_environment
         self.headers = self._get_request_headers(system_type)
-        if test_environment:
+        if self.test_environment:
             self.base_url = "https://api.sandbox.hibob.com/v1/"
         else:
             self.base_url = "https://api.hibob.com/v1/"
@@ -46,6 +47,15 @@ class Bob(BrynQ):
 
     def _get_request_headers(self, system_type):
         credentials = self.interfaces.credentials.get(system='bob', system_type=system_type)
+        # multiple creds possible, not fetched by environment test status, get first occurence
+        if isinstance(credentials, list):
+            credentials = next(
+                (
+                    element for element in credentials
+                    if element.get('data', {}).get('Test Environment') == self.test_environment
+                ),
+                credentials[0]
+            )
         auth_token = base64.b64encode(f"{credentials.get('data').get('User ID')}:{credentials.get('data').get('API Token')}".encode()).decode('utf-8')
         headers = {
             "accept": "application/json",

brynq_sdk_bob-2.9.1/brynq_sdk_bob/schemas/employment.py

@@ -0,0 +1,32 @@
+import pandas as pd
+import pandera as pa
+from typing import Optional
+from pandera import Bool
+from pandera.typing import Series, String, Float, DateTime
+from brynq_sdk_functions import BrynQPanderaDataFrameModel
+
+class EmploymentSchema(BrynQPanderaDataFrameModel):
+    id: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Employment ID", alias="id")
+    employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
+    active_effective_date: Series[DateTime] = pa.Field(coerce=True, description="Active Effective Date", alias="activeEffectiveDate")
+    contract: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Contract", alias="contract") # has a list of possible values
+    creation_date: Optional[Series[DateTime]] = pa.Field(coerce=True, nullable=True, description="Creation Date", alias="creationDate")
+    effective_date: Series[DateTime] = pa.Field(coerce=True, description="Effective Date", alias="effectiveDate")
+    end_effective_date: Optional[Series[DateTime]] = pa.Field(coerce=True, nullable=True, description="End Effective Date", alias="endEffectiveDate")
+    fte: Series[Float] = pa.Field(coerce=True, description="FTE", alias="fte")
+    is_current: Series[Bool] = pa.Field(coerce=True, description="Is Current", alias="isCurrent")
+    modification_date: Optional[Series[DateTime]] = pa.Field(coerce=True, nullable=True, description="Modification Date", alias="modificationDate")
+    salary_pay_type: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Salary Pay Type", alias="salaryPayType")
+    weekly_hours: Optional[Series[Float]] = pa.Field(coerce=True, nullable=True, description="Weekly Hours", alias="weeklyHours")
+    # weekly_hours_sort_factor: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=False)
+    actual_working_pattern_working_pattern_type: Optional[Series[pa.String]] = pa.Field(nullable=True, description="Actual Working Pattern Working Pattern Type", alias="actualWorkingPattern.workingPatternType")
+    actual_working_pattern_days_sunday: Optional[Series[Float]] = pa.Field(nullable=True, description="Actual Working Pattern Days Sunday", alias="actualWorkingPattern.days.sunday")
+    actual_working_pattern_days_tuesday: Optional[Series[Float]] = pa.Field(nullable=True, description="Actual Working Pattern Days Tuesday", alias="actualWorkingPattern.days.tuesday")
+    actual_working_pattern_days_wednesday: Optional[Series[Float]] = pa.Field(nullable=True, description="Actual Working Pattern Days Wednesday", alias="actualWorkingPattern.days.wednesday")
+    actual_working_pattern_days_monday: Optional[Series[Float]] = pa.Field(nullable=True, description="Actual Working Pattern Days Monday", alias="actualWorkingPattern.days.monday")
+    actual_working_pattern_days_friday: Optional[Series[Float]] = pa.Field(nullable=True, description="Actual Working Pattern Days Friday", alias="actualWorkingPattern.days.friday")
+    actual_working_pattern_days_thursday: Optional[Series[Float]] = pa.Field(nullable=True, description="Actual Working Pattern Days Thursday", alias="actualWorkingPattern.days.thursday")
+    actual_working_pattern_days_saturday: Optional[Series[Float]] = pa.Field(nullable=True, description="Actual Working Pattern Days Saturday", alias="actualWorkingPattern.days.saturday")
+
+    class Config:
+        coerce = True

{brynq_sdk_bob-2.8.5 → brynq_sdk_bob-2.9.1}/brynq_sdk_bob.egg-info/PKG-INFO

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

{brynq_sdk_bob-2.8.5 → brynq_sdk_bob-2.9.1}/setup.py

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

brynq_sdk_bob-2.8.5/brynq_sdk_bob/schemas/employment.py

@@ -1,31 +0,0 @@
-import pandas as pd
-import pandera as pa
-from pandera import Bool
-from pandera.typing import Series, String, Float, DateTime
-from brynq_sdk_functions import BrynQPanderaDataFrameModel
-
-class EmploymentSchema(BrynQPanderaDataFrameModel):
-    id: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Employment ID", alias="id")
-    employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
-    active_effective_date: Series[DateTime] = pa.Field(coerce=True, description="Active Effective Date", alias="activeEffectiveDate")
-    contract: Series[String] = pa.Field(coerce=True, nullable=True, description="Contract", alias="contract") # has a list of possible values
-    creation_date: Series[DateTime] = pa.Field(coerce=True, nullable=True, description="Creation Date", alias="creationDate")
-    effective_date: Series[DateTime] = pa.Field(coerce=True, description="Effective Date", alias="effectiveDate")
-    end_effective_date: Series[DateTime] = pa.Field(coerce=True, nullable=True, description="End Effective Date", alias="endEffectiveDate")
-    fte: Series[Float] = pa.Field(coerce=True, description="FTE", alias="fte")
-    is_current: Series[Bool] = pa.Field(coerce=True, description="Is Current", alias="isCurrent")
-    modification_date: Series[DateTime] = pa.Field(coerce=True, nullable=True, description="Modification Date", alias="modificationDate")
-    salary_pay_type: Series[String] = pa.Field(coerce=True, nullable=True, description="Salary Pay Type", alias="salaryPayType")
-    weekly_hours: Series[Float] = pa.Field(coerce=True, nullable=True, description="Weekly Hours", alias="weeklyHours")
-    # weekly_hours_sort_factor: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=False)
-    actual_working_pattern_working_pattern_type: Series[pa.String] = pa.Field(nullable=True, description="Actual Working Pattern Working Pattern Type", alias="actualWorkingPattern.workingPatternType")
-    actual_working_pattern_days_sunday: Series[Float] = pa.Field(nullable=True, description="Actual Working Pattern Days Sunday", alias="actualWorkingPattern.days.sunday")
-    actual_working_pattern_days_tuesday: Series[Float] = pa.Field(nullable=True, description="Actual Working Pattern Days Tuesday", alias="actualWorkingPattern.days.tuesday")
-    actual_working_pattern_days_wednesday: Series[Float] = pa.Field(nullable=True, description="Actual Working Pattern Days Wednesday", alias="actualWorkingPattern.days.wednesday")
-    actual_working_pattern_days_monday: Series[Float] = pa.Field(nullable=True, description="Actual Working Pattern Days Monday", alias="actualWorkingPattern.days.monday")
-    actual_working_pattern_days_friday: Series[Float] = pa.Field(nullable=True, description="Actual Working Pattern Days Friday", alias="actualWorkingPattern.days.friday")
-    actual_working_pattern_days_thursday: Series[Float] = pa.Field(nullable=True, description="Actual Working Pattern Days Thursday", alias="actualWorkingPattern.days.thursday")
-    actual_working_pattern_days_saturday: Series[Float] = pa.Field(nullable=True, description="Actual Working Pattern Days Saturday", alias="actualWorkingPattern.days.saturday")
-
-    class Config:
-        coerce = True