brynq-sdk-bob 2.6.2.dev10__py3-none-any.whl

brynq_sdk_bob/schemas/timeoff.py

@@ -0,0 +1,48 @@
+import pandera as pa
+from pandera.typing import Series, String, Float
+from typing import Optional
+import pandas as pd
+from brynq_sdk_functions import BrynQPanderaDataFrameModel
+
+class TimeOffSchema(BrynQPanderaDataFrameModel):
+    change_type: Series[String] = pa.Field(coerce=True, description="Change Type", alias="changeType")
+    employee_id: Series[pd.Int64Dtype] = 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")
+    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")
+    end_portion: Series[String] = pa.Field(coerce=True, nullable=True, description="End Portion", alias="endPortion")
+    day_portion: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Day Portion", alias="dayPortion")
+    date: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Date", alias="date")
+    hours_on_date: Optional[Series[Float]] = pa.Field(coerce=True, nullable=True, description="Hours on Date", alias="hoursOnDate")
+    daily_hours: Optional[Series[Float]] = pa.Field(coerce=True, nullable=True, description="Daily Hours", alias="dailyHours")
+    duration_unit: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Duration Unit", alias="durationUnit")
+    total_duration: Optional[Series[Float]] = pa.Field(coerce=True, nullable=True, description="Total Duration", alias="totalDuration")
+    total_cost: Optional[Series[Float]] = pa.Field(coerce=True, nullable=True, description="Total Cost", alias="totalCost")
+    change_reason: Optional[Series[String]] = pa.Field(nullable=True, coerce=True, description="Change Reason", alias="changeReason")
+    visibility: Optional[Series[String]] = pa.Field(coerce=True, nullable=True, description="Visibility", alias="visibility")
+
+    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/schemas/work.py

@@ -0,0 +1,32 @@
+import pandera as pa
+from pandera.typing import Series
+import pandas as pd
+from datetime import datetime
+from brynq_sdk_functions import BrynQPanderaDataFrameModel
+
+class WorkSchema(BrynQPanderaDataFrameModel):
+    can_be_deleted: Series[pa.Bool] = pa.Field(coerce=True, description="Can Be Deleted", alias="canBeDeleted")
+    work_change_type: Series[str] = pa.Field(coerce=True, description="Work Change Type", alias="workChangeType")
+    creation_date: Series[datetime] = pa.Field(coerce=True, nullable=True, description="Creation Date", alias="creationDate")
+    title: Series[str] = pa.Field(coerce=True, nullable=True, description="Title", alias="title")
+    is_current: Series[pa.Bool] = pa.Field(coerce=True, description="Is Current", alias="isCurrent")
+    modification_date: Series[datetime] = pa.Field(coerce=True, nullable=True, description="Modification Date", alias="modificationDate")
+    site: Series[str] = pa.Field(coerce=True, nullable=True, description="Site", alias="site")
+    site_id: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="Site ID", alias="siteId")
+    id: Series[pd.Int64Dtype] = pa.Field(coerce=True, description="ID", alias="id")
+    end_effective_date: Series[datetime] = pa.Field(coerce=True, nullable=True, description="End Effective Date", alias="endEffectiveDate")
+    active_effective_date: Series[datetime] = pa.Field(coerce=True, nullable=True, description="Active Effective Date", alias="activeEffectiveDate")
+    department: Series[str] = pa.Field(coerce=True, nullable=True, description="Department", alias="department")
+    effective_date: Series[datetime] = pa.Field(coerce=True, nullable=True, description="Effective Date", alias="effectiveDate")
+    change_reason: Series[str] = pa.Field(coerce=True, nullable=True, description="Change Reason", alias="change.reason")
+    change_changed_by: Series[str] = pa.Field(coerce=True, nullable=True, description="Change Changed By", alias="change.changedBy")
+    change_changed_by_id: Series[str] = pa.Field(coerce=True, nullable=True, description="Change Changed By ID", alias="change.changedById")
+    reports_to_id: Series[str] = pa.Field(coerce=True, nullable=True, description="Reports To ID", alias="reportsTo.id")
+    reports_to_first_name: Series[str] = pa.Field(coerce=True, nullable=True, description="Reports To First Name", alias="reportsTo.firstName")
+    reports_to_surname: Series[str] = pa.Field(coerce=True, nullable=True, description="Reports To Surname", alias="reportsTo.surname")
+    reports_to_email: Series[str] = pa.Field(coerce=True, nullable=True, description="Reports To Email", alias="reportsTo.email")
+    reports_to_display_name: Series[str] = pa.Field(coerce=True, nullable=True, description="Reports To Display Name", alias="reportsTo.displayName")
+    employee_id: Series[str] = pa.Field(coerce=True, description="Employee ID", alias="employeeId")
+
+    class Config:
+        coerce = True

brynq_sdk_bob/timeoff.py

@@ -0,0 +1,79 @@
+from datetime import datetime, timezone, timedelta
+import pandas as pd
+from brynq_sdk_functions import Functions
+from .schemas.timeoff import TimeOffSchema, TimeOffBalanceSchema
+import warnings
+
+
+class TimeOff:
+    def __init__(self, bob):
+        self.bob = bob
+        self.schema = TimeOffSchema
+        self.balance_schema = TimeOffBalanceSchema
+
+    def get(self, since: datetime = None, include_pending: bool = False) -> tuple[pd.DataFrame, pd.DataFrame]:
+        """
+        Get time off requests
+
+        Args:
+            since (datetime, optional): The start date of the time off requests max 6 months ago. Defaults to 6 months ago if not provided.
+
+        Returns:
+            tuple[pd.DataFrame, pd.DataFrame]: A tuple of (valid_timeoff, invalid_timeoff) as pandas DataFrames
+        """
+        max_lookback = timedelta(days=int((365 / 2) - 1))
+        now_utc = datetime.now(timezone.utc)
+        six_months_ago = now_utc - max_lookback
+        #if since is provided, use it and cap to 6 months max
+        if since is not None:
+            if since.tzinfo is None:
+                since = since.replace(tzinfo=timezone.utc)
+            if since < six_months_ago:
+                warnings.warn("The 'since' date is more than 6 months ago. Limiting to 6 months ago due to API restrictions.")
+                since = six_months_ago
+        #if no since is provided, use 6 months ago
+        else:
+            since = six_months_ago
+
+        since = since.replace(tzinfo=timezone.utc).isoformat(timespec='milliseconds').replace('+00:00', 'Z')
+        resp = self.bob.session.get(url=f"{self.bob.base_url}timeoff/requests/changes",
+                                    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)
+        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_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/work.py

@@ -0,0 +1,31 @@
+import pandas as pd
+import requests
+from typing import Optional, List
+from brynq_sdk_functions import Functions
+from .schemas.work import WorkSchema
+
+
+class Work:
+    def __init__(self, bob):
+        self.bob = bob
+        self.schema = WorkSchema
+
+    def get(self, employee_ids: Optional[List[str]] = None) ->(pd.DataFrame, pd.DataFrame):
+        params = {"limit": 100}
+
+        # Add employeeIds filter if provided
+        if employee_ids is not None:
+            params["employeeIds"] = ",".join(employee_ids)
+
+        request = requests.Request(method='GET',
+                                   url=f"{self.bob.base_url}bulk/people/work",
+                                   params=params)
+        data = self.bob.get_paginated_result(request)
+        df = pd.json_normalize(
+            data,
+            record_path='values',
+            meta=['employeeId']
+        )
+        valid_work, invalid_work = Functions.validate_data(df=df, schema=self.schema, debug=True)
+
+        return valid_work, invalid_work

brynq_sdk_bob-2.6.2.dev10.dist-info/METADATA

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: brynq_sdk_bob
+Version: 2.6.2.dev10
+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
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: summary
+
+Bob wrapper from BrynQ

brynq_sdk_bob-2.6.2.dev10.dist-info/RECORD

@@ -0,0 +1,28 @@
+brynq_sdk_bob/__init__.py,sha256=bw2Www_wgVxqZczJpz9K_94k7OdgqwWKF0aQaKG5TQE,3052
+brynq_sdk_bob/bank.py,sha256=5ncWsmeAvNoiLnVvIvPgFDRQ5DB5_VY2ipMC439GVHU,1460
+brynq_sdk_bob/company.py,sha256=rjOpkm0CZ1EeJ-jddBl36GrGKUQviC1ca1aUL2tl1_M,848
+brynq_sdk_bob/custom_tables.py,sha256=MvnR2mIcyK0rpwd0P7xV3BPIvCYQVEClBvo901GttPs,2642
+brynq_sdk_bob/documents.py,sha256=ww101owiBGARCxOANdDtmWrNedSBe9V-BEed6QspQPg,1756
+brynq_sdk_bob/employment.py,sha256=kBEKfUmKEw-A_FjC9fOuJqcsT7NxUKwXJs_V_-x9LbI,765
+brynq_sdk_bob/named_lists.py,sha256=ksLXV2ysBFegq4gZiiaC56gjkgdnPzL7WajZTGvjYIM,1069
+brynq_sdk_bob/payments.py,sha256=e1TnJcXlbotOfULukVUiaYZ1N-bDGynAtmPgpux7It0,5670
+brynq_sdk_bob/payroll_history.py,sha256=wHo6da7kLDe1ViL4egyMdyJBMZnWVhwjNjmh4cTCTeY,3972
+brynq_sdk_bob/people.py,sha256=6b7uCucl_xqtAq_4YJZU457fUY0qssMwfCCStzbNG0M,5036
+brynq_sdk_bob/salaries.py,sha256=8xq9XDTK473Az2MpuAPofz9CvZstjufSoWtF0bi1wC4,1766
+brynq_sdk_bob/timeoff.py,sha256=NbBZ39qy9D7jbS_z9bpmB-BKNuUGmNrkYTbEw034tZ0,3339
+brynq_sdk_bob/work.py,sha256=IADipEuI_ofhhLZZV9oI8PV10VN07yluPPkLZT19Ze8,1013
+brynq_sdk_bob/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+brynq_sdk_bob/schemas/bank.py,sha256=Yfq-ouzcAyaqcZj5_lWLcaQgtLikI9zlcsHhsuPUQzY,1935
+brynq_sdk_bob/schemas/custom_tables.py,sha256=638NH2n50vHzw1LFlbKVCBKBhfSsfGqtEGuor0Z-QS4,1567
+brynq_sdk_bob/schemas/employment.py,sha256=uErbSyl8xYaYo5Hu0FZ7_tl1WCOaytcU9pjkIWJbmOo,3226
+brynq_sdk_bob/schemas/named_lists.py,sha256=HJBRKrAI2vhrkq-5MVXqQcmpGNzFtoOnaZI2Ii_6_vs,725
+brynq_sdk_bob/schemas/payments.py,sha256=N4Ylommed-hsw0aX7rjuZuVZQc4XU0iDBdSIxz8eAxc,4292
+brynq_sdk_bob/schemas/payroll_history.py,sha256=JdAq0XaArHHEw8EsXo3GD0EhSAyBhPtYQMmdvjCiY8g,806
+brynq_sdk_bob/schemas/people.py,sha256=cy0mLKRdUhjOv52WB53AWd1XHn6mwrACGZAvK2ufsz4,38780
+brynq_sdk_bob/schemas/salary.py,sha256=TSaM1g92y3oiDcUrfJW7ushgKZenI9xB6XW3kKuU0dE,4540
+brynq_sdk_bob/schemas/timeoff.py,sha256=BHImTTT4n8j7bF7T5Ue_B0WHmmj1_QTPV9TKAlHeBZM,4124
+brynq_sdk_bob/schemas/work.py,sha256=klzJtQf-avwhOkRHuPTM-jMhTIsroDPE56da1H5xaYs,2926
+brynq_sdk_bob-2.6.2.dev10.dist-info/METADATA,sha256=hlLk5XIWFqHhiKBwQb8E-EPiyr34wWndRQvBxcSP5_M,408
+brynq_sdk_bob-2.6.2.dev10.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+brynq_sdk_bob-2.6.2.dev10.dist-info/top_level.txt,sha256=oGiWqOuAAiVoLIzGe6F-Lo4IJBYz5ftOwBft7HtPuoY,14
+brynq_sdk_bob-2.6.2.dev10.dist-info/RECORD,,

brynq_sdk_bob-2.6.2.dev10.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.10.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

brynq_sdk_bob-2.6.2.dev10.dist-info/top_level.txt

@@ -0,0 +1 @@
+brynq_sdk_bob