brynq-sdk-bob 2.2.1__tar.gz → 2.3.0__tar.gz

{brynq_sdk_bob-2.2.1 → brynq_sdk_bob-2.3.0}/PKG-INFO

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

{brynq_sdk_bob-2.2.1 → brynq_sdk_bob-2.3.0}/brynq_sdk_bob/schemas/timeoff.py

@@ -27,3 +27,21 @@ class TimeOffSchema(BrynQPanderaDataFrameModel):
 
     class Config:
         coerce = True
+
+
+class TimeOffBalanceSchema(BrynQPanderaDataFrameModel):
+    employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
+    policy_type_name: Series[String] = pa.Field(coerce=True, description="Policy Type Name", alias="policyTypeName")
+    policy_type_display_name: Series[String] = pa.Field(coerce=True, description="Policy Type Display Name", alias="policyTypeDisplayName")
+    balance: Series[Float] = pa.Field(coerce=True, description="Current Balance", alias="balance")
+    used: Series[Float] = pa.Field(coerce=True, description="Used Balance", alias="used")
+    available: Series[Float] = pa.Field(coerce=True, description="Available Balance", alias="available")
+    approved_requests: Series[Float] = pa.Field(coerce=True, description="Approved Requests", alias="approvedRequests")
+    pending_requests: Series[Float] = pa.Field(coerce=True, description="Pending Requests", alias="pendingRequests")
+    as_of_date: Series[String] = pa.Field(coerce=True, description="As of Date", alias="asOfDate")
+    accrual_start_date: Series[String] = pa.Field(nullable=True, coerce=True, description="Accrual Start Date", alias="accrualStartDate")
+    expiry_date: Series[String] = pa.Field(nullable=True, coerce=True, description="Expiry Date", alias="expiryDate")
+    duration_unit: Series[String] = pa.Field(coerce=True, description="Duration Unit", alias="durationUnit")
+
+    class Config:
+        coerce = True

{brynq_sdk_bob-2.2.1 → brynq_sdk_bob-2.3.0}/brynq_sdk_bob/timeoff.py

@@ -1,7 +1,7 @@
 from datetime import datetime, timezone, timedelta
 import pandas as pd
 from brynq_sdk_functions import Functions
-from .schemas.timeoff import TimeOffSchema
+from .schemas.timeoff import TimeOffSchema, TimeOffBalanceSchema
 import warnings
 
 
@@ -9,6 +9,7 @@ class TimeOff:
     def __init__(self, bob):
         self.bob = bob
         self.schema = TimeOffSchema
+        self.balance_schema = TimeOffBalanceSchema
 
     def get(self, since: datetime = None) -> tuple[pd.DataFrame, pd.DataFrame]:
         """
@@ -49,3 +50,34 @@ class TimeOff:
         valid_timeoff, invalid_timeoff = Functions.validate_data(df=df, schema=self.schema, debug=True)
 
         return valid_timeoff, invalid_timeoff
+
+    def get_balance(self, employee_id: str, policy_type: str = None, as_of_date: str = None) -> tuple[pd.DataFrame, pd.DataFrame]:
+        """
+        Get time off balance for a specific employee
+
+        Args:
+            employee_id (str): The ID of the employee
+            policy_type (str, optional): The policy type to filter by. Defaults to None.
+            as_of_date (str, optional): The date to get balance as of (YYYY-MM-DD format). Defaults to None.
+
+        Returns:
+            tuple[pd.DataFrame, pd.DataFrame]: A tuple of (valid_balance, invalid_balance) as pandas DataFrames
+        """
+        params = {}
+        if policy_type:
+            params['policyType'] = policy_type
+        if as_of_date:
+            params['asOfDate'] = as_of_date
+
+        resp = self.bob.session.get(
+            url=f"{self.bob.base_url}timeoff/employees/{employee_id}/balance",
+            params=params,
+            timeout=self.bob.timeout
+        )
+        resp.raise_for_status()
+        data = resp.json()
+        df = pd.DataFrame(data)
+
+        valid_balance, invalid_balance = Functions.validate_data(df=df, schema=self.balance_schema, debug=True)
+
+        return valid_balance, invalid_balance

{brynq_sdk_bob-2.2.1 → brynq_sdk_bob-2.3.0}/brynq_sdk_bob.egg-info/PKG-INFO

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

{brynq_sdk_bob-2.2.1 → brynq_sdk_bob-2.3.0}/setup.py

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