Vince Liem
b2e7c7ffeb
Migrated TagInjectorSingle, TagInjectorDuo, TagInjector, TagInjectorLarge, RandomFloat, RandomInt, and NextSeed from legacy node implementation to new ComfyExtension API with io.ComfyNode base class. Replaced INPUT_TYPES, RETURN_TYPES, RETURN_NAMES, FUNCTION, and CATEGORY class attributes with define_schema() method using io.Schema. Converted legacy methods to execute() with type-annotated parameters and io.NodeOutput return type
48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
from typing_extensions import override
|
|
|
|
from comfy_api.latest import ComfyExtension, io
|
|
|
|
from custom_nodes.comfyui_gr85.nodes.resolution.image_dimension_resizer import ImageDimensionResizer
|
|
from custom_nodes.comfyui_gr85.nodes.resolution.image_sizer import ImageSizer
|
|
from custom_nodes.comfyui_gr85.nodes.resolution.image_sizer_all import ImageSizerAll
|
|
from custom_nodes.comfyui_gr85.nodes.resolution.random_ratio import RandomRatio
|
|
from custom_nodes.comfyui_gr85.nodes.prompt_selection.seed_based_output_selector import SeedBasedOutputSelector
|
|
from custom_nodes.comfyui_gr85.nodes.prompt_wildcards.simple_wildcard_picker import SimpleWildcardPicker
|
|
from custom_nodes.comfyui_gr85.nodes.prompt_tags.tag_injector import TagInjector
|
|
from custom_nodes.comfyui_gr85.nodes.prompt_tags.tag_injector import TagInjectorSingle
|
|
from custom_nodes.comfyui_gr85.nodes.prompt_tags.tag_injector import TagInjectorDuo
|
|
from custom_nodes.comfyui_gr85.nodes.prompt_tags.tag_injector_large import TagInjectorLarge
|
|
from custom_nodes.comfyui_gr85.nodes.random_seed.next_seed import NextSeed
|
|
from custom_nodes.comfyui_gr85.nodes.random_numbers.random_float import RandomFloat
|
|
from custom_nodes.comfyui_gr85.nodes.random_numbers.random_int import RandomInt
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
|
|
class GR85Extension(ComfyExtension):
|
|
@override
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
SimpleWildcardPicker,
|
|
SeedBasedOutputSelector,
|
|
TagInjectorSingle,
|
|
TagInjectorDuo,
|
|
TagInjector,
|
|
TagInjectorLarge,
|
|
RandomFloat,
|
|
RandomInt,
|
|
NextSeed,
|
|
ImageDimensionResizer,
|
|
ImageSizerAll,
|
|
ImageSizer,
|
|
RandomRatio,
|
|
]
|
|
|
|
|
|
async def comfy_entrypoint() -> GR85Extension:
|
|
return GR85Extension()
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|