Skip to content
32 changes: 32 additions & 0 deletions comfy_cli/registry/config_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,6 +110,9 @@
project = document.get("project", tomlkit.table())
urls = project.get("urls", tomlkit.table())
urls["Repository"] = git_remote_url
urls["Documentation"] = git_remote_url + "/wiki"
urls["Bug Tracker"] = git_remote_url + "/issues"

Check warning on line 114 in comfy_cli/registry/config_parser.py

View check run for this annotation

Codecov / codecov/patch

comfy_cli/registry/config_parser.py#L113-L114

Added lines #L113 - L114 were not covered by tests

project["urls"] = urls
project["name"] = sanitize_node_name(repo_name)
project["description"] = ""
Expand All @@ -119,10 +122,39 @@
license_table = tomlkit.inline_table()
license_table["file"] = "LICENSE"
project["license"] = license_table

# Classifiers
# [project]
# classifiers = [
# # For OS-independent nodes (works on all operating systems)
# "Operating System :: OS Independent",

# # OR for OS-specific nodes, specify the supported systems:
# "Operating System :: Microsoft :: Windows", # Windows specific
# "Operating System :: POSIX :: Linux", # Linux specific
# "Operating System :: MacOS", # macOS specific

# # GPU Accelerator support
# "Environment :: GPU :: NVIDIA CUDA", # NVIDIA CUDA support
# "Environment :: GPU :: AMD ROCm", # AMD ROCm support
# "Environment :: GPU :: Intel Arc", # Intel Arc support
# "Environment :: NPU :: Huawei Ascend", # Huawei Ascend support
# "Environment :: GPU :: Apple Metal", # Apple Metal support
# ]
project["classifiers"] = tomlkit.array(

Check warning on line 144 in comfy_cli/registry/config_parser.py

View check run for this annotation

Codecov / codecov/patch

comfy_cli/registry/config_parser.py#L144

Added line #L144 was not covered by tests
"Operating System :: OS Independent",
"Environment :: GPU :: NVIDIA CUDA",
"Environment :: GPU :: AMD ROCm",
"Environment :: GPU :: Intel Arc",
"Environment :: NPU :: Huawei Ascend",
"Environment :: GPU :: Apple Metal",
)

tool = document.get("tool", tomlkit.table())
comfy = tool.get("comfy", tomlkit.table())
comfy["DisplayName"] = repo_name
comfy["requires-comfyui"] = ">=1.0.0" # ComfyUI version compatibility

Check warning on line 156 in comfy_cli/registry/config_parser.py

View check run for this annotation

Codecov / codecov/patch

comfy_cli/registry/config_parser.py#L156

Added line #L156 was not covered by tests

tool["comfy"] = comfy
document["tool"] = tool

Expand Down
Loading