matrice-streaming 0.1.14__py3-none-any.whl → 0.1.65__py3-none-any.whl

matrice_streaming/__init__.py

@@ -1,37 +1,49 @@
 """Module providing __init__ functionality."""
-from matrice_common.utils import dependencies_check
+import multiprocessing
 
-base = [
-    "httpx",
-    "fastapi", 
-    "uvicorn",
-    "pillow",
-    "confluent_kafka[snappy]",
-    "aiokafka",
-    "aiohttp",
-    "filterpy",
-    "scipy",
-    "scikit-learn", 
-    "matplotlib",
-    "scikit-image",
-    "python-snappy",
-    "pyyaml",
-    "imagehash",
-]
+# Only run dependency checks in the main process, NOT in spawned child processes
+# (ProcessPoolExecutor workers). Child processes re-import modules which would
+# trigger pip install commands, causing crashes with BrokenProcessPool errors.
+_is_main_process = multiprocessing.parent_process() is None
 
-# Install base dependencies first
-dependencies_check(base)
+if _is_main_process:
+    from matrice_common.utils import dependencies_check
 
-# Helper to attempt installation and verify importability
-def _install_and_verify(pkg: str, import_name: str):
-    """Install a package expression and return True if the import succeeds."""
-    if dependencies_check([pkg]):
-        try:
-            __import__(import_name)
-            return True
-        except ImportError:
-            return False
-    return False
+    base = [
+        "httpx",
+        "fastapi", 
+        "uvicorn",
+        "pillow",
+        "confluent_kafka[snappy]",
+        "aiokafka",
+        "aiohttp",
+        "filterpy",
+        "scipy",
+        "scikit-learn", 
+        "matplotlib",
+        "scikit-image",
+        "python-snappy",
+        "pyyaml",
+        "imagehash",
+        "psutil"
+    ]
 
-if not dependencies_check(["opencv-python"]):
-    dependencies_check(["opencv-python-headless"])
+    # Install base dependencies first
+    dependencies_check(base)
+
+    # Helper to attempt installation and verify importability
+    def _install_and_verify(pkg: str, import_name: str):
+        """Install a package expression and return True if the import succeeds."""
+        if dependencies_check([pkg]):
+            try:
+                __import__(import_name)
+                return True
+            except ImportError:
+                return False
+        return False
+
+    try:
+        import cv2
+    except ImportError:
+        if not dependencies_check(["opencv-python"]):
+            dependencies_check(["opencv-python-headless"])

matrice_streaming/streaming_gateway/camera_streamer/__init__.py

@@ -1,3 +1,70 @@
 from .camera_streamer import CameraStreamer
+from .worker_manager import WorkerManager
+from .async_camera_worker import AsyncCameraWorker
 
-__all__ = ["CameraStreamer"]
+# GStreamer components (optional - graceful import)
+try:
+    from .gstreamer_camera_streamer import (
+        GStreamerCameraStreamer,
+        GStreamerConfig,
+        GStreamerPipeline,
+        is_gstreamer_available,
+    )
+    from .gstreamer_worker import (
+        GStreamerAsyncWorker,
+        run_gstreamer_worker,
+    )
+    from .gstreamer_worker_manager import GStreamerWorkerManager
+
+    GSTREAMER_AVAILABLE = is_gstreamer_available()
+except (ImportError, ValueError):
+    # ImportError: gi module not available
+    # ValueError: gi.require_version fails when GStreamer not installed
+    GSTREAMER_AVAILABLE = False
+    GStreamerCameraStreamer = None
+    GStreamerConfig = None
+    GStreamerPipeline = None
+    GStreamerAsyncWorker = None
+    GStreamerWorkerManager = None
+
+    def is_gstreamer_available():
+        return False
+
+# NVDEC components (optional - graceful import)
+try:
+    from .nvdec_worker_manager import (
+        NVDECWorkerManager,
+        is_nvdec_available,
+        get_available_gpu_count,
+    )
+    NVDEC_AVAILABLE = is_nvdec_available()
+except ImportError:
+    # NVDEC not available (requires CuPy, PyNvVideoCodec)
+    NVDEC_AVAILABLE = False
+    NVDECWorkerManager = None
+
+    def is_nvdec_available():
+        return False
+
+    def get_available_gpu_count():
+        return 1
+
+__all__ = [
+    # Original components
+    "CameraStreamer",
+    "WorkerManager",
+    "AsyncCameraWorker",
+    # GStreamer components
+    "GStreamerCameraStreamer",
+    "GStreamerConfig",
+    "GStreamerPipeline",
+    "GStreamerAsyncWorker",
+    "GStreamerWorkerManager",
+    "is_gstreamer_available",
+    "GSTREAMER_AVAILABLE",
+    # NVDEC components
+    "NVDECWorkerManager",
+    "is_nvdec_available",
+    "get_available_gpu_count",
+    "NVDEC_AVAILABLE",
+]