def new_client(self, session_id, profiles):
'''
This method is invoked when the main window handles events from the controller (via the bridge).
Here we create a new subtree for the client and add it to the list of connections.
'''
# create the subtree containing profiles
newClientTreeItem = ClientTreeItem(session_id, self.rootItem)
for profile in profiles:
newConnProfileItem = ConnectionProfileTreeItem(newClientTreeItem, profile, RESULT_PENDING)
newClientTreeItem.appendChild(newConnProfileItem)
# insert the new node under the parent node
n = self.rootItem.childCount()
self.beginInsertRows(QtCore.QModelIndex(), n, n)
self.rootItem.appendChild(newClientTreeItem)
self.parents[session_id] = newClientTreeItem
self.endInsertRows()
def new_client(self, session_id, profiles):
'''
This method is invoked when the main window handles events from the controller (via the bridge).
Here we create a new subtree for the client and add it to the list of connections.
'''
# create the subtree containing profiles
newClientTreeItem = ClientTreeItem(session_id, self.rootItem)
for profile in profiles:
newConnProfileItem = ConnectionProfileTreeItem(
newClientTreeItem, profile, RESULT_PENDING)
newClientTreeItem.appendChild(newConnProfileItem)
# insert the new node under the parent node
n = self.rootItem.childCount()
self.beginInsertRows(QtCore.QModelIndex(), n, n)
self.rootItem.appendChild(newClientTreeItem)
self.parents[session_id] = newClientTreeItem
self.endInsertRows()
class ClientServerTestResultTreeTableModel(QtCore.QAbstractItemModel):
def __init__(self, parent=None):
super(ClientServerTestResultTreeTableModel, self).__init__(parent)
self.rootItem = ClientTreeItem('ALL', None) # XXX wtf
self.parents = {0: self.rootItem}
def columnCount(self, parent=None):
if parent and parent.isValid():
return parent.internalPointer().columnCount()
else:
return len(HORIZONTAL_HEADERS)
def data(self, index, role):
if not index.isValid():
return QtCore.QVariant()
item = index.internalPointer()
if role == QtCore.Qt.DisplayRole:
return item.data(index.column())
if role == QtCore.Qt.UserRole:
if item:
return item.result
return QtCore.QVariant()
def headerData(self, column, orientation, role):
if (orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole):
try:
return QtCore.QVariant(HORIZONTAL_HEADERS[column])
except IndexError:
pass
return QtCore.QVariant()
def index(self, row, column, parent):
if not self.hasIndex(row, column, parent):
return QtCore.QModelIndex()
if not parent.isValid():
parentItem = self.rootItem
else:
parentItem = parent.internalPointer()
childItem = parentItem.child(row)
if childItem:
return self.createIndex(row, column, childItem)
else:
return QtCore.QModelIndex()
def parent(self, index):
if not index.isValid():
return QtCore.QModelIndex()
childItem = index.internalPointer()
if not childItem:
return QtCore.QModelIndex()
parentItem = childItem.parent()
if parentItem == self.rootItem:
return QtCore.QModelIndex()
return self.createIndex(parentItem.row(), 0, parentItem)
def rowCount(self, parent=QtCore.QModelIndex()):
if parent.column() > 0:
return 0
if not parent.isValid():
p_Item = self.rootItem
else:
p_Item = parent.internalPointer()
return p_Item.childCount()
def new_client(self, session_id, profiles):
'''
This method is invoked when the main window handles events from the controller (via the bridge).
Here we create a new subtree for the client and add it to the list of connections.
'''
# create the subtree containing profiles
newClientTreeItem = ClientTreeItem(session_id, self.rootItem)
for profile in profiles:
newConnProfileItem = ConnectionProfileTreeItem(newClientTreeItem, profile, RESULT_PENDING)
newClientTreeItem.appendChild(newConnProfileItem)
# insert the new node under the parent node
n = self.rootItem.childCount()
self.beginInsertRows(QtCore.QModelIndex(), n, n)
self.rootItem.appendChild(newClientTreeItem)
self.parents[session_id] = newClientTreeItem
self.endInsertRows()
def new_conn_result(self, session_id, profile, result):
'''
This method is invoked when the main window handles events from the controller (via the bridge).
'''
if session_id in self.parents:
clientTreeItem = self.parents[session_id]
for profile_id in range(clientTreeItem.childCount()):
connProfileItem = clientTreeItem.child(profile_id)
if connProfileItem.profile == profile:
connProfileItem.result = result
# XXX need to call self.dataChanged() here?
return
logger.error('got "new_conn_result" event, but cannot find a row for it')
else:
logger.error('received "new_conn_result" event for session id "%s" but there is no subtree for it' % (session_id))
def client_done(self, session_id, results):
'''
This method is when the main window handles events from the controller (via bridge).
'''
print '*** client done ***'
class ClientServerTestResultTreeTableModel(QtCore.QAbstractItemModel):
def __init__(self, parent=None):
super(ClientServerTestResultTreeTableModel, self).__init__(parent)
self.rootItem = ClientTreeItem('ALL', None) # XXX wtf
self.parents = {0: self.rootItem}
def columnCount(self, parent=None):
if parent and parent.isValid():
return parent.internalPointer().columnCount()
else:
return len(HORIZONTAL_HEADERS)
def data(self, index, role):
if not index.isValid():
return QtCore.QVariant()
item = index.internalPointer()
if role == QtCore.Qt.DisplayRole:
return item.data(index.column())
if role == QtCore.Qt.UserRole:
if item:
return item.result
return QtCore.QVariant()
def headerData(self, column, orientation, role):
if (orientation == QtCore.Qt.Horizontal
and role == QtCore.Qt.DisplayRole):
try:
return QtCore.QVariant(HORIZONTAL_HEADERS[column])
except IndexError:
pass
return QtCore.QVariant()
def index(self, row, column, parent):
if not self.hasIndex(row, column, parent):
return QtCore.QModelIndex()
if not parent.isValid():
parentItem = self.rootItem
else:
parentItem = parent.internalPointer()
childItem = parentItem.child(row)
if childItem:
return self.createIndex(row, column, childItem)
else:
return QtCore.QModelIndex()
def parent(self, index):
if not index.isValid():
return QtCore.QModelIndex()
childItem = index.internalPointer()
if not childItem:
return QtCore.QModelIndex()
parentItem = childItem.parent()
if parentItem == self.rootItem:
return QtCore.QModelIndex()
return self.createIndex(parentItem.row(), 0, parentItem)
def rowCount(self, parent=QtCore.QModelIndex()):
if parent.column() > 0:
return 0
if not parent.isValid():
p_Item = self.rootItem
else:
p_Item = parent.internalPointer()
return p_Item.childCount()
def new_client(self, session_id, profiles):
'''
This method is invoked when the main window handles events from the controller (via the bridge).
Here we create a new subtree for the client and add it to the list of connections.
'''
# create the subtree containing profiles
newClientTreeItem = ClientTreeItem(session_id, self.rootItem)
for profile in profiles:
newConnProfileItem = ConnectionProfileTreeItem(
newClientTreeItem, profile, RESULT_PENDING)
newClientTreeItem.appendChild(newConnProfileItem)
# insert the new node under the parent node
n = self.rootItem.childCount()
self.beginInsertRows(QtCore.QModelIndex(), n, n)
self.rootItem.appendChild(newClientTreeItem)
self.parents[session_id] = newClientTreeItem
self.endInsertRows()
def new_conn_result(self, session_id, profile, result):
'''
This method is invoked when the main window handles events from the controller (via the bridge).
'''
if session_id in self.parents:
clientTreeItem = self.parents[session_id]
for profile_id in range(clientTreeItem.childCount()):
connProfileItem = clientTreeItem.child(profile_id)
if connProfileItem.profile == profile:
connProfileItem.result = result
# XXX need to call self.dataChanged() here?
return
logger.error(
'got "new_conn_result" event, but cannot find a row for it')
else:
logger.error(
'received "new_conn_result" event for session id "%s" but there is no subtree for it'
% (session_id))
def client_done(self, session_id, results):
'''
This method is when the main window handles events from the controller (via bridge).
'''
print '*** client done ***'