from meshroom.common import Property, Variant
from meshroom.core import attribute, desc

class TestAttribute(attribute.ChoiceParam):
    # custom implementation
    pass

class TestDescription(desc.ChoiceParam):
    def getInstanceType(self):
        return TestAttribute

    instanceType = Property(Variant, getInstanceType, constant=True)
    type = Property(str, lambda self: self.__class__.__base__.__name__, constant=True)  # allows the parameter to be interpreted as a ChoiceParam from a UI standpoint

__version__ = "1.0"

from meshroom.core import desc, custom

class TestAttributeType(desc.InputNode):
    documentation = """ """
    category = 'Test'

    # Inputs to the node
    inputs = [
        custom.TestDescription(
            name="testAttribute",
            label="Test Attribute",
            description="Custom attribute inheriting ChoiceParam.",
            value="a",
            values=["a", "b", "c", "d"],
            exclusive=True,
            uid=[0],
        )
    ]