comfy-env 0.0.45__tar.gz → 0.0.47__tar.gz

{comfy_env-0.0.45 → comfy_env-0.0.47}/PKG-INFO

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

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

{comfy_env-0.0.45 → comfy_env-0.0.47}/src/comfy_env/cli.py

@@ -42,6 +42,18 @@ def main(args: Optional[List[str]] = None) -> int:
 
     subparsers = parser.add_subparsers(dest="command", help="Available commands")
 
+    # init command
+    init_parser = subparsers.add_parser(
+        "init",
+        help="Create a default comfy-env.toml config file",
+        description="Initialize a new comfy-env.toml configuration file in the current directory",
+    )
+    init_parser.add_argument(
+        "--force", "-f",
+        action="store_true",
+        help="Overwrite existing config file",
+    )
+
     # install command
     install_parser = subparsers.add_parser(
         "install",
@@ -134,7 +146,9 @@ def main(args: Optional[List[str]] = None) -> int:
         return 0
 
     try:
-        if parsed.command == "install":
+        if parsed.command == "init":
+            return cmd_init(parsed)
+        elif parsed.command == "install":
             return cmd_install(parsed)
         elif parsed.command == "info":
             return cmd_info(parsed)
@@ -155,6 +169,42 @@ def main(args: Optional[List[str]] = None) -> int:
         return 1
 
 
+DEFAULT_CONFIG = """\
+# comfy-env.toml - Environment configuration for ComfyUI custom nodes
+# Documentation: https://github.com/PozzettiAndrea/comfy-env
+
+[system]
+# System packages required (apt on Linux, brew on macOS)
+linux = []
+
+[environment]
+python = "3.11"
+cuda_version = "auto"
+pytorch_version = "auto"
+
+[environment.cuda]
+# CUDA packages from comfy-env registry
+# Example: nvdiffrast = "0.4.0"
+
+[environment.packages]
+requirements = []
+"""
+
+
+def cmd_init(args) -> int:
+    """Handle init command."""
+    config_path = Path.cwd() / "comfy-env.toml"
+
+    if config_path.exists() and not args.force:
+        print(f"Config file already exists: {config_path}", file=sys.stderr)
+        print("Use --force to overwrite", file=sys.stderr)
+        return 1
+
+    config_path.write_text(DEFAULT_CONFIG)
+    print(f"Created {config_path}")
+    return 0
+
+
 def cmd_install(args) -> int:
     """Handle install command."""
     from .install import install

{comfy_env-0.0.45 → comfy_env-0.0.47}/src/comfy_env/pixi.py

@@ -224,6 +224,21 @@ def create_pixi_toml(
 
     # PyPI dependencies section
     pypi_deps = []
+    special_deps = {}  # For dependencies that need special syntax (path, etc.)
+
+    # Always include comfy-env for worker support
+    # Use local wheel if available (for ct test --local)
+    local_wheels_dir = os.environ.get("COMFY_LOCAL_WHEELS")
+    if local_wheels_dir:
+        local_wheels = list(Path(local_wheels_dir).glob("comfy_env-*.whl"))
+        if local_wheels:
+            # Use relative path from node_dir
+            rel_path = os.path.relpath(local_wheels[0], node_dir)
+            special_deps["comfy-env"] = f'{{ path = "{rel_path}" }}'
+        else:
+            pypi_deps.append("comfy-env")
+    else:
+        pypi_deps.append("comfy-env")
 
     # Add regular requirements
     if env_config.requirements:
@@ -241,8 +256,13 @@ def create_pixi_toml(
     elif sys.platform == "win32" and env_config.windows_requirements:
         pypi_deps.extend(env_config.windows_requirements)
 
-    if pypi_deps:
+    if pypi_deps or special_deps:
         lines.append("[pypi-dependencies]")
+
+        # Add special dependencies first (path-based, etc.)
+        for name, value in special_deps.items():
+            lines.append(f'{name} = {value}')
+
         for dep in pypi_deps:
             # Parse pip requirement format to pixi format
             dep_clean = dep.strip()