comfy-env 0.0.72__py3-none-any.whl → 0.0.74__py3-none-any.whl

comfy_env/install.py

@@ -55,7 +55,15 @@ def install(
             "Create comfy-env.toml to define dependencies."
         )
 
-    # Install node dependencies first
+    # Install apt packages first (Linux only)
+    if cfg.apt_packages:
+        _install_apt_packages(cfg.apt_packages, log, dry_run)
+
+    # Set persistent env vars (for OpenMP settings, etc.)
+    if cfg.env_vars:
+        _set_persistent_env_vars(cfg.env_vars, log, dry_run)
+
+    # Install node dependencies
     if cfg.node_reqs:
         _install_node_dependencies(cfg.node_reqs, node_dir, log, dry_run)
 
@@ -69,6 +77,162 @@ def install(
     return True
 
 
+def _install_apt_packages(
+    packages: List[str],
+    log: Callable[[str], None],
+    dry_run: bool,
+) -> None:
+    """Install apt packages (Linux only)."""
+    import os
+    import platform
+    import shutil
+    import subprocess
+
+    if platform.system() != "Linux":
+        log(f"[apt] Skipping apt packages (not Linux)")
+        return
+
+    log(f"\n[apt] Installing {len(packages)} system package(s):")
+    for pkg in packages:
+        log(f"  - {pkg}")
+
+    if dry_run:
+        log("  (dry run - no changes made)")
+        return
+
+    # Determine if we need sudo
+    is_root = os.geteuid() == 0
+    has_sudo = shutil.which("sudo") is not None
+    use_sudo = not is_root and has_sudo
+    prefix = ["sudo"] if use_sudo else []
+
+    if not is_root and not has_sudo:
+        log(f"[apt] Warning: No root access. Install manually:")
+        log(f"  sudo apt-get update && sudo apt-get install -y {' '.join(packages)}")
+        return
+
+    # Run apt-get update (suppress output, just show errors)
+    log("[apt] Updating package lists...")
+    result = subprocess.run(
+        prefix + ["apt-get", "update"],
+        capture_output=True,
+        text=True,
+    )
+    if result.returncode != 0:
+        log(f"[apt] Warning: apt-get update failed: {result.stderr.strip()}")
+
+    # Install each package individually (some may not exist on all distros)
+    log("[apt] Installing packages...")
+    installed = []
+    skipped = []
+    for pkg in packages:
+        result = subprocess.run(
+            prefix + ["apt-get", "install", "-y", pkg],
+            capture_output=True,
+            text=True,
+        )
+        if result.returncode == 0:
+            installed.append(pkg)
+            log(f"  [apt] Installed {pkg}")
+        else:
+            skipped.append(pkg)
+            log(f"  [apt] Skipped {pkg} (not available)")
+
+    if installed:
+        log(f"[apt] Installed {len(installed)} package(s)")
+    if skipped:
+        log(f"[apt] Skipped {len(skipped)} unavailable package(s)")
+
+
+def _set_persistent_env_vars(
+    env_vars: dict,
+    log: Callable[[str], None],
+    dry_run: bool,
+) -> None:
+    """Set env vars permanently (survives restarts)."""
+    import os
+    import platform
+    import subprocess
+    from pathlib import Path
+
+    if not env_vars:
+        return
+
+    system = platform.system()
+    log(f"\n[env] Setting {len(env_vars)} persistent environment variable(s)...")
+
+    for key, value in env_vars.items():
+        log(f"  - {key}={value}")
+
+    if dry_run:
+        log("  (dry run - no changes made)")
+        return
+
+    if system == "Windows":
+        # Windows: use setx (writes to registry)
+        for key, value in env_vars.items():
+            result = subprocess.run(
+                ["setx", key, value],
+                capture_output=True, text=True
+            )
+            if result.returncode == 0:
+                log(f"  [env] Set {key} (Windows registry)")
+            else:
+                log(f"  [env] Warning: Failed to set {key}: {result.stderr.strip()}")
+        log("[env] Restart terminal/ComfyUI for changes to take effect")
+
+    elif system == "Darwin":  # macOS
+        # macOS: launchctl for GUI apps + zshrc for terminal
+        for key, value in env_vars.items():
+            subprocess.run(["launchctl", "setenv", key, value], capture_output=True)
+            log(f"  [env] Set {key} (launchctl)")
+
+        # Also add to zshrc for terminal (zsh is default on macOS)
+        _add_to_shell_profile(env_vars, log)
+
+    else:  # Linux
+        _add_to_shell_profile(env_vars, log)
+
+
+def _add_to_shell_profile(
+    env_vars: dict,
+    log: Callable[[str], None],
+) -> None:
+    """Add env vars to shell profile (Linux/macOS)."""
+    import os
+    from pathlib import Path
+
+    # Determine shell profile
+    shell = os.environ.get("SHELL", "/bin/bash")
+    if "zsh" in shell:
+        rc_file = Path.home() / ".zshrc"
+    else:
+        rc_file = Path.home() / ".bashrc"
+
+    profile_file = Path.home() / ".comfy-env-profile"
+
+    # Write env vars to our dedicated file
+    with open(profile_file, "w") as f:
+        f.write("# Generated by comfy-env - do not edit manually\n")
+        for key, value in env_vars.items():
+            f.write(f'export {key}="{value}"\n')
+    log(f"  [env] Wrote {profile_file}")
+
+    # Add source line to shell rc (only once)
+    source_line = f'source "{profile_file}"'
+    existing = rc_file.read_text() if rc_file.exists() else ""
+
+    if source_line not in existing and str(profile_file) not in existing:
+        with open(rc_file, "a") as f:
+            f.write(f'\n# comfy-env environment variables\n')
+            f.write(f'{source_line}\n')
+        log(f"  [env] Added source line to {rc_file}")
+    else:
+        log(f"  [env] Already configured in {rc_file}")
+
+    log("[env] Restart terminal/ComfyUI for changes to take effect")
+
+
 def _install_node_dependencies(
     node_reqs: List[NodeReq],
     node_dir: Path,

{comfy_env-0.0.72.dist-info → comfy_env-0.0.74.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.72
+Version: 0.0.74
 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.0.72.dist-info → comfy_env-0.0.74.dist-info}/RECORD

@@ -1,7 +1,7 @@
 comfy_env/__init__.py,sha256=s0RkyKsBlDiSI4ZSwivtWLvYhc8DS0CUEWgFLVJbtLA,2176
 comfy_env/cli.py,sha256=ty4HYlzollCUCS0o6Sha6eczPAsW_gHRVgvck3IfA2w,12723
 comfy_env/errors.py,sha256=q-C3vyrPa_kk_Ao8l17mIGfJiG2IR0hCFV0GFcNLmcI,9924
-comfy_env/install.py,sha256=Fkjk3ci9_mAabUFqyuXShV9-Ry2iTX-JWAx9fnr5q_Q,5098
+comfy_env/install.py,sha256=N7eBj8wB2DrGepVYk-Hks2mSf6UuGzj34pfVLNYJgQ4,10357
 comfy_env/nodes.py,sha256=nBkG2pESeOt5kXSgTDIHAHaVdHK8-rEueR3BxXWn6TE,4354
 comfy_env/prestartup.py,sha256=rP0TtT9lxiU7Py27U79Ew7954VpHsYL6lfx4IpQY1Uc,3152
 comfy_env/config/__init__.py,sha256=4Guylkb-FV8QxhFwschzpzbr2eu8y-KNgNT3_JOm9jc,403
@@ -25,8 +25,8 @@ comfy_env/workers/base.py,sha256=4ZYTaQ4J0kBHCoO_OfZnsowm4rJCoqinZUaOtgkOPbw,230
 comfy_env/workers/mp.py,sha256=d6PFVrgqp3MK7Gkt08a8LQD_VSwHtoIcGx2Lovou3vM,25972
 comfy_env/workers/subprocess.py,sha256=UMhKhaJoSjv2-FWnwQq9TeMWtaDLIs3ajAFTq1ngdJw,57844
 comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaFLLq8,6409
-comfy_env-0.0.72.dist-info/METADATA,sha256=Z0aiB9t0vBMRBP71_5wt6AeOd13lW1gkX8IE0Mg-kg0,6946
-comfy_env-0.0.72.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.0.72.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.0.72.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.0.72.dist-info/RECORD,,
+comfy_env-0.0.74.dist-info/METADATA,sha256=rmFALPp_CocGIiAe5vF03FMHrJvFtrSt27UhPGVmmUc,6946
+comfy_env-0.0.74.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.74.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.74.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.74.dist-info/RECORD,,