comfy-env 0.0.8__tar.gz → 0.0.10__tar.gz

{comfy_env-0.0.8 → comfy_env-0.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.8
+Version: 0.0.10
 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.8 → comfy_env-0.0.10}/pyproject.toml

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

{comfy_env-0.0.8 → comfy_env-0.0.10}/src/comfy_env/env/config.py

@@ -98,6 +98,10 @@ class IsolatedEnv:
     cuda: Optional[str] = None
     requirements: list[str] = field(default_factory=list)
     no_deps_requirements: list[str] = field(default_factory=list)  # Install with --no-deps
+    # Platform-specific requirements (merged at install time)
+    windows_requirements: list[str] = field(default_factory=list)
+    linux_requirements: list[str] = field(default_factory=list)
+    darwin_requirements: list[str] = field(default_factory=list)
     requirements_file: Optional[Path] = None
     wheel_sources: list[str] = field(default_factory=list)
     index_urls: list[str] = field(default_factory=list)

{comfy_env-0.0.8 → comfy_env-0.0.10}/src/comfy_env/env/config_file.py

@@ -529,6 +529,30 @@ def _parse_single_env(name: str, env_data: Dict[str, Any], base_dir: Path) -> Is
     elif isinstance(packages_section, list):
         requirements = packages_section
 
+    # Parse platform-specific packages [envname.packages.windows], etc.
+    windows_reqs = []
+    linux_reqs = []
+    darwin_reqs = []
+
+    if isinstance(packages_section, dict):
+        win_section = packages_section.get("windows", {})
+        if isinstance(win_section, dict):
+            windows_reqs = win_section.get("requirements", [])
+        elif isinstance(win_section, list):
+            windows_reqs = win_section
+
+        linux_section = packages_section.get("linux", {})
+        if isinstance(linux_section, dict):
+            linux_reqs = linux_section.get("requirements", [])
+        elif isinstance(linux_section, list):
+            linux_reqs = linux_section
+
+        darwin_section = packages_section.get("darwin", {})
+        if isinstance(darwin_section, dict):
+            darwin_reqs = darwin_section.get("requirements", [])
+        elif isinstance(darwin_section, list):
+            darwin_reqs = darwin_section
+
     return IsolatedEnv(
         name=name,
         python=python,
@@ -536,6 +560,9 @@ def _parse_single_env(name: str, env_data: Dict[str, Any], base_dir: Path) -> Is
         pytorch_version=pytorch,
         requirements=requirements,
         no_deps_requirements=no_deps_requirements,
+        windows_requirements=windows_reqs,
+        linux_requirements=linux_reqs,
+        darwin_requirements=darwin_reqs,
     )
 
 

{comfy_env-0.0.8 → comfy_env-0.0.10}/src/comfy_env/env/manager.py

@@ -6,6 +6,7 @@ Uses uv for fast environment creation and package installation.
 
 import subprocess
 import shutil
+import sys
 from pathlib import Path
 from typing import Optional, Callable
 
@@ -263,13 +264,29 @@ class IsolatedEnvManager:
         """
         python_exe = self.get_python(env)
 
-        if not env.requirements and not env.requirements_file:
+        # Merge platform-specific requirements
+        if sys.platform == 'win32':
+            platform_reqs = env.windows_requirements
+            platform_name = 'Windows'
+        elif sys.platform == 'darwin':
+            platform_reqs = env.darwin_requirements
+            platform_name = 'macOS'
+        else:
+            platform_reqs = env.linux_requirements
+            platform_name = 'Linux'
+
+        all_requirements = list(env.requirements) + list(platform_reqs)
+
+        if platform_reqs:
+            self.log(f"Including {len(platform_reqs)} {platform_name}-specific packages")
+
+        if not all_requirements and not env.requirements_file:
             self.log("No requirements to install")
             return
 
         # Validate requirements for security
-        if env.requirements:
-            validate_dependencies(env.requirements)
+        if all_requirements:
+            validate_dependencies(all_requirements)
 
         # Validate wheel sources
         for wheel_source in env.wheel_sources:
@@ -311,11 +328,11 @@ class IsolatedEnvManager:
             self.log(f"Installing {len(env.no_deps_requirements)} CUDA packages")
             self._install_cuda_packages(env, pip_args)
 
-        # Install individual requirements
-        if env.requirements:
-            self.log(f"Installing {len(env.requirements)} packages")
+        # Install individual requirements (including platform-specific)
+        if all_requirements:
+            self.log(f"Installing {len(all_requirements)} packages")
             result = subprocess.run(
-                pip_args + env.requirements,
+                pip_args + all_requirements,
                 capture_output=True,
                 text=True,
             )