brynq-sdk-bob 2.6.2.dev6__tar.gz → 2.6.2.dev8__tar.gz

{brynq_sdk_bob-2.6.2.dev6 → brynq_sdk_bob-2.6.2.dev8}/PKG-INFO

@@ -1,12 +1,13 @@
 Metadata-Version: 2.4
 Name: brynq_sdk_bob
-Version: 2.6.2.dev6
+Version: 2.6.2.dev8
 Summary: Bob wrapper from BrynQ
 Author: BrynQ
 Author-email: support@brynq.com
 License: BrynQ License
 Requires-Dist: brynq-sdk-brynq<5,>=4
 Requires-Dist: pandas<3.0.0,>=2.2.0
+Requires-Dist: tenacity==8.2.3
 Dynamic: author
 Dynamic: author-email
 Dynamic: description

{brynq_sdk_bob-2.6.2.dev6 → brynq_sdk_bob-2.6.2.dev8}/brynq_sdk_bob/bank.py

@@ -4,17 +4,24 @@ from .schemas.bank import BankSchema
 
 import time
 from tqdm import tqdm
+from tenacity import retry, stop_after_delay, wait_exponential
 
 class Bank:
     def __init__(self, bob):
         self.bob = bob
         self.schema = BankSchema
 
+    @retry(stop=stop_after_delay(120), wait=wait_exponential(multiplier=1, min=1, max=10))
+    def _get_bank_accounts(self, person_id: str):
+        """Fetch bank accounts for a person with retry logic (max 2 minutes)."""
+        resp = self.bob.session.get(url=f"{self.bob.base_url}people/{person_id}/bank-accounts", timeout=self.bob.timeout)
+        resp.raise_for_status()
+        return resp
+
     def get(self, person_ids: pd.Series, field_selection: list[str] = []) -> (pd.DataFrame, pd.DataFrame):
         data = []
         for person_id in tqdm(person_ids, desc="Fetching 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()
+            resp = self._get_bank_accounts(person_id)
             temp_data = resp.json()['values']
             # when an employee has one or more bank accounts, the response is a list of dictionaries.
             for account in temp_data:

{brynq_sdk_bob-2.6.2.dev6 → brynq_sdk_bob-2.6.2.dev8}/brynq_sdk_bob/payments.py

@@ -1,10 +1,12 @@
+import time
+from typing import List, Optional
+
 import pandas as pd
-from typing import Optional, List
+from tqdm import tqdm
+
 from brynq_sdk_functions import Functions
-from .schemas.payments import VariablePaymentSchema, ActualPaymentsSchema
 
-import time
-from tqdm import tqdm
+from .schemas.payments import ActualPaymentsSchema, VariablePaymentSchema
 
 
 class Payments:
@@ -31,6 +33,9 @@ class Payments:
             for key, value in named_lists.items()
         }
 
+        # rename payrollVariableType to variableType in named lists
+        named_lists['variableType'] = named_lists['payrollVariableType']
+
         for field in df.columns:
             # Fields in the response and in the named-list have different building blocks
             # but they both end with the same last block

{brynq_sdk_bob-2.6.2.dev6 → brynq_sdk_bob-2.6.2.dev8}/brynq_sdk_bob.egg-info/PKG-INFO

@@ -1,12 +1,13 @@
 Metadata-Version: 2.4
 Name: brynq_sdk_bob
-Version: 2.6.2.dev6
+Version: 2.6.2.dev8
 Summary: Bob wrapper from BrynQ
 Author: BrynQ
 Author-email: support@brynq.com
 License: BrynQ License
 Requires-Dist: brynq-sdk-brynq<5,>=4
 Requires-Dist: pandas<3.0.0,>=2.2.0
+Requires-Dist: tenacity==8.2.3
 Dynamic: author
 Dynamic: author-email
 Dynamic: description

{brynq_sdk_bob-2.6.2.dev6 → brynq_sdk_bob-2.6.2.dev8}/brynq_sdk_bob.egg-info/requires.txt

@@ -1,2 +1,3 @@
 brynq-sdk-brynq<5,>=4
 pandas<3.0.0,>=2.2.0
+tenacity==8.2.3

{brynq_sdk_bob-2.6.2.dev6 → brynq_sdk_bob-2.6.2.dev8}/setup.py

@@ -2,7 +2,7 @@ from setuptools import find_namespace_packages, setup
 
 setup(
     name='brynq_sdk_bob',
-    version='2.6.2.dev6',
+    version='2.6.2.dev8',
     description='Bob wrapper from BrynQ',
     long_description='Bob wrapper from BrynQ',
     author='BrynQ',
@@ -12,6 +12,7 @@ setup(
     install_requires=[
         'brynq-sdk-brynq>=4,<5',
         'pandas>=2.2.0,<3.0.0',
+        'tenacity==8.2.3',
     ],
     zip_safe=False,
 )