|
|
from PyQt4 import QtCore
|
|
|
|
|
|
class person_class(object):
|
|
|
'''
|
|
|
a trivial custom data object
|
|
|
'''
|
|
|
def __init__(self, caracteristica, principal, descripcion):
|
|
|
self.caracteristica = caracteristica
|
|
|
self.principal = principal
|
|
|
self.descripcion = descripcion
|
|
|
|
|
|
def __repr__(self):
|
|
|
return "PERSON - %s %s"% (self.principal, self.caracteristica)
|
|
|
|
|
|
class TreeItem(object):
|
|
|
'''
|
|
|
a python object used to return row/column data, and keep note of
|
|
|
it's parents and/or children
|
|
|
'''
|
|
|
def __init__(self, person, header, parentItem):
|
|
|
self.person = person
|
|
|
self.parentItem = parentItem
|
|
|
self.header = header
|
|
|
self.childItems = []
|
|
|
|
|
|
def appendChild(self, item):
|
|
|
self.childItems.append(item)
|
|
|
|
|
|
def child(self, row):
|
|
|
return self.childItems[row]
|
|
|
|
|
|
def childCount(self):
|
|
|
return len(self.childItems)
|
|
|
|
|
|
def columnCount(self):
|
|
|
return 2
|
|
|
|
|
|
def data(self, column):
|
|
|
if self.person == None:
|
|
|
if column == 0:
|
|
|
return QtCore.QVariant(self.header)
|
|
|
if column == 1:
|
|
|
return QtCore.QVariant("")
|
|
|
else:
|
|
|
if column == 0:
|
|
|
return QtCore.QVariant(self.person.principal)
|
|
|
if column == 1:
|
|
|
return QtCore.QVariant(self.person.descripcion)
|
|
|
return QtCore.QVariant()
|
|
|
|
|
|
def parent(self):
|
|
|
return self.parentItem
|
|
|
|
|
|
def row(self):
|
|
|
if self.parentItem:
|
|
|
return self.parentItem.childItems.index(self)
|
|
|
return 0
|
|
|
|
