brynq-sdk-bob 2.6.6__tar.gz → 2.6.7__tar.gz

{brynq_sdk_bob-2.6.6 → brynq_sdk_bob-2.6.7}/PKG-INFO

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

{brynq_sdk_bob-2.6.6 → brynq_sdk_bob-2.6.7}/brynq_sdk_bob/people.py

@@ -1,4 +1,5 @@
 import pandas as pd
+import re
 from typing import Optional
 from brynq_sdk_functions import Functions
 from brynq_sdk_functions import BrynQPanderaDataFrameModel
@@ -21,6 +22,7 @@ class People:
         self.schema = PeopleSchema
 
 
+
         # Build API fields using column metadata if present (api_field), otherwise use the column (alias) name
     def __build_api_fields(self, schema_model: BrynQPanderaDataFrameModel) -> list[str]:
         schema = schema_model.to_schema()
@@ -74,6 +76,30 @@ class People:
 
 
         valid_people, invalid_people = Functions.validate_data(df=valid_people, schema=PeopleSchema, debug=True)
-        valid_people = valid_people.loc[:, ~valid_people.columns.str.contains(r'\.value|_get')]
 
+        # For columns ending with .value.value, use fillna to fill the corresponding base column since that's not done automatically
+        def _normalize_key(key: str) -> str:
+            # Build mapping from custom schema alias (API path) to the real column name used after validation
+            key = key.lstrip('.')
+            key = key.replace('/', '.')
+            key = key.replace('.', '_')
+            key = re.sub(r'(?<!^)([A-Z])', r'_\1', key).lower()
+            return key
+
+        alias_to_real: dict = {}
+        if schema_custom_fields is not None:
+            alias_map = getattr(schema_custom_fields, "_alias_map", {})  # real_name -> alias
+            alias_to_real = {_normalize_key(alias): real for real, alias in alias_map.items()}
+
+        for col in valid_people.columns:
+            if col.endswith('value.value'):
+                # Compute the base alias (remove suffix), normalise, and resolve to real column name via custom schema
+                base_alias = col[:-12]
+                target_key = _normalize_key(base_alias)
+                target_col = alias_to_real.get(target_key, target_key)
+                if target_col in valid_people.columns:
+                    valid_people[target_col] = valid_people[target_col].fillna(valid_people[col])
+
+        # Remove columns that contain '.value' or '_get'
+        valid_people = valid_people.loc[:, ~valid_people.columns.str.contains(r'\.value|_get')]
         return valid_people, pd.concat([invalid_people, invalid_people_custom])

{brynq_sdk_bob-2.6.6 → brynq_sdk_bob-2.6.7}/brynq_sdk_bob.egg-info/PKG-INFO

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

{brynq_sdk_bob-2.6.6 → brynq_sdk_bob-2.6.7}/setup.py

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