Thomas Peetz
116 lines
4.3 KiB
Python
116 lines
4.3 KiB
Python
from datetime import datetime
|
|
|
|
import mariadb
|
|
from PySide6.QtCore import QAbstractTableModel, QModelIndex, Qt
|
|
from PySide6.QtGui import QColor
|
|
|
|
|
|
class MediaFileTableModel(QAbstractTableModel):
|
|
|
|
def __init__(self, db_config, main_window):
|
|
super().__init__()
|
|
self.main_window = main_window
|
|
self._data = []
|
|
self.mariadb_conn = mariadb.connect(
|
|
host=db_config['mariadb']['host'],
|
|
port=db_config['mariadb']['port'],
|
|
user=db_config['mariadb']['user'],
|
|
password=db_config['mariadb']['password'],
|
|
database=db_config['mariadb']['database']
|
|
)
|
|
self.refresh()
|
|
|
|
def refresh(self):
|
|
data = []
|
|
cursor = self.mariadb_conn.cursor()
|
|
filter_rule = ""
|
|
print(self.main_window.filter["download"].isChecked())
|
|
if self.main_window.filter["download"].isChecked():
|
|
print(self.main_window.filter["download"].isChecked())
|
|
filter_rule = "WHERE should_download is true"
|
|
if self.main_window.filter["review"].isChecked():
|
|
if len(filter_rule) > 0:
|
|
filter_rule += " AND "
|
|
else:
|
|
filter_rule += "WHERE "
|
|
filter_rule += "review is true"
|
|
print(f"{filter_rule=}")
|
|
try:
|
|
cursor.execute(f"SELECT id, url, review, should_download, file_name, cloud_link FROM media_file {filter_rule}")
|
|
except mariadb.Error as error:
|
|
print(error)
|
|
rows = cursor.fetchall()
|
|
print(len(rows))
|
|
if len(rows) > 0:
|
|
self.beginResetModel()
|
|
for row in rows:
|
|
data.append(list(row))
|
|
self._data = data
|
|
self.endResetModel()
|
|
else:
|
|
self._data = None
|
|
try:
|
|
self.layoutChanged.emit()
|
|
except:
|
|
|
|
self.main_window.statusBar.showMessage(f"{len(rows)} Einträge geladen", 3000)
|
|
|
|
def rowCount(self, parent=QModelIndex()):
|
|
# The length of the outer list.
|
|
return len(self._data)
|
|
|
|
def headerData(self, col, orientation, role=Qt.ItemDataRole.DisplayRole):
|
|
if orientation == Qt.Orientation.Horizontal and role == Qt.ItemDataRole.DisplayRole:
|
|
match col:
|
|
case 0:
|
|
return "ID"
|
|
case 1:
|
|
return "URL"
|
|
case 2:
|
|
return "Review"
|
|
case 3:
|
|
return "Download"
|
|
case 4:
|
|
return "Filename"
|
|
case 5:
|
|
return "Cloud Link"
|
|
if orientation == Qt.Orientation.Vertical and role == Qt.ItemDataRole.DisplayRole:
|
|
return str(col+1)
|
|
|
|
def data(self, index, role=Qt.ItemDataRole.DisplayRole):
|
|
value = self._data[index.row()][index.column()]
|
|
if role == Qt.ItemDataRole.DisplayRole:
|
|
if isinstance(value, datetime):
|
|
return value.strftime("%Y-%m-%d %M:%M:%S")
|
|
if isinstance(value, str):
|
|
return value
|
|
if isinstance(value, bytes):
|
|
if value == b'\x01':
|
|
return self.main_window.tick
|
|
else:
|
|
return self.main_window.cross
|
|
return value
|
|
if role == Qt.ItemDataRole.DecorationRole:
|
|
if isinstance(value, bytes):
|
|
# print('{}: {}'.format(value, type(value)))
|
|
if value == b'\x01':
|
|
return self.main_window.tick
|
|
else:
|
|
return self.main_window.cross
|
|
|
|
def columnCount(self, index=QModelIndex()):
|
|
# The following takes the first sub-list, and returns
|
|
# the length (only works if all rows are an equal length)
|
|
return len(self._data[0])
|
|
|
|
def setData(self, index, value, role=Qt.ItemDataRole.EditRole):
|
|
if role == Qt.ItemDataRole.EditRole:
|
|
self._data[index.row()][index.column()] = value
|
|
if role == Qt.ItemDataRole.CheckStateRole:
|
|
checked = value == Qt.CheckState.Checked
|
|
self._data[index.row()][index.column()] = checked
|
|
return True
|
|
|
|
def flags(self, index):
|
|
return Qt.ItemFlag.ItemIsSelectable | Qt.ItemFlag.ItemIsEnabled | Qt.ItemFlag.ItemIsEditable | Qt.ItemFlag.ItemIsUserTristate
|