brynq-sdk-bob 2.6.8__tar.gz → 2.8.0__tar.gz

{brynq_sdk_bob-2.6.8 → brynq_sdk_bob-2.8.0}/PKG-INFO

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

{brynq_sdk_bob-2.6.8 → brynq_sdk_bob-2.8.0}/brynq_sdk_bob/__init__.py

@@ -2,6 +2,7 @@ import base64
 from typing import List, Optional, Literal
 import requests
 import os
+from .reports import Reports
 from brynq_sdk_brynq import BrynQ
 from .bank import Bank
 from .company import Company
@@ -39,6 +40,7 @@ class Bob(BrynQ):
         self.named_lists = NamedLists(self)
         self.custom_tables = CustomTables(self)
         self.payroll_history = History(self)
+        self.reports = Reports(self)
         self.data_interface_id = os.getenv("DATA_INTERFACE_ID")
         self.debug = debug
 

brynq_sdk_bob-2.8.0/brynq_sdk_bob/reports.py

@@ -0,0 +1,38 @@
+from datetime import datetime
+from io import BytesIO
+from typing import Optional, TYPE_CHECKING
+
+import pandas as pd
+from brynq_sdk_functions import Functions
+if TYPE_CHECKING:
+    from brynq_sdk_bob import Bob
+
+
+class Reports:
+    def __init__(self, bob):
+        self.bob: Bob = bob
+
+    def get(self) -> pd.DataFrame:
+        resp = self.bob.session.get(url=f"{self.bob.base_url}company/reports", timeout=self.bob.timeout)
+        resp.raise_for_status()
+        data = resp.json()
+        df = pd.json_normalize(
+            data,
+            record_path='views'
+        )
+        # df = self.bob.rename_camel_columns_to_snake_case(df)
+        # valid_documents, invalid_documents = Functions.validate_data(df=df, schema=DocumentsSchema, debug=True)
+
+        return df
+
+    def download(self, report_id: int | str = None) -> bytes:
+        if report_id:
+            url = f"{self.bob.base_url}company/reports/{report_id}/download"
+        else:
+            raise ValueError("Either report_id or report_name must be provided")
+
+        resp = self.bob.session.get(url=url, timeout=self.bob.timeout, params={"format": "csv"})
+        resp.raise_for_status()
+        data = resp.content
+
+        return data

{brynq_sdk_bob-2.6.8 → brynq_sdk_bob-2.8.0}/brynq_sdk_bob.egg-info/PKG-INFO

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

{brynq_sdk_bob-2.6.8 → brynq_sdk_bob-2.8.0}/brynq_sdk_bob.egg-info/SOURCES.txt

@@ -9,6 +9,7 @@ brynq_sdk_bob/named_lists.py
 brynq_sdk_bob/payments.py
 brynq_sdk_bob/payroll_history.py
 brynq_sdk_bob/people.py
+brynq_sdk_bob/reports.py
 brynq_sdk_bob/salaries.py
 brynq_sdk_bob/timeoff.py
 brynq_sdk_bob/work.py

{brynq_sdk_bob-2.6.8 → brynq_sdk_bob-2.8.0}/setup.py

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