Source

gtkarchitect-main / gtkarchitect / view / projecttree.py

Full commit

from kiwi.ui.objectlist import ObjectTree, Column

from ..model.entity import get_entity_type
from .base import Slave


class EntityColumn(Column):

    def get_attribute(self, entity, *args):
        return '%s %s' % (entity.tag, get_entity_type(entity))

class ProjectTree(Slave):

    def create_ui(self):
        self.tree = ObjectTree(EntityColumn('name'))
        t = self.get_toplevel()
        t.add(self.tree)

    def set_project(self, project):
        self.project = project
        self.add_entity(self.project.tree)

    def add_entity(self, entity, parent=None):
        self.tree.append(parent, entity)
        for child in entity:
            self.add_entity(child, entity)