Spaces:
Runtime error
Runtime error
File size: 1,320 Bytes
c19ca42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
from typing import Any, Dict, Generic, List, NewType, Optional, TypeVar, Union
from base_types import InputId
from nodes.base_input import BaseInput
T = TypeVar("T")
GroupId = NewType("GroupId", int)
class GroupInfo:
def __init__(
self,
group_id: GroupId,
kind: str,
options: Optional[Dict[str, Any]] = None,
) -> None:
self.id: GroupId = group_id
self.kind: str = kind
self.options: Dict[str, Any] = {} if options is None else options
class Group(Generic[T]):
def __init__(self, info: GroupInfo, items: List[T]) -> None:
self.info: GroupInfo = info
self.items: List[T] = items
def toDict(self):
return {
"id": self.info.id,
"kind": self.info.kind,
"options": self.info.options,
"items": [i.toDict() if isinstance(i, Group) else i for i in self.items],
}
NestedGroup = Group[Union[BaseInput, "NestedGroup"]]
NestedIdGroup = Group[Union[InputId, "NestedIdGroup"]]
# pylint: disable-next=redefined-builtin
def group(kind: str, options: Optional[Dict[str, Any]] = None, id: int = -1):
info = GroupInfo(GroupId(id), kind, options)
def ret(*items: Union[BaseInput, NestedGroup]) -> NestedGroup:
return Group(info, list(items))
return ret
|