def __init__(self, db):
super(TagModel, self).__init__(None, db)
self.setTable("tags")
self.select()
logging.debug("%d rows in TagModel" % self.rowCount())
self.tagParentsModel = TagParentsModel(db)
self.tree = MultiParentTree()
self.updateTree()
def updateTree(self, filterIDs=None):
self.tree = MultiParentTree()
self.setFilter(getFilterFromIDs(filterIDs, "ID"))
self.select()
logging.debug("FILTER=" + self.filter())
logging.debug("self.rowCount()=%d" % self.rowCount())
for i in range(self.rowCount()):
ID = self.record(i).value("ID")
tag = self.record(i).value("name")
logging.debug("ID=%d" % ID)
logging.debug("tag=%s" % tag)
self.tree.insertElement(ID, tag)
self.tagParentsModel.setFilter(getFilterFromIDs(filterIDs, "tagID"))
self.tagParentsModel.select()
for i in range(self.tagParentsModel.rowCount()):
parentID = self.tagParentsModel.record(i).value("parentTagID")
childID = self.tagParentsModel.record(i).value("tagID")
self.tree.setRelationship(parentID, childID)
class TagModel(QtSql.QSqlTableModel):
def __init__(self, db):
super(TagModel, self).__init__(None, db)
self.setTable("tags")
self.select()
logging.debug("%d rows in TagModel" % self.rowCount())
self.tagParentsModel = TagParentsModel(db)
self.tree = MultiParentTree()
self.updateTree()
def addTag(self, name):
self.setFilter("")
self.setSort(0, QtCore.Qt.SortOrder.AscendingOrder)
self.select()
record = QtSql.QSqlRecord()
record.append(QtSql.QSqlField("ID"))
record.append(QtSql.QSqlField("name"))
record.setValue(1, name)
if not self.insertRecord(self.rowCount() - 1, record):
raise "Tag could not be inserted."
newID = int(self.record(self.rowCount() - 1).value(0))
logging.debug("newID=%d" % newID)
return newID
def removeTag(self, ID):
self.setFilter("ID=%d" % ID)
self.select()
logging.debug("removeTag: self.rowCount()=%d" % self.rowCount())
if self.rowCount() == 1:
self.removeRow(0)
def getTagNameFromID(self, ID):
elem = self.tree.getElementByID(ID)
if elem is not None:
return elem.data
def hasID(self, ID):
return self.tree.hasID(ID)
def getChildIDs(self, ID, filterIDs=None):
#TODO: Implement filter
logging.debug("getChildIDs(%d)" % ID)
if not self.tree.hasID(ID):
return []
return self.tree.getElementByID(ID).getChildIDs()
def getAllChildIDs(self, ID):
childIDs = self.getChildIDs(ID)
allChildIDs = childIDs
for childID in childIDs:
allChildIDs.extend(self.getAllChildIDs(childID))
return allChildIDs
def fillTreeWidgetWithTags(self, treeWidget, checkable=False, IDstoCheck=set(), filterIDs=None):
self.updateTree(filterIDs)
# Insert root tags
rootTags = self.getRootTags(filterIDs)
logging.debug("rootTags=%s" % str(rootTags))
for rootID in sorted(rootTags.keys(), reverse=True):
newElem = QtGui.QTreeWidgetItem([rootTags[rootID], str(rootID)])
if checkable:
newElem.setFlags(newElem.flags() | QtCore.Qt.ItemIsUserCheckable)
newElem.setCheckState(0, QtCore.Qt.Checked)
if rootID not in IDstoCheck:
newElem.setCheckState(0, QtCore.Qt.Unchecked)
else:
self.expandDownToRoot(newElem)
treeWidget.insertTopLevelItem(0, newElem)
self.insertChildTags(rootID, newElem, checkable, IDstoCheck, filterIDs)
def insertChildTags(self, ID, treeWidgetItem, checkable, IDstoCheck, filterIDs):
if not self.hasID(ID):
return
childTags = self.getChildTags(ID, filterIDs)
logging.debug(self.getTagNameFromID(ID) + ": " + str(childTags))
for childID in childTags.keys():
if self.hasID(childID):
newElem = QtGui.QTreeWidgetItem(treeWidgetItem, [self.getTagNameFromID(childID), str(childID)])
if checkable:
newElem.setFlags(newElem.flags() | QtCore.Qt.ItemIsUserCheckable)
newElem.setCheckState(0, QtCore.Qt.Checked)
if childID not in IDstoCheck:
newElem.setCheckState(0, QtCore.Qt.Unchecked)
else:
self.expandDownToRoot(newElem)
if self.hasChildTags(childID):
self.insertChildTags(childID, newElem, checkable, IDstoCheck, filterIDs)
def expandDownToRoot(self, treeWidgetItem):
treeWidgetItem.setExpanded(True)
if treeWidgetItem.parent() is not None:
self.expandDownToRoot(treeWidgetItem.parent())
def getRootTags(self, filterIDs):
#TODO: Implement filter
return self.tree.getRootElementsDict()
def getChildTags(self, ID, filterIDs=None):
#TODO: Implement filter
return self.tree.getElementByID(ID).getChildDict()
def hasChildTags(self, ID):
return self.tree.getElementByID(ID).hasParents()
def getIDsFilteredByName(self, name):
self.setFilter('name LIKE "%' + name + '%"')
self.select()
logging.debug("Found %d tags with name %s" % (self.rowCount(), name))
IDs = [self.record(i).value("ID") for i in range(self.rowCount())]
return IDs
def getParentIDs(self, ID):
if self.tree.getElementByID(ID) is not None:
return self.tree.getElementByID(ID).getParentIDs()
else:
return []
def getParentIDsDownToRoot(self, ID):
allParentIDs = []
if self.tree.getElementByID(ID) is not None:
allParentIDs = self.tree.getElementByID(ID).getParentIDs()
for parentID in allParentIDs:
allParentIDs.extend(self.getParentIDsDownToRoot(parentID))
return allParentIDs
def getParentsTags(self, tagID):
self.tree.getElementByID(tagID).getParentIDs()
def updateTree(self, filterIDs=None):
self.tree = MultiParentTree()
self.setFilter(getFilterFromIDs(filterIDs, "ID"))
self.select()
logging.debug("FILTER=" + self.filter())
logging.debug("self.rowCount()=%d" % self.rowCount())
for i in range(self.rowCount()):
ID = self.record(i).value("ID")
tag = self.record(i).value("name")
logging.debug("ID=%d" % ID)
logging.debug("tag=%s" % tag)
self.tree.insertElement(ID, tag)
self.tagParentsModel.setFilter(getFilterFromIDs(filterIDs, "tagID"))
self.tagParentsModel.select()
for i in range(self.tagParentsModel.rowCount()):
parentID = self.tagParentsModel.record(i).value("parentTagID")
childID = self.tagParentsModel.record(i).value("tagID")
self.tree.setRelationship(parentID, childID)