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

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

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

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

@@ -35,6 +35,7 @@ class Bob(BrynQ):
         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)
@@ -53,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.1.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.1.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.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.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.1.0 → brynq_sdk_bob-1.1.1}/brynq_sdk_bob/people.py

@@ -18,7 +18,7 @@ class People:
         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",
@@ -49,7 +49,8 @@ class People:
                                                        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.1.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.1.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.1.0
+Version: 1.1.1
 Summary: Bob wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_bob-1.1.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.1.0',
+    version='1.1.1',
     description='Bob wrapper from BrynQ',
     long_description='Bob wrapper from BrynQ',
     author='BrynQ',