Refactor tableview out of MainWindow

MainWindow.py

@@ -1,10 +1,11 @@
-from typing import Any, cast
-from PySide6.QtCore import QDate, QEvent, QModelIndex, QObject, QPersistentModelIndex, QRect, QSize, Signal, Qt, Slot
-from PySide6.QtGui import QColor, QPainter, QTextDocument
+from typing import Any, Self, cast
+from PySide6.QtCore import QDate, QEvent, QModelIndex, QObject, QPersistentModelIndex, QPoint, QRect, QSize, Signal, Qt, Slot
+from PySide6.QtGui import QColor, QFont, QPainter, QTextDocument, QTextDocumentFragment
 from PySide6.QtSql import QSqlTableModel
-from PySide6.QtWidgets import QAbstractItemView, QHeaderView, QMainWindow, QStyle, QStyleOptionViewItem, QStyledItemDelegate, QWidget
+from PySide6.QtWidgets import QAbstractItemView, QHeaderView, QMainWindow, QStyle, QStyleOptionViewItem, QStyledItemDelegate, QTableView, QWidget
 
 from docketModel import docketModel
+from dockettableview import documentDelegate
 from ui.MainWindow import Ui_MainWindow
 from workers import loadCases, updateThread
 
@@ -12,6 +13,7 @@ class QStyleOptionViewItemInit(QStyleOptionViewItem):
     backgroundBrush: QColor
     rect: QRect
     widget: QWidget
+    font: QFont
     text: str
 
 class dateDelegate(QStyledItemDelegate):
@@ -47,69 +49,6 @@ class activeDelegate(QStyledItemDelegate):
         return
 
 
-class documentDelegate(QStyledItemDelegate):
-    def initStyleOption(
-        self,
-        option: QStyleOptionViewItem,
-        index: QModelIndex | QPersistentModelIndex,
-        /,
-    ) -> None:
-        options = cast(QStyleOptionViewItemInit, option)
-        super().initStyleOption(options, index)
-        assert index.isValid() and isinstance(index, QModelIndex)
-        if index.siblingAtColumn(6).data() == 0:
-            options.backgroundBrush = QColor(0x444444)
-        return
-
-    def eventFilter(self, object: QObject, event: QEvent) -> bool:
-        if event.type() in [
-            QEvent.Type.MouseButtonPress,
-            QEvent.Type.MouseButtonRelease,
-        ]:
-            return False
-        return super().eventFilter(object, event)
-
-    def event(self, ev: QEvent) -> bool:
-        print(ev)
-        return super(documentDelegate, self).event(ev)
-
-    def paint(
-        self,
-        painter: QPainter,
-        option: QStyleOptionViewItem,
-        index: QModelIndex | QPersistentModelIndex,
-    ) -> None:
-        options = cast(QStyleOptionViewItemInit, option)
-        self.initStyleOption(options, index)
-        painter.save()
-        doc = QTextDocument()
-        doc.setTextWidth(options.rect.width())
-        doc.setHtml(options.text)
-        options.text = ""
-        options.widget.style().drawControl(
-            QStyle.ControlElement.CE_ItemViewItem,
-            options,
-            painter,
-        )
-        painter.translate(options.rect.left(), options.rect.top())
-        clip = QRect(0, 0, options.rect.width(), options.rect.height())
-        doc.drawContents(painter, clip)
-        painter.restore()
-        return
-
-    def sizeHint(
-        self,
-        option: QStyleOptionViewItem,
-        index: QModelIndex | QPersistentModelIndex,
-    ) -> QSize:
-        options = cast(QStyleOptionViewItemInit, option)
-        self.initStyleOption(options, index)
-        doc = QTextDocument()
-        doc.setTextWidth(options.rect.width())
-        doc.setHtml(options.text)
-        doc.setTextWidth(options.rect.width())
-        return QSize(int(doc.idealWidth()), int(doc.size().height()))
-
 
 class MainWindow(QMainWindow, Ui_MainWindow):
     show_entries = Signal(int)
@@ -119,12 +58,13 @@ class MainWindow(QMainWindow, Ui_MainWindow):
     def __init__(self) -> None:
         super(MainWindow, self).__init__()
         self.setupUi(self)
-        self.loadThread = loadCases()
-        self.loadThread.finished.connect(self.updateDone)
-        self.loadThread.start()
+        #self.loadThread = loadCases()
+        #self.loadThread.finished.connect(self.updateDone)
+        #self.loadThread.start()
 
         model = QSqlTableModel()
         model.setTable("cases")
+        model.sort(1, Qt.SortOrder.AscendingOrder)
         model.select()
         model.setHeaderData(1, Qt.Orientation.Horizontal, "Docket")
         model.setHeaderData(2, Qt.Orientation.Horizontal, "Petitioners")
@@ -141,7 +81,7 @@ class MainWindow(QMainWindow, Ui_MainWindow):
         self.casesView.setItemDelegate(activeDelegate())
         self.casesView.setItemDelegateForColumn(4, dateDelegate())
         self.casesView.resizeColumnToContents(1)
-        self.casesView.resizeColumnToContents(5)
+        self.casesView.resizeColumnToContents(4)
         header = self.casesView.horizontalHeader()
         header.setSectionResizeMode(2, QHeaderView.ResizeMode.Fixed)
         header.setSectionResizeMode(3, QHeaderView.ResizeMode.Fixed)
@@ -149,7 +89,7 @@ class MainWindow(QMainWindow, Ui_MainWindow):
         remaining = (
             self.casesView.width()
             - header.sectionSize(1)
-            - header.sectionSize(5)
+            - header.sectionSize(4)
             - 5
         )
         self.casesView.setColumnWidth(2, int(remaining * 0.5))
@@ -159,12 +99,13 @@ class MainWindow(QMainWindow, Ui_MainWindow):
         self.casesView.doubleClicked.connect(self.rowClicked)
         self.casesView.clicked.connect(self.rowClicked)
 
+        self.docketView.clickedEvent.connect(self.clickedEvent)
         self.docketView.setModel(docketModel())
         self.docketView.horizontalHeader().setSectionResizeMode(
             1, QHeaderView.ResizeMode.Stretch
         )
         self.docketView.resizeRowsToContents()
-        self.docketView.setItemDelegateForColumn(1, documentDelegate())
+        #self.docketView.setItemDelegateForColumn(1, documentDelegate())
         return
 
     @Slot(QModelIndex)  # type: ignore
@@ -174,8 +115,8 @@ class MainWindow(QMainWindow, Ui_MainWindow):
         docket = index.siblingAtColumn(1).data()
         self.docketLabel.setText(docket)
         self.show_entries.emit(index.siblingAtColumn(0).data())
-        model = docketModel(index.siblingAtColumn(0).data())
-        self.docketView.setModel(model)
+        model = cast(docketModel, self.docketView.model())
+        model.newCase(index.siblingAtColumn(0).data())
         self.docketView.resizeColumnToContents(0)
         self.docketView.resizeRowsToContents()
         return
@@ -194,11 +135,12 @@ class MainWindow(QMainWindow, Ui_MainWindow):
         self.updateThread.start()
         return
 
-    @Slot(QModelIndex)  # type: ignore
-    def on_docketView_clicked(self, index: QModelIndex) -> None:
-        print(f"view clicked, cell: {index.column()}, {index.row()}")
+    @Slot()
+    def clickedEvent(self, pos:QPoint) -> None:
+        print(pos)
+        viewport = self.docketView.viewport()
+        print(viewport,viewport.children())
         return
-
     @Slot()
     def updateDone(self) -> None:
         self.updateThread = None