brynq-sdk-bob 1.0.0__tar.gz → 1.1.1__tar.gz

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq_sdk_bob
-Version: 1.0.0
+Version: 1.1.1
 Summary: Bob wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob/__init__.py

@@ -8,15 +8,15 @@ from .bank import Bank
 from .company import Company
 from .documents import CustomDocuments
 from .employment import Employment
+from .named_lists import NamedLists
 from .payments import Payments
 from .people import People
 from .salaries import Salaries
 from .timeoff import TimeOff
 from .work import Work
 
-
 class Bob(BrynQ):
-    def __init__(self, label: Union[str, List], test_environment: bool = True, debug: bool = False):
+    def __init__(self, label: Union[str, List], test_environment: bool = True, debug: bool = False, target_system: str = None):
         super().__init__()
         self.headers = self._get_request_headers(label=label)
         if test_environment:
@@ -34,6 +34,8 @@ class Bob(BrynQ):
         self.time_off = TimeOff(self)
         self.documents = CustomDocuments(self)
         self.companies = Company(self)
+        self.named_lists = NamedLists(self)
+        self.timeout = 3600
 
     def _get_request_headers(self, label):
         credentials = self.get_system_credential(system='bob', label=label)
@@ -52,7 +54,7 @@ class Bob(BrynQ):
         while has_next_page:
             prepped = request.prepare()
             prepped.headers.update(self.session.headers)
-            resp = self.session.send(prepped)
+            resp = self.session.send(prepped, timeout=self.timeout)
             resp.raise_for_status()
             response_data = resp.json()
             result_data += response_data['results']

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob/bank.py

@@ -10,7 +10,7 @@ class Bank:
     def get(self, person_ids: pd.Series) -> (pd.DataFrame, pd.DataFrame):
         data = []
         for person_id in person_ids:
-            resp = self.bob.session.get(url=f"{self.bob.base_url}people/{person_id}/bank-accounts")
+            resp = self.bob.session.get(url=f"{self.bob.base_url}people/{person_id}/bank-accounts", timeout=self.bob.timeout)
             resp.raise_for_status()
             temp_data = resp.json()['values']
             # when an employee has one or more bank accounts, the response is a list of dictionaries.

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob/company.py

@@ -9,12 +9,12 @@ class Company:
         values = {}
 
         if list_name is not None:
-            resp = self.bob.session.get(url=f"{self.bob.base_url}company/named-lists/{list_name}")
+            resp = self.bob.session.get(url=f"{self.bob.base_url}company/named-lists/{list_name}", timeout=self.bob.timeout)
             resp.raise_for_status()
             data = resp.json()
             values.update({data["name"]: [value['id'] for value in data['values']]})
         else:
-            resp = self.bob.session.get(url=f"{self.bob.base_url}company/named-lists")
+            resp = self.bob.session.get(url=f"{self.bob.base_url}company/named-lists", timeout=self.bob.timeout)
             resp.raise_for_status()
             data = resp.json()
             for list_key, list_data in data.items():

brynq_sdk_bob-1.1.1/brynq_sdk_bob/custom_tables.py

@@ -0,0 +1,36 @@
+from datetime import datetime
+import pandas as pd
+from brynq_sdk_functions import Functions
+from .schemas.custom_tables import CustomTableSchema
+
+
+class CustomTables:
+    def __init__(self, bob):
+        self.bob = bob
+
+    def get(self, employee_id: str, custom_table_id: str) -> (pd.DataFrame, pd.DataFrame):
+        """
+        Get custom table data for an employee
+        
+        Args:
+            employee_id: The employee ID
+            custom_table_id: The custom table ID
+            
+        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()
+        
+        # Normalize the nested JSON response
+        df = pd.json_normalize(
+            data,
+            record_path=['values']
+        )
+        
+        df['employee_id'] = employee_id
+        df = self.bob.rename_camel_columns_to_snake_case(df)
+        valid_data, invalid_data = Functions.validate_data(df=df, schema=CustomTableSchema, debug=True)
+
+        return valid_data, invalid_data 

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob/documents.py

@@ -13,7 +13,7 @@ class CustomDocuments:
         # self.headers_upload['Accept'] = 'application/json'
 
     def get(self, person_id: datetime) -> pd.DataFrame:
-        resp = self.bob.session.get(url=f"{self.bob.base_url}docs/people/{person_id}")
+        resp = self.bob.session.get(url=f"{self.bob.base_url}docs/people/{person_id}", timeout=self.bob.timeout)
         resp.raise_for_status()
         data = resp.json()['documents']
         df = pd.DataFrame(data)
@@ -29,7 +29,7 @@ class CustomDocuments:
         return df
 
     def get_folders(self) -> dict:
-        resp = self.bob.session.get(url=f"{self.bob.base_url}docs/folders/metadata")
+        resp = self.bob.session.get(url=f"{self.bob.base_url}docs/folders/metadata", timeout=self.bob.timeout)
         resp.raise_for_status()
         data = resp.json()
 
@@ -42,5 +42,6 @@ class CustomDocuments:
                file_object: BytesIO):
         files = {"file": (file_name, file_object, "application/pdf")}
         resp = self.bob.session.post(url=f"{self.bob.base_url}docs/people/{person_id}/folders/{folder_id}/upload",
-                                     files=files)
+                                     files=files,
+                                     timeout=self.bob.timeout)
         resp.raise_for_status()

brynq_sdk_bob-1.1.1/brynq_sdk_bob/named_lists.py

@@ -0,0 +1,37 @@
+from datetime import datetime
+import pandas as pd
+from brynq_sdk_functions import Functions
+from .schemas.named_lists import NamedListSchema
+
+
+class NamedLists:
+    def __init__(self, bob):
+        self.bob = bob
+
+    def get(self) -> (pd.DataFrame, pd.DataFrame):
+        """
+        Get custom table data for an employee
+        
+        Args:
+            list_name: The list name
+            
+        Returns:
+            A tuple of (valid_data, invalid_data) as pandas DataFrames
+        """
+        url = f"{self.bob.base_url}company/named-lists/"
+        resp = self.bob.session.get(url=url)
+        resp.raise_for_status()
+        data = resp.json()
+
+        df = pd.DataFrame([
+            {**item, "type": key}
+            for key, group in data.items()
+            for item in group["values"]
+        ])
+        
+        # Normalize the nested JSON response
+        # df = pd.DataFrame(data.get('values'))
+        df = self.bob.rename_camel_columns_to_snake_case(df)
+        valid_data, invalid_data = Functions.validate_data(df=df, schema=NamedListSchema, debug=True)
+
+        return valid_data, invalid_data 

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob/payments.py

@@ -9,7 +9,7 @@ class Payments:
         self.bob = bob
 
     def get(self, person_id: str) -> (pd.DataFrame, pd.DataFrame):
-        resp = self.bob.session.get(url=f"{self.bob.base_url}people/{person_id}/variable")
+        resp = self.bob.session.get(url=f"{self.bob.base_url}people/{person_id}/variable", timeout=self.bob.timeout)
         resp.raise_for_status()
         data = resp.json()
         df = pd.json_normalize(

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob/people.py

@@ -1,11 +1,11 @@
 import pandas as pd
 from brynq_sdk_functions import Functions
-
 from .bank import Bank
 from .employment import Employment
 from .salaries import Salaries
 from .schemas.people import PeopleSchema
 from .work import Work
+from .custom_tables import CustomTables
 
 
 class People:
@@ -15,9 +15,10 @@ class People:
         self.employment = Employment(bob)
         self.bank = Bank(bob)
         self.work = Work(bob)
+        self.custom_tables = CustomTables(bob)
 
     def get(self) -> pd.DataFrame:
-        resp = self.bob.session.get(url=f"{self.bob.base_url}profiles")
+        resp = self.bob.session.get(url=f"{self.bob.base_url}profiles", timeout=self.bob.timeout)
         # Bob sucks with default fields so you need to do a search call to retrieve additional fields.
         additional_fields = [
             "personal.birthDate",
@@ -27,21 +28,29 @@ class People:
             "address.line2",
             "address.activeEffectiveDate",
             "address.country",
-            "home.legalGender",
+            # "home.legalGender",
             "home.spouse.firstName",
             "home.spouse.surname",
-            "home.spouse.birthDate",
+            # "home.spouse.birthDate",
             "home.spouse.gender",
             "internal.terminationReason",
             "internal.terminationDate",
             "internal.terminationType",
-            "employee.lastDayOfWork"
+            "employee.lastDayOfWork",
+            # housenumber addition
+            "address.customColumns.column_1740046184782",
+            # contract end date (bob only fills this when you get a new contract normally)
+            "payroll.employment.customColumns.column_1680013460318",
+            # iban
+            "financial.iban"
+            # ploegentoeslag
         ]
         resp_additional_fields = self.bob.session.post(url=f"{self.bob.base_url}people/search",
                                                        json={
                                                            "fields": ["root.id"] + additional_fields,
                                                            "filters": []
-                                                       })
+                                                       },
+                                                       timeout=self.bob.timeout)
         df_extra_fields = pd.json_normalize(resp_additional_fields.json()['employees'])
         resp.raise_for_status()
         data = resp.json()

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob/schemas/bank.py

@@ -5,6 +5,7 @@ import pandas as pd
 
 class BankSchema(pa.DataFrameModel):
     id: Series[pd.Int64Dtype] = pa.Field(coerce=True)
+    employee_id: Series[pd.Int64Dtype] = pa.Field(coerce=True)
     amount: Series[pd.Int64Dtype] = pa.Field(coerce=True)
     allocation: Series[String] = pa.Field(coerce=True, nullable=True)
     branch_address: Series[String] = pa.Field(coerce=True, nullable=True)

brynq_sdk_bob-1.1.1/brynq_sdk_bob/schemas/custom_tables.py

@@ -0,0 +1,10 @@
+import pandera as pa
+from pandera.typing import Series
+import pandas as pd
+
+class CustomTableSchema(pa.DataFrameModel):
+    id: Series[pd.Int64Dtype] = pa.Field(coerce=True)
+    employee_id: Series[pd.Int64Dtype] = pa.Field(coerce=True)
+    
+    class Config:
+        coerce = True 

brynq_sdk_bob-1.1.1/brynq_sdk_bob/schemas/named_lists.py

@@ -0,0 +1,13 @@
+import pandera as pa
+from pandera.typing import Series
+
+class NamedListSchema(pa.DataFrameModel):
+    id: Series[str] = pa.Field(coerce=True)
+    value: Series[str] = pa.Field(coerce=True)
+    name: Series[str] = pa.Field(coerce=True)
+    archived: Series[bool] = pa.Field(coerce=True)
+    # children: Series[list] = pa.Field(coerce=True)
+    type: Series[str] = pa.Field(coerce=True)
+
+    class Config:
+        coerce = True 

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob/schemas/people.py

@@ -24,10 +24,10 @@ class PeopleSchema(pa.DataFrameModel):
     personal_pronouns: Series[String] = pa.Field(coerce=True, nullable=True)
     personal_honorific: Series[String] = pa.Field(coerce=True, nullable=True)
     personal_nationality: Series[object] = pa.Field(coerce=True, check_name=check_list)
-    employee_payroll_manager: Series[String] = pa.Field(coerce=True, nullable=True)
-    employee_hrbp: Series[String] = pa.Field(coerce=True, nullable=True)
-    employee_it_admin: Series[String] = pa.Field(coerce=True, nullable=True)
-    employee_buddy: Series[String] = pa.Field(coerce=True, nullable=True)
+    # employee_payroll_manager: Series[String] = pa.Field(coerce=True, nullable=True)
+    # employee_hrbp: Series[String] = pa.Field(coerce=True, nullable=True)
+    # employee_it_admin: Series[String] = pa.Field(coerce=True, nullable=True)
+    # employee_buddy: Series[String] = pa.Field(coerce=True, nullable=True)
     employee_veteran_status: Series[object] = pa.Field(coerce=True, check_name=check_list)
     employee_disability_status: Series[String] = pa.Field(coerce=True, nullable=True)
     work_start_date: Series[DateTime] = pa.Field(coerce=True, nullable=True)
@@ -58,12 +58,12 @@ class PeopleSchema(pa.DataFrameModel):
     work_original_start_date: Series[DateTime] = pa.Field(coerce=True, nullable=True)
     work_active_effective_date: Series[DateTime] = pa.Field(coerce=True, nullable=True)
     work_direct_reports: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True)
-    work_work_change_type: Series[String] = pa.Field(coerce=True, nullable=True)
+    # work_work_change_type: Series[String] = pa.Field(coerce=True, nullable=True)
     work_second_level_manager: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True)
     work_days_of_previous_service: Series[pd.Int64Dtype] = pa.Field(coerce=True, nullable=True)
     work_years_of_service: Series[Float] = pa.Field(coerce=True, nullable=True)
     about_food_preferences: Series[object] = pa.Field(coerce=True, check_name=check_list)
-    about_social_data_linkedin: Series[String] = pa.Field(coerce=True, nullable=True)
+    # about_social_data_linkedin: Series[String] = pa.Field(coerce=True, nullable=True)
     about_social_data_twitter: Series[String] = pa.Field(coerce=True, nullable=True)
     about_social_data_facebook: Series[String] = pa.Field(coerce=True, nullable=True)
     about_superpowers: Series[object] = pa.Field(coerce=True, check_name=check_list)
@@ -74,12 +74,12 @@ class PeopleSchema(pa.DataFrameModel):
     address_post_code: Series[String] = pa.Field(coerce=True, nullable=True)
     address_line1: Series[String] = pa.Field(coerce=True, nullable=True)
     address_line2: Series[String] = pa.Field(coerce=True, nullable=True)
-    address_country: Series[String] = pa.Field(coerce=True, nullable=True)
+    # address_country: Series[String] = pa.Field(coerce=True, nullable=True)
     address_active_effective_date: Series[DateTime] = pa.Field(coerce=True, nullable=True)
-    home_legal_gender: Series[String] = pa.Field(coerce=True, nullable=True)
+    # home_legal_gender: Series[String] = pa.Field(coerce=True, nullable=True)
     home_spouse_first_name: Series[String] = pa.Field(coerce=True, nullable=True)
     home_spouse_surname: Series[String] = pa.Field(coerce=True, nullable=True)
-    home_spouse_birth_date: Series[DateTime] = pa.Field(coerce=True, nullable=True)
+    # home_spouse_birth_date: Series[DateTime] = pa.Field(coerce=True, nullable=True)
     home_spouse_gender: Series[String] = pa.Field(coerce=True, nullable=True)
     internal_termination_reason: Series[String] = pa.Field(coerce=True, nullable=True)
     internal_termination_date: Series[DateTime] = pa.Field(coerce=True, nullable=True)

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob/timeoff.py

@@ -10,7 +10,8 @@ class TimeOff:
 
     def get(self, since: datetime) -> (pd.DataFrame, pd.DataFrame):
         resp = self.bob.session.get(url=f"{self.bob.base_url}timeoff/requests/changes",
-                                    params={'since': since.replace(tzinfo=timezone.utc).isoformat(timespec='milliseconds')})
+                                    params={'since': since.replace(tzinfo=timezone.utc).isoformat(timespec='milliseconds')},
+                                    timeout=self.bob.timeout)
         resp.raise_for_status()
         data = resp.json()['changes']
         # data = self.bob.get_paginated_result(request)

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq-sdk-bob
-Version: 1.0.0
+Version: 1.1.1
 Summary: Bob wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob.egg-info/SOURCES.txt

@@ -2,8 +2,10 @@ setup.py
 brynq_sdk_bob/__init__.py
 brynq_sdk_bob/bank.py
 brynq_sdk_bob/company.py
+brynq_sdk_bob/custom_tables.py
 brynq_sdk_bob/documents.py
 brynq_sdk_bob/employment.py
+brynq_sdk_bob/named_lists.py
 brynq_sdk_bob/payments.py
 brynq_sdk_bob/people.py
 brynq_sdk_bob/salaries.py
@@ -17,7 +19,9 @@ brynq_sdk_bob.egg-info/requires.txt
 brynq_sdk_bob.egg-info/top_level.txt
 brynq_sdk_bob/schemas/__init__.py
 brynq_sdk_bob/schemas/bank.py
+brynq_sdk_bob/schemas/custom_tables.py
 brynq_sdk_bob/schemas/employment.py
+brynq_sdk_bob/schemas/named_lists.py
 brynq_sdk_bob/schemas/payments.py
 brynq_sdk_bob/schemas/people.py
 brynq_sdk_bob/schemas/salary.py

{brynq_sdk_bob-1.0.0 → brynq_sdk_bob-1.1.1}/setup.py

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