lkj 0.1.48__py3-none-any.whl → 0.1.49__py3-none-any.whl

lkj/dicts.py

@@ -48,17 +48,21 @@ def truncate_dict_values(
     max_list_size: int | None = 2,
     max_string_size: int | None = 66,
     middle_marker: str = "...",
+    d_ingress: callable = lambda d: d,
 ) -> dict:
     """
     Returns a new dictionary with the same nested keys structure, where:
     - List values are reduced to a maximum size of max_list_size.
     - String values longer than max_string_size are truncated in the middle.
+    - Values are preprocessed with d_ingress before type checking.
 
     Parameters:
     d (dict): The input dictionary.
     max_list_size (int, optional): Maximum size for lists. Defaults to 2.
     max_string_size (int, optional): Maximum length for strings. Defaults to None (no truncation).
     middle_marker (str, optional): String to insert in the middle of truncated strings. Defaults to '...'.
+    d_ingress (callable, optional): Function applied to each value before type checking.
+        Defaults to identity function. Useful for handling special types like pandas Series.
 
     Returns:
     dict: A new dictionary with truncated lists and strings.
@@ -79,6 +83,14 @@ def truncate_dict_values(
     ...     == large_dict
     ... )
 
+    For handling special types like pandas Series:
+
+    >>> def handle_pandas(obj):
+    ...     if hasattr(obj, 'head'):  # pandas Series/DataFrame
+    ...         return f"<{type(obj).__name__}: {len(obj)} items>"
+    ...     return obj
+    >>> # truncate_dict_values(data_with_pandas, d_ingress=handle_pandas)
+
     """
 
     def truncate_string(value, max_len, marker):
@@ -87,19 +99,26 @@ def truncate_dict_values(
         half_len = (max_len - len(marker)) // 2
         return value[:half_len] + marker + value[-half_len:]
 
+    # Apply d_ingress to preprocess the value
+    d = d_ingress(d)
+
     kwargs = dict(
         max_list_size=max_list_size,
         max_string_size=max_string_size,
         middle_marker=middle_marker,
+        d_ingress=d_ingress,
     )
     if isinstance(d, dict):
         return {k: truncate_dict_values(v, **kwargs) for k, v in d.items()}
-    elif isinstance(d, list):
-        return (
-            [truncate_dict_values(v, **kwargs) for v in d[:max_list_size]]
-            if max_list_size is not None
-            else d
-        )
+    elif isinstance(d, (list, tuple)):
+        container_type = type(d)
+        if max_list_size is not None:
+            truncated_items = [
+                truncate_dict_values(v, **kwargs) for v in d[:max_list_size]
+            ]
+        else:
+            truncated_items = [truncate_dict_values(v, **kwargs) for v in d]
+        return container_type(truncated_items)
     elif isinstance(d, str):
         return truncate_string(d, max_string_size, middle_marker)
     else:

{lkj-0.1.48.dist-info → lkj-0.1.49.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lkj
-Version: 0.1.48
+Version: 0.1.49
 Summary: A dump of homeless useful utils
 Home-page: https://github.com/thorwhalen/lkj
 Author: Thor Whalen

{lkj-0.1.48.dist-info → lkj-0.1.49.dist-info}/RECORD

@@ -1,6 +1,6 @@
 lkj/__init__.py,sha256=z8IZi5ql-bCM0xRgcEERg48oi2SRkgX9qSZdyaKb62Y,7300
 lkj/chunking.py,sha256=K7yakDuTgyn6t2rvFeDIWS5pO636BKGOe1spKk7v4Mc,3746
-lkj/dicts.py,sha256=0Ow8AgMHx2Og7hAAkBjFd3f2YTBhRbSvZ9fO-xkGkpU,10430
+lkj/dicts.py,sha256=67E9qwxPLLsBeRj67cSDgkAGvB6xfmNsi8igXdFGwjA,11307
 lkj/filesys.py,sha256=hYWztDoJUvVcPMD5AxRyYKkTB-sy5hx6_RDw4OyRcos,14628
 lkj/funcs.py,sha256=LXJlj3gMMsbD0t2gn2NZZ6mOqmW5bxM-94uGoYgrhzI,8930
 lkj/importing.py,sha256=TcW3qUDmw7jqswpxXnksjlHkkbOJq70NbUk1ZyaafT0,6658
@@ -10,8 +10,8 @@ lkj/misc.py,sha256=RqJg4HgcT1ZZRBtxOppwYibvvLIyn6pNqI-WMg0uKgI,1457
 lkj/strings.py,sha256=JukQ4pvLlwGajBEK0PXyqsumEFKm8i1rx6SgASz7tqo,43798
 lkj/tests/__init__.py,sha256=kReYfWiyz1T79AuZAy5m4PIBj3Oj_Dlc0E-T8HVQ504,20
 lkj/tests/test_strings.py,sha256=Ix54io8WqWmYTY4guBgZShtKabtjd5uVpdxzxwJkgNA,1707
-lkj-0.1.48.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-lkj-0.1.48.dist-info/METADATA,sha256=qACdUbDx6cImMq9DmVvGqYiFFO2mxH3ImUwQfTGjY5U,8267
-lkj-0.1.48.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-lkj-0.1.48.dist-info/top_level.txt,sha256=3DZOUwYmyurJFBXQCvCmEIVm8z2b42O5Sx3RDQyePfg,4
-lkj-0.1.48.dist-info/RECORD,,
+lkj-0.1.49.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+lkj-0.1.49.dist-info/METADATA,sha256=pURwBZHm355AXQBzLViKArzBWbJhCFLxiNiHRaNRih8,8267
+lkj-0.1.49.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+lkj-0.1.49.dist-info/top_level.txt,sha256=3DZOUwYmyurJFBXQCvCmEIVm8z2b42O5Sx3RDQyePfg,4
+lkj-0.1.49.dist-info/RECORD,,