import pillarsdk


class SpecialFolderNode(pillarsdk.Node):
    NODE_TYPE = "SPECIAL"


class UpNode(SpecialFolderNode):
    NODE_TYPE = "UP"

    def __init__(self):
        super().__init__()
        self["_id"] = "UP"
        self["node_type"] = self.NODE_TYPE


class ProjectNode(SpecialFolderNode):
    NODE_TYPE = "PROJECT"

    def __init__(self, project):
        super().__init__()

        assert isinstance(
            project, pillarsdk.Project
        ), "wrong type for project: %r" % type(project)

        self.merge(project.to_dict())
        self["node_type"] = self.NODE_TYPE