Remove Table Model specific code
This commit is contained in:
BasioMeusPuga
139
models.py
139
models.py
@@ -184,145 +184,6 @@ class ProxyModelsCommonFunctions:
|
||||
return False
|
||||
|
||||
|
||||
class MostExcellentTableModel(QtCore.QAbstractTableModel):
|
||||
# Sorting is taken care of by the QSortFilterProxy model
|
||||
# which has an inbuilt sort method
|
||||
|
||||
# Modifying data in the table model is a case of modifying the
|
||||
# data sent to it as a list
|
||||
# In this case, that's self.data_list
|
||||
|
||||
def __init__(self, header_data, display_data, temp_dir=None, parent=None):
|
||||
super(MostExcellentTableModel, self).__init__(parent)
|
||||
self.header_data = header_data
|
||||
self.display_data = display_data
|
||||
self.temp_dir = temp_dir
|
||||
|
||||
def rowCount(self, parent):
|
||||
if self.display_data:
|
||||
return len(self.display_data)
|
||||
else:
|
||||
return 0
|
||||
|
||||
def columnCount(self, parent):
|
||||
return len(self.header_data)
|
||||
|
||||
def data(self, index, role):
|
||||
if not index.isValid():
|
||||
return None
|
||||
|
||||
# This block specializes this function for the library
|
||||
# Not having a self.temp_dir allows for its reuse elsewhere
|
||||
if self.temp_dir:
|
||||
if role == QtCore.Qt.DecorationRole and index.column() == 2:
|
||||
return_pixmap = None
|
||||
file_exists = self.display_data[index.row()][5]['file_exists']
|
||||
position = self.display_data[index.row()][5]['position']
|
||||
|
||||
if not file_exists:
|
||||
return_pixmap = pie_chart.pixmapper(
|
||||
-1, None, None, QtCore.Qt.SizeHintRole + 10)
|
||||
|
||||
if position:
|
||||
current_chapter = position['current_chapter']
|
||||
total_chapters = position['total_chapters']
|
||||
|
||||
return_pixmap = pie_chart.pixmapper(
|
||||
current_chapter, total_chapters, self.temp_dir,
|
||||
QtCore.Qt.SizeHintRole + 10)
|
||||
|
||||
return return_pixmap
|
||||
|
||||
# The rest of the roles can be accomodated here.
|
||||
elif role == QtCore.Qt.UserRole:
|
||||
value = self.display_data[index.row()][5] # File metadata
|
||||
return value
|
||||
|
||||
elif role == QtCore.Qt.UserRole + 1:
|
||||
value = self.display_data[index.row()][6] # File hash
|
||||
return value
|
||||
|
||||
#_________________________________
|
||||
# The EditRole is so that editing a cell doesn't clear its contents
|
||||
if role == QtCore.Qt.DisplayRole or role == QtCore.Qt.EditRole:
|
||||
value = self.display_data[index.row()][index.column()]
|
||||
return value
|
||||
|
||||
else:
|
||||
return QtCore.QVariant()
|
||||
|
||||
def headerData(self, col, orientation, role):
|
||||
if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
|
||||
return self.header_data[col]
|
||||
return None
|
||||
|
||||
def flags(self, index):
|
||||
# This means only the Tags column is editable
|
||||
if self.temp_dir and index.column() == 4:
|
||||
return QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable
|
||||
else:
|
||||
# These are standard select but don't edit values
|
||||
return QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable
|
||||
|
||||
def setData(self, index, value, role=QtCore.Qt.EditRole):
|
||||
# We don't need to connect this to dataChanged since the underlying
|
||||
# table model (not the proxy model) is the one that's being updated
|
||||
|
||||
# Database tags for files should not be updated each time
|
||||
# a new folder gets added or deleted from the directory
|
||||
# This will be done @ runtime
|
||||
# Individually set file tags will be preserved
|
||||
# Duplicate file tags will be removed
|
||||
|
||||
row = index.row()
|
||||
col = index.column()
|
||||
self.display_data[row][col] = value
|
||||
return True
|
||||
|
||||
|
||||
class TableProxyModel2(QtCore.QSortFilterProxyModel):
|
||||
def __init__(self, parent=None):
|
||||
super(TableProxyModel2, self).__init__(parent)
|
||||
self.filter_string = None
|
||||
self.filter_columns = None
|
||||
self.active_library_filters = None
|
||||
|
||||
def setFilterParams(self, filter_text, filter_columns, active_library_filters):
|
||||
self.filter_string = filter_text.lower()
|
||||
self.filter_columns = filter_columns
|
||||
self.active_library_filters = [i.lower() for i in active_library_filters]
|
||||
|
||||
def filterAcceptsRow(self, row_num, parent):
|
||||
if self.filter_string is None or self.filter_columns is None:
|
||||
return True
|
||||
|
||||
model = self.sourceModel()
|
||||
|
||||
valid_indices = [model.index(row_num, i) for i in self.filter_columns]
|
||||
valid_data = [
|
||||
model.data(i, QtCore.Qt.DisplayRole).lower() for i in valid_indices if model.data(
|
||||
i, QtCore.Qt.DisplayRole) is not None]
|
||||
|
||||
try:
|
||||
valid_data.extend([model.display_data[row_num][7], model.display_data[row_num][8]])
|
||||
except IndexError: # Columns 7 and 8 are added after creation of the model
|
||||
pass
|
||||
|
||||
# Filter out all books not in the active library filters
|
||||
if self.active_library_filters:
|
||||
current_library_name = valid_data[-2].lower()
|
||||
if current_library_name not in self.active_library_filters:
|
||||
return False
|
||||
else:
|
||||
return False
|
||||
|
||||
for i in valid_data:
|
||||
if i:
|
||||
if self.filter_string in i:
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
class MostExcellentFileSystemModel(QtWidgets.QFileSystemModel):
|
||||
# Directories are tracked on the basis of their paths
|
||||
# Poll the tag_data dictionary to get User selection
|
||||
|
||||
Reference in New Issue
Block a user