|
|
| def MakeSmartType(t):
|
| if isinstance(t, str):
|
| return SmartType(t)
|
| return t
|
|
|
| class SmartType(str):
|
| def __ne__(self, other):
|
| if self == "*" or other == "*":
|
| return False
|
| selfset = set(self.split(','))
|
| otherset = set(other.split(','))
|
| return not selfset.issubset(otherset)
|
|
|
| def VariantSupport():
|
| def decorator(cls):
|
| if hasattr(cls, "INPUT_TYPES"):
|
| old_input_types = getattr(cls, "INPUT_TYPES")
|
| def new_input_types(*args, **kwargs):
|
| types = old_input_types(*args, **kwargs)
|
| for category in ["required", "optional"]:
|
| if category not in types:
|
| continue
|
| for key, value in types[category].items():
|
| if isinstance(value, tuple):
|
| types[category][key] = (MakeSmartType(value[0]),) + value[1:]
|
| return types
|
| setattr(cls, "INPUT_TYPES", new_input_types)
|
| if hasattr(cls, "RETURN_TYPES"):
|
| old_return_types = cls.RETURN_TYPES
|
| setattr(cls, "RETURN_TYPES", tuple(MakeSmartType(x) for x in old_return_types))
|
| if hasattr(cls, "VALIDATE_INPUTS"):
|
|
|
| raise NotImplementedError("VariantSupport does not support VALIDATE_INPUTS yet")
|
| else:
|
| def validate_inputs(input_types):
|
| inputs = cls.INPUT_TYPES()
|
| for key, value in input_types.items():
|
| if isinstance(value, SmartType):
|
| continue
|
| if "required" in inputs and key in inputs["required"]:
|
| expected_type = inputs["required"][key][0]
|
| elif "optional" in inputs and key in inputs["optional"]:
|
| expected_type = inputs["optional"][key][0]
|
| else:
|
| expected_type = None
|
| if expected_type is not None and MakeSmartType(value) != expected_type:
|
| return f"Invalid type of {key}: {value} (expected {expected_type})"
|
| return True
|
| setattr(cls, "VALIDATE_INPUTS", validate_inputs)
|
| return cls
|
| return decorator
|
|
|
|
|
