Troglodite/scotus-watch
3
0
Fork 0
Code Issues 10 Pull Requests Actions Packages Projects Releases Wiki Activity

Lint

Browse Source
This commit is contained in:
Christopher T. Johnson
2025-02-08 16:42:19 -05:00
parent ffc840dc66
commit 488203d868
3 changed files with 139 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

128
scotus-pull.py
View File

@@ -37,25 +37,70 @@ translate = QCoreApplication.translate
class dateDelegate(QStyledItemDelegate):
def displayText(self, value, locale) -> str:
def displayText(self, value, locale) -> str: # type: ignore
date = datetime.date.fromtimestamp(value)
return date.strftime("%B %-d, %Y")
class documentDelegate(QStyledItemDelegate):
def paint(
self,
painter: QPainter,
option: QStyleOptionViewItem,
index: QModelIndex | QPersistentModelIndex,
):
options = option
self.initStyleOption(options, index)
painter.save()
doc = QTextDocument()
doc.setTextWidth(option.rect.width()) # type: ignore
doc.setHtml(option.text) # type: ignore
option.text = "" # type: ignore
option.widget.style().drawControl( # type: ignore
QStyle.ControlElement.CE_ItemViewItem,
option,
painter,
)
painter.translate(option.rect.left(), options.rect.top()) # type: ignore
clip = QRect(0, 0, option.rect.width(), option.rect.height()) # type: ignore
doc.drawContents(painter, clip)
painter.restore()
return
def sizeHint(
self,
option: QStyleOptionViewItem,
index: QModelIndex | QPersistentModelIndex,
) -> QSize:
self.initStyleOption(option, index)
doc = QTextDocument()
doc.setTextWidth(option.rect.width()) # type: ignore
doc.setHtml(option.text) # type: ignore
doc.setTextWidth(option.rect.width()) # type: ignore
return QSize(int(doc.idealWidth()), int(doc.size().height()))
class MainWindow(QMainWindow, Ui_MainWindow):
show_entries = Signal(int)
def __init__(self) -> None:
super(MainWindow, self).__init__()
self.setupUi(self)
model = QSqlQueryModel()
# model = QSqlQueryModel()
model = QSqlTableModel()
query = QSqlQuery("SELECT * FROM cases ORDER BY docket_id")
if not query.exec():
query_error(query)
model.setQuery(query)
self.casesView.setModel(model)
self.casesView.setSelectionMode(QAbstractItemView.SelectionMode.SingleSelection)
self.casesView.setSelectionBehavior(QAbstractItemView.SelectionBehavior.SelectRows)
self.casesView.setSelectionMode(
QAbstractItemView.SelectionMode.SingleSelection
)
self.casesView.setSelectionBehavior(
QAbstractItemView.SelectionBehavior.SelectRows
)
self.casesView.hideColumn(0)
self.casesView.hideColumn(2)
self.casesView.setItemDelegateForColumn(5, dateDelegate())
@@ -65,21 +110,29 @@ class MainWindow(QMainWindow, Ui_MainWindow):
header.setSectionResizeMode(3, QHeaderView.ResizeMode.Fixed)
header.setSectionResizeMode(4, QHeaderView.ResizeMode.Fixed)
self.show()
remaining = self.casesView.width() - header.sectionSize(1) - header.sectionSize(5) - 5
self.casesView.setColumnWidth(3,int(remaining * 0.5))
self.casesView.setColumnWidth(4,int(remaining * 0.5))
remaining = (
self.casesView.width()
- header.sectionSize(1)
- header.sectionSize(5)
- 5
)
self.casesView.setColumnWidth(3, int(remaining * 0.5))
self.casesView.setColumnWidth(4, int(remaining * 0.5))
self.casesView.verticalHeader().hide()
self.casesView.resizeRowsToContents()
self.casesView.doubleClicked.connect(self.rowClicked)
self.casesView.clicked.connect(self.rowClicked)
self.docketView.setModel(docketModel())
self.docketView.horizontalHeader().setSectionResizeMode(1, QHeaderView.ResizeMode.Stretch)
self.docketView.horizontalHeader().setSectionResizeMode(
1, QHeaderView.ResizeMode.Stretch
)
self.docketView.resizeRowsToContents()
self.docketView.setItemDelegateForColumn(1, documentDelegate())
return
@Slot(QModelIndex)
def rowClicked(self, index:QModelIndex) -> None:
@Slot(QModelIndex) # type: ignore
def rowClicked(self, index: QModelIndex) -> None:
if not index.isValid():
raise Exception("Bad index")
docket = index.siblingAtColumn(1).data()
@@ -91,15 +144,32 @@ class MainWindow(QMainWindow, Ui_MainWindow):
self.docketView.resizeRowsToContents()
return
updateThread = None
@Slot()
def on_updateButton_clicked(self):
text = self.docketInput.toPlainText()
update_db(text)
self.update()
print(f"on_updateButton_clicked(): {text}")
if not self.updateThread:
self.updateThread = updateThread()
assert isinstance(self.updateThread, updateThread)
self.updateThread.finished.connect(self.updateDone)
self.updateThread.setDocketId(text)
self.updateThread.start()
return
@Slot()
def updateDone(self):
self.updateThread = None
print("Done updating")
model: QSqlTableModel = self.casesView.model() # type: ignore
query = model.query()
query.exec()
return
SQL_CMDS = [
#"PRAGMA foreign_keys=ON",
# "PRAGMA foreign_keys=ON",
"CREATE TABLE IF NOT EXISTS cases "
"(case_id INTEGER PRIMARY KEY AUTOINCREMENT, "
"docket_id TEXT, "
@@ -121,9 +191,15 @@ SQL_CMDS = [
"name TEXT, "
"url TEXT, "
"FOREIGN KEY(entry_id) REFERENCES entries(entry_id))",
]
#
"CREATE TABLE IF NOT EXISTS history ("
"history_id INTEGER PRIMARY KEY AUTOINCREMENT, "
"year TEXT, "
"edocket INTEGER, "
"number INTEGER)",
]
def schema_update(db: QSqlDatabase) -> None:
query = QSqlQuery()
@@ -148,7 +224,7 @@ def schema_update(db: QSqlDatabase) -> None:
else:
raise AttributeError(f"No match found: {create_cmd}")
print("Table name = {}".format(table_name))
print(f"Table name = {table_name}")
query.prepare("SELECT sql FROM sqlite_schema WHERE tbl_name = :tbl")
query.bindValue(":tbl", table_name)
if not query.exec():
@@ -178,7 +254,10 @@ def schema_update(db: QSqlDatabase) -> None:
query_error(query)
# step 5 transfer content
coldefs = re.search(r"\((.+)\)", old).group(1).split(", ") # type: ignore[union-attr]
cols = [x.split(" ")[0] for x in filter(lambda s: not s.startswith('FOREIGN '),coldefs)]
cols = [
x.split(" ")[0]
for x in filter(lambda s: not s.startswith("FOREIGN "), coldefs)
]
cols_str = ", ".join(cols)
sql = f"INSERT INTO {new_table_name} ({cols_str}) SELECT {cols_str} FROM {table_name}"
query.prepare(sql)
@@ -190,15 +269,17 @@ def schema_update(db: QSqlDatabase) -> None:
if not query.exec():
query_error(query)
# step 6 rename new table to old table
query.prepare("ALTER TABLE " + new_table_name + " RENAME TO " + table_name)
query.prepare(
"ALTER TABLE " + new_table_name + " RENAME TO " + table_name
)
if not query.exec():
query_error(query)
# step 8 create indexes, triggers, and views
# step 9 rebuild affected views
# step 10 turn foreign key constrants back on
# if not query.exec("PRAGMA foreign_keys=ON"):
# query_error(query)
# if not query.exec("PRAGMA foreign_keys=ON"):
# query_error(query)
# step 11 commit the changes
db.commit()
return
@@ -207,12 +288,13 @@ def schema_update(db: QSqlDatabase) -> None:
def main() -> int:
app = QApplication(sys.argv)
db = QSqlDatabase.addDatabase("QSQLITE")
#db.setConnectOptions("PRAGMA foreign_keys = ON")
# db.setConnectOptions("PRAGMA foreign_keys = ON")
db.setDatabaseName("scotus.db")
db.open()
schema_update(db)
window = MainWindow()
return app.exec()
if __name__ == "__main__":
sys.exit(main())