Lint
This commit is contained in:
@@ -1,19 +1,28 @@
|
||||
import datetime
|
||||
from PySide6.QtCore import QAbstractTableModel, QModelIndex, Qt
|
||||
from PySide6.QtGui import QColor, QFont
|
||||
|
||||
from PySide6.QtCore import (
|
||||
QAbstractTableModel,
|
||||
QModelIndex,
|
||||
QPersistentModelIndex,
|
||||
Qt,
|
||||
)
|
||||
from PySide6.QtGui import QFont
|
||||
from PySide6.QtSql import QSqlQuery
|
||||
|
||||
from lib.utils import query_error
|
||||
|
||||
|
||||
class docketModel(QAbstractTableModel):
|
||||
entries = []
|
||||
def __init__(self, case_id:int|None = None) -> None:
|
||||
super(docketModel,self).__init__()
|
||||
entries: list[list[str | None]] = []
|
||||
|
||||
def __init__(self, case_id: int | None = None) -> None:
|
||||
super(docketModel, self).__init__()
|
||||
if case_id == None:
|
||||
return
|
||||
query = QSqlQuery()
|
||||
query.prepare("SELECT * FROM entries WHERE case_id = :cid "
|
||||
"ORDER BY entry_id")
|
||||
query.prepare(
|
||||
"SELECT * FROM entries WHERE case_id = :cid " "ORDER BY entry_id"
|
||||
)
|
||||
q2 = QSqlQuery()
|
||||
q2.prepare("SELECT * FROM documents WHERE entry_id = :eid")
|
||||
|
||||
@@ -23,38 +32,40 @@ class docketModel(QAbstractTableModel):
|
||||
while query.next():
|
||||
date = datetime.date.fromtimestamp(query.value(2))
|
||||
assert isinstance(date, datetime.date)
|
||||
row = [
|
||||
date.strftime("%B %-d, %Y"),
|
||||
query.value(3)
|
||||
]
|
||||
row = [date.strftime("%B %-d, %Y"), query.value(3)]
|
||||
self.entries.append(row)
|
||||
q2.bindValue(":eid", query.value("entry_id"))
|
||||
if not q2.exec():
|
||||
query_error(q2)
|
||||
row = []
|
||||
while q2.next():
|
||||
row.append(q2.value('name'))
|
||||
row.append("<u>" + q2.value("name") + "</u>")
|
||||
if len(row) > 0:
|
||||
self.entries.append([None, " ".join(row)])
|
||||
return
|
||||
|
||||
def rowCount(self, parent:QModelIndex|None = None) -> int:
|
||||
|
||||
def rowCount(
|
||||
self, _: QModelIndex | QPersistentModelIndex = QModelIndex()
|
||||
) -> int:
|
||||
return len(self.entries)
|
||||
|
||||
def columnCount(self, parent:QModelIndex|None = None) -> int:
|
||||
def columnCount(
|
||||
self, _: QModelIndex | QPersistentModelIndex = QModelIndex()
|
||||
) -> int:
|
||||
return 2
|
||||
|
||||
def data(self, index: QModelIndex, role:int):
|
||||
def data(self, index: QModelIndex | QPersistentModelIndex, role: int = 0):
|
||||
if not index.isValid():
|
||||
return ''
|
||||
return ""
|
||||
if role == Qt.ItemDataRole.DisplayRole:
|
||||
return self.entries[index.row()][index.column()]
|
||||
if role == Qt.ItemDataRole.TextAlignmentRole and index.column() == 0:
|
||||
return Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignTop
|
||||
return
|
||||
|
||||
|
||||
def headerData(self, section:int, orientation:Qt.Orientation, role:int):
|
||||
def headerData(
|
||||
self, section: int, orientation: Qt.Orientation, role: int = 0
|
||||
):
|
||||
if orientation == Qt.Orientation.Vertical:
|
||||
return
|
||||
if role == Qt.ItemDataRole.FontRole:
|
||||
@@ -63,4 +74,4 @@ class docketModel(QAbstractTableModel):
|
||||
return font
|
||||
if role != Qt.ItemDataRole.DisplayRole:
|
||||
return
|
||||
return ['Date', 'Proceedings and Orders'][section]
|
||||
return ["Date", "Proceedings and Orders"][section]
|
||||
|
||||
Reference in New Issue
Block a user