brynq-sdk-bob 2.4.3__tar.gz → 2.5.0__tar.gz

{brynq_sdk_bob-2.4.3 → brynq_sdk_bob-2.5.0}/PKG-INFO

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

{brynq_sdk_bob-2.4.3 → brynq_sdk_bob-2.5.0}/brynq_sdk_bob/people.py

@@ -196,7 +196,8 @@ class People:
 
         Args:
             additional_fields (list[str]): Additional fields to get (not defined in the schema)
-            field_selection (list[str]): Fields to get (defined in the schema), if not provided, all fields are returned
+            field_selection (list[str]): Fields to get (defined in the schema), if not provided, all fields are returned.
+                Can be either pythonic field names (e.g., 'work_start_date') or Bob API field names (e.g., 'work.startDate').
             add_payroll_information (list[str]): List of payroll information types to include (valid options: 'entitlement', 'variable')
 
         Returns:
@@ -210,6 +211,9 @@ class People:
         response_fields = list(set(self.field_name_in_response + additional_fields))
 
         if field_selection:
+            # Convert pythonic field names to aliases (e.g., 'work_start_date' -> 'work.startDate')
+            field_selection = PeopleSchema.convert_pythonic_to_alias(field_selection)
+
             body_fields = [field for field in body_fields if field in field_selection]
             response_fields = [self.endpoint_to_response.get(field) for field in field_selection if field in self.endpoint_to_response]
 

{brynq_sdk_bob-2.4.3 → brynq_sdk_bob-2.5.0}/brynq_sdk_bob/schemas/payments.py

@@ -9,7 +9,7 @@ class VariablePaymentSchema(BrynQPanderaDataFrameModel):
     department_percent: Series[Float] = pa.Field(nullable=True, coerce=True, description="Department Percent", alias="departmentPercent")
     payout_type: Series[String] = pa.Field(coerce=True, description="Payout Type", alias="payoutType")
     num_of_salaries: Optional[Series[pd.Int64Dtype]] = pa.Field(nullable=True, coerce=True, description="Number of Salaries", alias="numOfSalaries")
-    end_date: Series[DateTime] = pa.Field(nullable=True, coerce=True, description="End Date", alias="endDate")
+    end_date: Optional[Series[DateTime]] = pa.Field(nullable=True, coerce=True, description="End Date", alias="endDate")
     creation_date: Series[DateTime] = pa.Field(coerce=True, description="Creation Date", alias="creationDate")
     percentage_of_annual_salary: Series[Float] = pa.Field(nullable=True, coerce=True, description="Percentage of Annual Salary", alias="percentageOfAnnualSalary")
     individual_percent: Series[Float] = pa.Field(nullable=True, coerce=True, description="Individual Percent", alias="individualPercent")

{brynq_sdk_bob-2.4.3 → brynq_sdk_bob-2.5.0}/brynq_sdk_bob/schemas/people.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Optional
+from typing import Optional, List, Dict
 
 import pandas as pd
 import pandera as pa
@@ -98,3 +98,39 @@ class PeopleSchema(BrynQPanderaDataFrameModel):
 
     class Config:
         coerce = True
+
+    @classmethod
+    def convert_pythonic_to_alias(cls, pythonic_fields: List[str]) -> List[str]:
+        """
+        Convert pythonic field names to their Bob API aliases.
+
+        Args:
+            pythonic_fields: List of pythonic field names (e.g., ['work_start_date', 'first_name'])
+
+        Returns:
+            List of Bob API field aliases (e.g., ['work.startDate', 'root.firstName'])
+        """
+        # Get the alias mapping from the class (populated by BrynQPanderaDataFrameModel.__init_subclass__)
+        alias_map = getattr(cls, "_alias_map", {})
+
+        # Convert the provided pythonic fields to aliases
+        result = []
+        for field in pythonic_fields:
+            if field in alias_map:
+                result.append(alias_map[field])
+            else:
+                # If not found in schema, keep the field as-is (could be a custom field)
+                result.append(field)
+
+        return result
+
+    @classmethod
+    def get_pythonic_to_alias_mapping(cls) -> Dict[str, str]:
+        """
+        Get the complete mapping from pythonic field names to Bob API aliases.
+
+        Returns:
+            Dictionary mapping pythonic field names to aliases
+        """
+        # Get the alias mapping from the class (populated by BrynQPanderaDataFrameModel.__init_subclass__)
+        return getattr(cls, "_alias_map", {})

{brynq_sdk_bob-2.4.3 → brynq_sdk_bob-2.5.0}/brynq_sdk_bob.egg-info/PKG-INFO

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

{brynq_sdk_bob-2.4.3 → brynq_sdk_bob-2.5.0}/setup.py

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