mirror of
https://github.com/olivierkes/manuskript.git
synced 2024-05-17 11:22:28 +12:00
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
#!/usr/bin/env python
|
|
# --!-- coding: utf8 --!--
|
|
from PyQt5.QtCore import QModelIndex, QSortFilterProxyModel
|
|
|
|
|
|
class characterPOVModel(QSortFilterProxyModel):
|
|
|
|
def __init__(self, sourceModel, parent=None):
|
|
QSortFilterProxyModel.__init__(self, parent)
|
|
|
|
self.setSourceModel(sourceModel)
|
|
|
|
if sourceModel:
|
|
sourceModel.dataChanged.connect(self.sourceDataChanged)
|
|
|
|
def filterAcceptsRow(self, sourceRow, sourceParent):
|
|
return self.sourceModel().pov(sourceRow)
|
|
|
|
def rowToSource(self, row):
|
|
index = self.index(row, 0)
|
|
sourceIndex = self.mapToSource(index)
|
|
return sourceIndex.row()
|
|
|
|
def sourceDataChanged(self, topLeft, bottomRight):
|
|
self.invalidateFilter()
|
|
|
|
###############################################################################
|
|
# CHARACTER QUERIES
|
|
###############################################################################
|
|
|
|
def character(self, row):
|
|
return self.sourceModel().character(self.rowToSource(row))
|
|
|
|
def name(self, row):
|
|
return self.sourceModel().name(self.rowToSource(row))
|
|
|
|
def icon(self, row):
|
|
return self.sourceModel().icon(self.rowToSource(row))
|
|
|
|
def ID(self, row):
|
|
return self.sourceModel().ID(self.rowToSource(row))
|
|
|
|
def importance(self, row):
|
|
return self.sourceModel().importance(self.rowToSource(row))
|
|
|
|
def pov(self, row):
|
|
return self.sourceModel().pov(self.rowToSource(row))
|