comfy-env 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

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()

{comfy_env-0.1.4.dist-info → comfy_env-0.1.6.dist-info}/METADATA

@@ -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.dist-info → comfy_env-0.1.6.dist-info}/RECORD

@@ -22,11 +22,11 @@ comfy_env/templates/comfy-env-instructions.txt,sha256=ve1RAthW7ouumU9h6DM7mIRX1M
 comfy_env/templates/comfy-env.toml,sha256=ROIqi4BlPL1MEdL1VgebfTHpdwPNYGHwWeigI9Kw-1I,4831
 comfy_env/workers/__init__.py,sha256=TMVG55d2XLP1mJ3x1d16H0SBDJZtk2kMC5P4HLk9TrA,1073
 comfy_env/workers/base.py,sha256=4ZYTaQ4J0kBHCoO_OfZnsowm4rJCoqinZUaOtgkOPbw,2307
-comfy_env/workers/mp.py,sha256=lKCdFJ0-OE5-ItJxcfXxoBTiESGU-5mUcTerU1WnUbk,27404
+comfy_env/workers/mp.py,sha256=xqSLnpR5zSVdZKt10nM6LmNyJ2SmlSr1yQ5I9HX5OJ4,29514
 comfy_env/workers/subprocess.py,sha256=UMhKhaJoSjv2-FWnwQq9TeMWtaDLIs3ajAFTq1ngdJw,57844
 comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaFLLq8,6409
-comfy_env-0.1.4.dist-info/METADATA,sha256=pkg8IsTN2rxSPTDlg3WlMuiRJ74VcQ3O4qvFaa5FC8E,6945
-comfy_env-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.1.4.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.1.4.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.1.4.dist-info/RECORD,,
+comfy_env-0.1.6.dist-info/METADATA,sha256=IT7HN3q_Q7hdsZqcGRbgtoeKBTN6z_1Gzr8KhVtPpg0,6945
+comfy_env-0.1.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.1.6.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.1.6.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.1.6.dist-info/RECORD,,