comfy-env 0.1.4__tar.gz → 0.1.6__tar.gz

{comfy_env-0.1.4 → comfy_env-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.1.4
+Version: 0.1.6
 Summary: Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation
 Project-URL: Homepage, https://github.com/PozzettiAndrea/comfy-env
 Project-URL: Repository, https://github.com/PozzettiAndrea/comfy-env

{comfy_env-0.1.4 → comfy_env-0.1.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "comfy-env"
-version = "0.1.4"
+version = "0.1.6"
 description = "Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation"
 readme = "README.md"
 license = {text = "MIT"}

{comfy_env-0.1.4 → comfy_env-0.1.6}/src/comfy_env/workers/mp.py

@@ -40,7 +40,55 @@ _SHUTDOWN = object()
 _CALL_METHOD = "call_method"
 
 
-def _worker_loop(queue_in, queue_out, sys_path_additions=None, lib_path=None, env_vars=None):
+def _dump_worker_env(worker_name: str = "unknown", print_to_terminal: bool = False):
+    """Dump worker environment to .comfy-env/logs/ (always) and optionally print."""
+    import json
+    import os
+    import platform
+    import sys
+    from datetime import datetime
+    from pathlib import Path
+
+    log_dir = Path.cwd() / ".comfy-env" / "logs"
+    log_dir.mkdir(parents=True, exist_ok=True)
+
+    debug_info = {
+        "timestamp": datetime.now().isoformat(),
+        "worker_name": worker_name,
+        "pid": os.getpid(),
+        "cwd": os.getcwd(),
+        "python": {
+            "executable": sys.executable,
+            "version": sys.version,
+            "prefix": sys.prefix,
+        },
+        "platform": {
+            "system": platform.system(),
+            "machine": platform.machine(),
+            "release": platform.release(),
+        },
+        "env_vars": dict(os.environ),
+        "sys_path": sys.path,
+        "modules_loaded": sorted(sys.modules.keys()),
+    }
+
+    log_file = log_dir / f"worker_{worker_name}_{os.getpid()}.json"
+    log_file.write_text(json.dumps(debug_info, indent=2, default=str))
+
+    if print_to_terminal:
+        print(f"[comfy-env] === WORKER ENV DEBUG: {worker_name} ===")
+        print(f"[comfy-env] Python: {sys.executable}")
+        print(f"[comfy-env] Version: {sys.version.split()[0]}")
+        print(f"[comfy-env] PID: {os.getpid()}, CWD: {os.getcwd()}")
+        for var in ['PATH', 'LD_LIBRARY_PATH', 'PYTHONPATH', 'OMP_NUM_THREADS', 'KMP_DUPLICATE_LIB_OK']:
+            val = os.environ.get(var, '<unset>')
+            if len(val) > 100:
+                val = val[:100] + '...'
+            print(f"[comfy-env] {var}={val}")
+        print(f"[comfy-env] Env dumped to: {log_file}")
+
+
+def _worker_loop(queue_in, queue_out, sys_path_additions=None, lib_path=None, env_vars=None, worker_name=None):
     """
     Worker process main loop.
 
@@ -57,6 +105,7 @@ def _worker_loop(queue_in, queue_out, sys_path_additions=None, lib_path=None, en
         sys_path_additions: Paths to add to sys.path
         lib_path: Path to add to LD_LIBRARY_PATH (for conda libraries)
         env_vars: Environment variables to set (from comfy-env.toml)
+        worker_name: Name of the worker (for logging)
     """
     import os
     import sys
@@ -69,6 +118,10 @@ def _worker_loop(queue_in, queue_out, sys_path_additions=None, lib_path=None, en
     # Set worker mode env var
     os.environ["COMFYUI_ISOLATION_WORKER"] = "1"
 
+    # Always dump env to file, print to terminal if debug enabled
+    print_debug = os.environ.get("COMFY_ENV_DEBUG", "").lower() in ("1", "true", "yes")
+    _dump_worker_env(worker_name or "unknown", print_to_terminal=print_debug)
+
     # DLL/library isolation - match SubprocessWorker's isolation level
     # Filter out conflicting paths from conda/mamba/etc and use proper DLL registration
     path_sep = ";" if sys.platform == "win32" else ":"
@@ -511,7 +564,7 @@ class MPWorker(Worker):
             self._queue_out = ctx.Queue()
             self._process = ctx.Process(
                 target=_worker_loop,
-                args=(self._queue_in, self._queue_out, self._sys_path, self._lib_path, self._env_vars),
+                args=(self._queue_in, self._queue_out, self._sys_path, self._lib_path, self._env_vars, self.name),
                 daemon=True,
             )
             self._process.start()