thds.core 1.45.20250818180904__py3-none-any.whl → 1.45.20250821162245__py3-none-any.whl

thds/core/thread_debug.py

@@ -0,0 +1,114 @@
+import os
+import re
+import sys
+import threading
+import types
+import typing as ty
+
+from thds.core import log
+
+logger = log.getLogger(__name__)
+
+
+def _format_frame(frame: types.FrameType) -> str:
+    """Format a stack frame as a single readable line."""
+    return f"{frame.f_code.co_filename}:{frame.f_lineno} in {frame.f_code.co_name}"
+
+
+def _get_thread_stack_frames(thread: threading.Thread) -> list[str]:
+    """Get stack frames for a specific thread."""
+    frames = list[str]()
+
+    if thread.ident is None:
+        logger.warning('Thread "%s" has no identifier; cannot get stack frames.', thread.name)
+        return frames
+
+    # Get the frame for this thread
+    current_frames = sys._current_frames()
+    frame = current_frames.get(thread.ident)
+
+    if frame is None:
+        logger.warning("unable to get stack frames for thread %s (id: %s)")
+        return frames
+
+    # Get the path of this debug module
+    debug_module_path = os.path.abspath(__file__)
+
+    # Walk the stack
+    while frame:
+        if not (debug_module_path and os.path.abspath(frame.f_code.co_filename) == debug_module_path):
+
+            frames.append(_format_frame(frame))
+        frame = frame.f_back
+    return frames
+
+
+def _get_thread_info(thread: threading.Thread) -> dict[str, ty.Any]:
+    """Extract thread information including stack trace."""
+    stack_frames = _get_thread_stack_frames(thread)
+    return {
+        "thread_id": thread.ident,
+        "thread_name": thread.name,
+        "stack_frames": stack_frames,
+    }
+
+
+def _is_thread_pool_thread(thread: threading.Thread) -> bool:
+    """Check if thread appears to belong to a ThreadPoolExecutor."""
+    tpe_patterns = [
+        r"ThreadPoolExecutor-\d+_\d+",  # Default naming
+        r".*-\d+_\d+",  # Custom prefix with TPE suffix pattern
+    ]
+
+    for pattern in tpe_patterns:
+        if re.match(pattern, thread.name):
+            return True
+
+    return False
+
+
+def _find_potential_parent_threads() -> list[dict[str, ty.Any]]:
+    """Find threads that could be parents of the current thread pool thread."""
+    potential_parents = []
+    current_thread = threading.current_thread()
+
+    for thread in threading.enumerate():
+        # Skip self
+        if thread == current_thread:
+            continue
+
+        # Skip daemon threads
+        if thread.daemon:
+            continue
+
+        # Skip other thread pool threads
+        if _is_thread_pool_thread(thread):
+            continue
+
+        thread_info = _get_thread_info(thread)
+        potential_parents.append(thread_info)
+
+    return potential_parents
+
+
+def capture_thread_context() -> dict[str, ty.Any]:
+    """
+    Capture stack trace for current thread and potential parent threads.
+
+    Returns:
+        Dict with 'current_thread' info and 'potential_parents' list.
+    """
+    current_thread = threading.current_thread()
+
+    # Always capture current thread info
+    current_info = _get_thread_info(current_thread)
+
+    # Only look for parents if we think we're in a thread pool
+    potential_parents = []
+    if _is_thread_pool_thread(current_thread):
+        potential_parents = _find_potential_parent_threads()
+
+    return {
+        "current_thread": current_info,
+        "potential_parents": potential_parents,
+    }

{thds_core-1.45.20250818180904.dist-info → thds_core-1.45.20250821162245.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: thds.core
-Version: 1.45.20250818180904
+Version: 1.45.20250821162245
 Summary: Core utilities.
 Author-email: Trilliant Health <info@trillianthealth.com>
 License: MIT

{thds_core-1.45.20250818180904.dist-info → thds_core-1.45.20250821162245.dist-info}/RECORD

@@ -41,6 +41,7 @@ thds/core/scaling.py,sha256=f7CtdgK0sN6nroTq5hLAkG8xwbWhbCZUULstSKjoxO0,1615
 thds/core/scope.py,sha256=9RWWCFRqsgjTyH6rzRm_WnO69N_sEBRaykarc2PAnBY,10834
 thds/core/source_serde.py,sha256=X4c7LiT3VidejqtTel9YB6dWGB3x-ct39KF9E50Nbx4,139
 thds/core/stack_context.py,sha256=17lPOuYWclUpZ-VXRkPgI4WbiMzq7_ZY6Kj1QK_1oNo,1332
+thds/core/thread_debug.py,sha256=ox0XmZQxkwoeIVyM6-BV4PNic5VbucJF3GhjkmaP5UQ,3240
 thds/core/thunks.py,sha256=p1OvMBJ4VGMsD8BVA7zwPeAp0L3y_nxVozBF2E78t3M,1053
 thds/core/timer.py,sha256=aOpNP-wHKaKs6ONK5fOtIOgx00FChVZquG4PeaEYH_k,5376
 thds/core/tmp.py,sha256=jA8FwDbXo3hx8o4kRjAlkwpcI77X86GY4Sktkps29ho,3166
@@ -73,8 +74,8 @@ thds/core/sqlite/structured.py,sha256=SvZ67KcVcVdmpR52JSd52vMTW2ALUXmlHEeD-VrzWV
 thds/core/sqlite/types.py,sha256=oUkfoKRYNGDPZRk29s09rc9ha3SCk2SKr_K6WKebBFs,1308
 thds/core/sqlite/upsert.py,sha256=BmKK6fsGVedt43iY-Lp7dnAu8aJ1e9CYlPVEQR2pMj4,5827
 thds/core/sqlite/write.py,sha256=z0219vDkQDCnsV0WLvsj94keItr7H4j7Y_evbcoBrWU,3458
-thds_core-1.45.20250818180904.dist-info/METADATA,sha256=ZpphogKhoUCtgZWuIDmHCyoejG7qNHOfiRRw4_8yurE,2216
-thds_core-1.45.20250818180904.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-thds_core-1.45.20250818180904.dist-info/entry_points.txt,sha256=bOCOVhKZv7azF3FvaWX6uxE6yrjK6FcjqhtxXvLiFY8,161
-thds_core-1.45.20250818180904.dist-info/top_level.txt,sha256=LTZaE5SkWJwv9bwOlMbIhiS-JWQEEIcjVYnJrt-CriY,5
-thds_core-1.45.20250818180904.dist-info/RECORD,,
+thds_core-1.45.20250821162245.dist-info/METADATA,sha256=VsnlsTYqIWz8IhH0Ol1fV6SNMna2E-u7chdnqJ9WOeI,2216
+thds_core-1.45.20250821162245.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+thds_core-1.45.20250821162245.dist-info/entry_points.txt,sha256=bOCOVhKZv7azF3FvaWX6uxE6yrjK6FcjqhtxXvLiFY8,161
+thds_core-1.45.20250821162245.dist-info/top_level.txt,sha256=LTZaE5SkWJwv9bwOlMbIhiS-JWQEEIcjVYnJrt-CriY,5
+thds_core-1.45.20250821162245.dist-info/RECORD,,