99 lines
2.7 KiB
Python
99 lines
2.7 KiB
Python
"""
|
|
Converts int / float / boolean
|
|
Note that it depends on the order of the conversion
|
|
"""
|
|
from .autonode import validate, node_wrapper, get_node_names_mappings, anytype
|
|
classes = []
|
|
node = node_wrapper(classes)
|
|
|
|
conversion_operators = {
|
|
"Int" : int,
|
|
"Float" : float,
|
|
"Boolean" : bool,
|
|
"String" : str,
|
|
"Dict": dict,
|
|
"List": list,
|
|
"Tuple": tuple,
|
|
"Set": set,
|
|
}
|
|
def create_class(type_to):
|
|
class_name = "ConvertAny2{}".format(type_to)
|
|
class CustomClass:
|
|
FUNCTION = "convert"
|
|
RETURN_TYPES = (type_to.upper(),)
|
|
CATEGORY = "Conversion"
|
|
custom_name = "Convert to {}".format(type_to)
|
|
@staticmethod
|
|
def convert(input1):
|
|
return (conversion_operators[type_to](input1),)
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"input1": (anytype, {"default": 0.0}),
|
|
}
|
|
}
|
|
CustomClass.__name__ = class_name
|
|
node(CustomClass)
|
|
return CustomClass
|
|
|
|
|
|
@node
|
|
class StringListToCombo:
|
|
"""
|
|
Converts raw string, separates with separator, then picks the first element or the element at index
|
|
"""
|
|
RETURN_TYPES = (anytype,)
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"string": ("STRING", {"default": ""}),
|
|
"separator": ("STRING", {"default": "$"}),
|
|
},
|
|
"optional": {
|
|
"index": ("INT", {"default": 0}),
|
|
}
|
|
}
|
|
FUNCTION = "stringListToCombo"
|
|
CATEGORY = "Logic Gates"
|
|
custom_name = "String List to Combo"
|
|
def stringListToCombo(self, string, separator, index = 0):
|
|
if isinstance(string, (float, int, bool)):
|
|
return (string,)
|
|
if separator == "" or separator == None or separator not in string:
|
|
return (string,)
|
|
# check length
|
|
splitted = string.split(separator)
|
|
if index >= len(splitted):
|
|
return (splitted[-1],)
|
|
return (splitted[index],)
|
|
|
|
@node
|
|
class ConvertComboToString:
|
|
"""
|
|
Converts raw list to string, separated with separator
|
|
"""
|
|
RETURN_TYPES = ("STRING",)
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"combo": (anytype, {"default": []}),
|
|
"separator": ("STRING", {"default": "$"}),
|
|
}
|
|
}
|
|
FUNCTION = "convertComboToString"
|
|
CATEGORY = "Logic Gates"
|
|
custom_name = "Convert Combo to String"
|
|
def convertComboToString(self, combo, separator):
|
|
if isinstance(combo, (list, tuple)):
|
|
return (separator.join(str(item) for item in combo),)
|
|
return (str(combo),)
|
|
|
|
for type_to in conversion_operators:
|
|
create_class(type_to)
|
|
|
|
CLASS_MAPPINGS, CLASS_NAMES = get_node_names_mappings(classes)
|
|
validate(classes)
|