from adorn.orchestrator.base import Base
from adorn.params import Params
from adorn.unit.complex import Complex
from adorn.unit.constructor_value import ConstructorValue
from adorn.unit.parameter_value import ParameterValue
from adorn.unit.python import Python


@Complex.root()
class Example(Complex):
   pass

@Example.register(None)
class Parent(Example):
    def __init__(self, parent_value: str) -> None:
        super().__init__()
        self.parent_value = parent_value


@Parent.register("child")
class Child(Parent):
    def __init__(self, child_value: int, **kwargs) -> None:
        super().__init__(**kwargs)
        self.child_value = child_value


base = Base(
    [
        ConstructorValue(),
        ParameterValue(),
        Example(),
        Python()
    ]
)

params = Params(
        {
            "type": "child",
            "child_value": 0,
            "parent_value": "abc"
        }
)

# well specified configuration
# we can type check from any level in the
# class hierarchy
assert base.type_check(Example, params) is None
assert base.type_check(Parent, params) is None
assert base.type_check(Child, params) is None

# instantiate
# we can instantiate from any level in the
# class hierarchy
example_obj = base.from_obj(
    Example,
    params
)

assert isinstance(example_obj, Child)


parent_obj = base.from_obj(
    Parent,
    params
)

assert isinstance(parent_obj, Child)


child_obj = base.from_obj(
    Child,
    params
)

assert isinstance(child_obj, Child)