brynq-sdk-bob 2.8.4__py3-none-any.whl → 2.9.0__py3-none-any.whl

brynq_sdk_bob/__init__.py

@@ -21,8 +21,9 @@ class Bob(BrynQ):
     def __init__(self, system_type: Optional[Literal['source', 'target']] = None, test_environment: bool = True, debug: bool = False, target_system: str = None):
         super().__init__()
         self.timeout = 3600
+        self.test_environment = test_environment
         self.headers = self._get_request_headers(system_type)
-        if test_environment:
+        if self.test_environment:
             self.base_url = "https://api.sandbox.hibob.com/v1/"
         else:
             self.base_url = "https://api.hibob.com/v1/"
@@ -46,6 +47,15 @@ class Bob(BrynQ):
 
     def _get_request_headers(self, system_type):
         credentials = self.interfaces.credentials.get(system='bob', system_type=system_type)
+        # multiple creds possible, not fetched by environment test status, get first occurence
+        if isinstance(credentials, list):
+            credentials = next(
+                (
+                    element for element in credentials
+                    if element.get('data', {}).get('Test Environment') == self.test_environment
+                ),
+                credentials[0]
+            )
         auth_token = base64.b64encode(f"{credentials.get('data').get('User ID')}:{credentials.get('data').get('API Token')}".encode()).decode('utf-8')
         headers = {
             "accept": "application/json",

brynq_sdk_bob/schemas/timeoff.py

@@ -4,14 +4,20 @@ from typing import Optional
 import pandas as pd
 from brynq_sdk_functions import BrynQPanderaDataFrameModel
 
+
+# =============================================================================
+# TimeOffSchema - For /timeoff/requests/changes endpoint (change events)
+# =============================================================================
+
 class TimeOffSchema(BrynQPanderaDataFrameModel):
+    """Schema for time off change events from /timeoff/requests/changes endpoint."""
     change_type: Series[String] = pa.Field(coerce=True, description="Change Type", alias="changeType")
     employee_id: Series[String] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
     employee_display_name: Series[String] = pa.Field(coerce=True, description="Employee Display Name", alias="employeeDisplayName")
     employee_email: Series[String] = pa.Field(coerce=True, description="Employee Email", alias="employeeEmail")
     request_id: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Request ID", alias="requestId")
     policy_type_display_name: Series[String] = pa.Field(coerce=True, description="Policy Type Display Name", alias="policyTypeDisplayName")
-    type: Series[String] = pa.Field(coerce=True, description="Type", alias="type")
+    type: Series[String] = pa.Field(coerce=True, description="Request type", alias="type")
     start_date: Series[String] = pa.Field(coerce=True, nullable=True, description="Start Date", alias="startDate")
     start_portion: Series[String] = pa.Field(coerce=True, nullable=True, description="Start Portion", alias="startPortion")
     end_date: Series[String] = pa.Field(coerce=True, nullable=True, description="End Date", alias="endDate")
@@ -30,7 +36,191 @@ class TimeOffSchema(BrynQPanderaDataFrameModel):
         coerce = True
 
 
+# =============================================================================
+# TimeOffRequest - For /timeoff/employees/{id}/requests/{requestId} endpoint
+# =============================================================================
+
+class TimeOffRequest(BrynQPanderaDataFrameModel):
+    """
+    Schema for time off request details from Bob API.
+
+    Based on: https://apidocs.hibob.com/reference/get_timeoff-employees-id-requests-requestid
+
+    Supports all request types (discriminated by 'type' field):
+    - days: Request for X days
+    - hours: Request for X hours during the day (policy types measured in hours)
+    - portionOnRange: Every morning or afternoon during days requested
+    - hoursOnRange: X hours every day during days requested
+    - differentDayDurations: Different hours on each day requested
+    - specificHoursDayDurations: Specific hours per day
+    - differentSpecificHoursDayDurations: Different specific hours on each day
+    - percentageOnRange: X percent of every day during days requested
+    - openEnded: Request without an end date yet
+
+    All type-specific fields are optional since they vary by request type.
+
+    Note: Complex nested fields (attachmentLinks, durations arrays) are not included
+    """
+
+    # -------------------------------------------------------------------------
+    # IDENTIFIERS
+    # -------------------------------------------------------------------------
+    employee_id: Series[String] = pa.Field(
+        coerce=True, description="Employee ID", alias="employeeId"
+    )
+    request_id: Series[pd.Int64Dtype] = pa.Field(
+        coerce=True, description="Time Off Request ID", alias="requestId"
+    )
+
+    # -------------------------------------------------------------------------
+    # REQUEST METADATA
+    # -------------------------------------------------------------------------
+    policy_type_display_name: Series[String] = pa.Field(
+        coerce=True, description="Display name of the policy type", alias="policyTypeDisplayName"
+    )
+    created_on: Series[String] = pa.Field(
+        coerce=True, description="Date and time the request was created", alias="createdOn"
+    )
+    description: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True, description="Request description", alias="description"
+    )
+
+    # -------------------------------------------------------------------------
+    # TYPE DISCRIMINATOR
+    # Valid values: days, hours, portionOnRange, hoursOnRange, differentDayDurations,
+    #               specificHoursDayDurations, differentSpecificHoursDayDurations,
+    #               percentageOnRange, openEnded
+    # -------------------------------------------------------------------------
+    type: Series[String] = pa.Field(
+        coerce=True, description="Request type discriminator", alias="type"
+    )
+
+    # GENERAL INFO
+    duration_unit: Series[String] = pa.Field(
+        coerce=True, description="Unit for totalDuration/totalCost: 'days' or 'hours'", alias="durationUnit"
+    )
+    total_duration: Series[Float] = pa.Field(
+        coerce=True, description="Total time including regular days off", alias="totalDuration"
+    )
+    total_cost: Series[Float] = pa.Field(
+        coerce=True, description="Amount deducted from balance", alias="totalCost"
+    )
+    status: Series[String] = pa.Field(
+        coerce=True, description="Request status: approved, pending, canceled, etc.", alias="status"
+    )
+    approved: Series[pd.BooleanDtype] = pa.Field(
+        coerce=True, description="Whether request is approved", alias="approved"
+    )
+
+    has_attachment: Series[pd.BooleanDtype] = pa.Field(
+        coerce=True, description="Whether request has attachments", alias="hasAttachment"
+    )
+    # Note: attachmentLinks array is not included (complex nested structure)
+
+    reason_code: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True, description="Reason code from policy type's list", alias="reasonCode"
+    )
+
+    previous_request_id: Optional[Series[pd.Int64Dtype]] = pa.Field(
+        nullable=True, coerce=True,
+        description="ID of replaced request when date/time updated", alias="previousRequestId"
+    )
+    original_request_id: Optional[Series[pd.Int64Dtype]] = pa.Field(
+        nullable=True, coerce=True,
+        description="ID of the very first request in history chain", alias="originalRequestId"
+    )
+
+    approved_by: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True, description="Who approved the request", alias="approvedBy"
+    )
+    approved_at: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True, description="When request was approved", alias="approvedAt"
+    )
+
+    declined_by: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True, description="Who declined the request", alias="declinedBy"
+    )
+    declined_at: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True, description="When request was declined", alias="declinedAt"
+    )
+    decline_reason: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True, description="Why request was declined", alias="declineReason"
+    )
+
+    visibility: Series[String] = pa.Field(
+        coerce=True, description="Visibility: 'Public', 'Private' or 'Custom name'", alias="visibility"
+    )
+    time_zone_offset: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True,
+        description="GMT offset (e.g., 'GMT -5:00') for requests with specific times", alias="timeZoneOffset"
+    )
+
+    # -------------------------------------------------------------------------
+    # TYPE-SPECIFIC FIELDS (optional, presence depends on 'type' value)
+    # -------------------------------------------------------------------------
+
+    # For types: days, portionOnRange, hoursOnRange, differentDayDurations,
+    #            specificHoursDayDurations, differentSpecificHoursDayDurations,
+    #            percentageOnRange, openEnded
+    start_date: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True, description="First day of time off", alias="startDate"
+    )
+    end_date: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True, description="Last day of time off (null for openEnded)", alias="endDate"
+    )
+
+    # For types: days, openEnded
+    start_portion: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True,
+        description="First day portion: all_day, morning, afternoon", alias="startPortion"
+    )
+    end_portion: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True,
+        description="Last day portion: all_day, morning, afternoon (null for openEnded)", alias="endPortion"
+    )
+
+    # For type: hours
+    date: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True, description="Date for single-day hours request", alias="date"
+    )
+    hours_on_date: Optional[Series[Float]] = pa.Field(
+        nullable=True, coerce=True, description="Hours for single-day request", alias="hoursOnDate"
+    )
+
+    # For type: portionOnRange
+    day_portion: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True,
+        description="Portion for range: morning or afternoon", alias="dayPortion"
+    )
+
+    # For type: hoursOnRange
+    daily_hours: Optional[Series[Float]] = pa.Field(
+        nullable=True, coerce=True, description="Hours per day for range", alias="dailyHours"
+    )
+
+    # For type: percentageOnRange
+    percentage_of_day: Optional[Series[pd.Int64Dtype]] = pa.Field(
+        nullable=True, coerce=True, description="Percent of each day requested", alias="percentageOfDay"
+    )
+
+    # For types: specificHoursDayDurations, differentSpecificHoursDayDurations, openEnded
+    time_zone: Optional[Series[String]] = pa.Field(
+        nullable=True, coerce=True,
+        description="Time zone name (e.g., 'Europe/London')", alias="timeZone"
+    )
+
+    # Note: 'durations' array is not included (complex nested structure with per-day details)
+
+    class Config:
+        coerce = True
+
+
+# =============================================================================
+# TimeOffBalanceSchema - For /timeoff/employees/{id}/balance endpoint
+# =============================================================================
+
 class TimeOffBalanceSchema(BrynQPanderaDataFrameModel):
+    """Schema for time off balance from /timeoff/employees/{id}/balance endpoint."""
     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")

brynq_sdk_bob/timeoff.py

@@ -1,7 +1,8 @@
 from datetime import datetime, timezone, timedelta
+from typing import Union
 import pandas as pd
 from brynq_sdk_functions import Functions
-from .schemas.timeoff import TimeOffSchema, TimeOffBalanceSchema
+from .schemas.timeoff import TimeOffSchema, TimeOffBalanceSchema, TimeOffRequest
 import warnings
 
 
@@ -40,13 +41,43 @@ class TimeOff:
                                     params={'since': since, 'includePending': 'true' if include_pending else 'false'},
                                     timeout=self.bob.timeout)
         resp.raise_for_status()
-        data = resp.json()['changes']
-        # data = self.bob.get_paginated_result(request)
+        data = resp.json().get('changes', [])
         df = pd.DataFrame(data)
         valid_timeoff, invalid_timeoff = Functions.validate_data(df=df, schema=self.schema, debug=True)
 
         return valid_timeoff, invalid_timeoff
 
+    def get_by_request_id(
+        self,
+        employee_id: Union[str, int],
+        request_id: Union[str, int],
+    ) -> tuple[pd.DataFrame, pd.DataFrame]:
+        """
+        Get time off request details by request ID.
+
+        Args:
+            employee_id: The Employee ID (from database or Bob URL).
+                Example: "3332883884017713238" from URL "https://app.hibob.com/employee-profile/3332883884017713238"
+            request_id: The time off request ID.
+
+        Returns:
+            tuple[pd.DataFrame, pd.DataFrame]: (valid_request, invalid_request) as single-row DataFrames.
+        """
+        resp = self.bob.session.get(
+            url=f"{self.bob.base_url}timeoff/employees/{employee_id}/requests/{request_id}",
+            timeout=self.bob.timeout
+        )
+        resp.raise_for_status()
+        data = resp.json()
+
+        # Single request returns a dict, wrap in list for DataFrame
+        df = pd.DataFrame([data])
+
+        valid_request, invalid_request = Functions.validate_data(df=df, schema=TimeOffRequest, debug=True)
+
+        return valid_request, invalid_request
+
+
     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

{brynq_sdk_bob-2.8.4.dist-info → brynq_sdk_bob-2.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: brynq_sdk_bob
-Version: 2.8.4
+Version: 2.9.0
 Summary: Bob wrapper from BrynQ
 Author: BrynQ
 Author-email: support@brynq.com

{brynq_sdk_bob-2.8.4.dist-info → brynq_sdk_bob-2.9.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-brynq_sdk_bob/__init__.py,sha256=M2y-juQ-GgzglQa0HvbWmeA9fBzv5L3MShHbTfb9rsk,3014
+brynq_sdk_bob/__init__.py,sha256=AFIEA5akwZqu-USQbMt26Rp8krqumvOsG1J83jhMQSg,3470
 brynq_sdk_bob/bank.py,sha256=zTdfe_qCZt2FB7SZbQ7njIDspwTinLFdbeH_xUby2FY,966
 brynq_sdk_bob/company.py,sha256=rjOpkm0CZ1EeJ-jddBl36GrGKUQviC1ca1aUL2tl1_M,848
 brynq_sdk_bob/custom_tables.py,sha256=MvnR2mIcyK0rpwd0P7xV3BPIvCYQVEClBvo901GttPs,2642
@@ -10,7 +10,7 @@ brynq_sdk_bob/payroll_history.py,sha256=wHo6da7kLDe1ViL4egyMdyJBMZnWVhwjNjmh4cTC
 brynq_sdk_bob/people.py,sha256=t1A1dABX6UZ0pyLTGOL-Sp5pHY630KWIyIO3JQ_Pjdk,5970
 brynq_sdk_bob/reports.py,sha256=Tawmqm_ZmQ487loyk-29-A_fTCrgImbWCEf6zfwuaq4,1245
 brynq_sdk_bob/salaries.py,sha256=BGQm-PT9QuKKJ9DP5nX6wmC8SZRAlm9M9I2EJhoZaII,1523
-brynq_sdk_bob/timeoff.py,sha256=NbBZ39qy9D7jbS_z9bpmB-BKNuUGmNrkYTbEw034tZ0,3339
+brynq_sdk_bob/timeoff.py,sha256=JtTu14PWFqQIEn9r-Z8ipeNE-5p7hqPz5N6wjjBeLTs,4438
 brynq_sdk_bob/work.py,sha256=0bVZkQ0I6z-z2_ql-EsOpFExx8VgsJvpcCQdOfiJYQM,712
 brynq_sdk_bob/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 brynq_sdk_bob/schemas/bank.py,sha256=lDmXP4P8091N20fL2CmhPU2wFuaK60Z-p-dvYSNCMaQ,1846
@@ -21,9 +21,9 @@ brynq_sdk_bob/schemas/payments.py,sha256=LrSr8WApYxqbMDmhYmh0EISEbWJinWovsULZV6s
 brynq_sdk_bob/schemas/payroll_history.py,sha256=JdAq0XaArHHEw8EsXo3GD0EhSAyBhPtYQMmdvjCiY8g,806
 brynq_sdk_bob/schemas/people.py,sha256=42BJVgJmT-h5kzuQl6iI7wZDSGNA0KTQQVIAqeeyHNk,40149
 brynq_sdk_bob/schemas/salary.py,sha256=TSaM1g92y3oiDcUrfJW7ushgKZenI9xB6XW3kKuU0dE,4540
-brynq_sdk_bob/schemas/timeoff.py,sha256=uswH42djiiWE2H0wXKHoneDvOJQ5BlaIeNdNl-kYd-s,4117
+brynq_sdk_bob/schemas/timeoff.py,sha256=gTYu_bNcfHrkTz4eIHCZ4WzgMTj2U4nI3X6JTzDovhk,12817
 brynq_sdk_bob/schemas/work.py,sha256=YgtBJ0WXJOq55bFlT_kY_IbHh0SlQEtaa0W8vms-xA4,3048
-brynq_sdk_bob-2.8.4.dist-info/METADATA,sha256=i8xKzW4Se01HsfeIOHSe_uHIxGVpornQ0Elqhdz0kHE,371
-brynq_sdk_bob-2.8.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-brynq_sdk_bob-2.8.4.dist-info/top_level.txt,sha256=oGiWqOuAAiVoLIzGe6F-Lo4IJBYz5ftOwBft7HtPuoY,14
-brynq_sdk_bob-2.8.4.dist-info/RECORD,,
+brynq_sdk_bob-2.9.0.dist-info/METADATA,sha256=aYpW-QeJZdjrvK9wfDMu3y228359XcZVyZLim7TNgc8,371
+brynq_sdk_bob-2.9.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+brynq_sdk_bob-2.9.0.dist-info/top_level.txt,sha256=oGiWqOuAAiVoLIzGe6F-Lo4IJBYz5ftOwBft7HtPuoY,14
+brynq_sdk_bob-2.9.0.dist-info/RECORD,,