1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
from PyQt5 import QtWidgets
from PyQt5.QtCore import Qt
import db.tasks
import gui.tasks.signal
import gui.tasks.table.menu
import gui.tasks.table.model
import gui.tasks.dialog
from model.task import Task, ValidTaskForm
class Widget(QtWidgets.QTableView):
def __init__(self, database, parent, add_task_signal):
super().__init__(parent)
self._database = database
self._update_task_signal = gui.tasks.signal.UpdateTask()
tasks = db.tasks.get(self._database.cursor())
table_model = gui.tasks.table.model.TableModel(tasks)
self.setModel(table_model)
self.sortByColumn(
gui.tasks.table.model.default_sort[0],
gui.tasks.table.model.default_sort[1])
self.setSortingEnabled(True)
self.setSelectionBehavior(QtWidgets.QTableView.SelectRows)
self.horizontalHeader().setStretchLastSection(True)
self.resizeColumns()
self.doubleClicked.connect(lambda index: self.on_double_click(index.row()))
# Menu
self.setContextMenuPolicy(Qt.CustomContextMenu)
self.customContextMenuRequested.connect(lambda position: gui.tasks.table.menu.open(self._database, self, self._update_task_signal, position))
add_task_signal.get().connect(lambda task: self.insert(task))
self._update_task_signal.get().connect(lambda row, task: self.update(row, task))
def insert(self, task):
self.model().insert_task(self.horizontalHeader(), task)
self.resizeColumns()
def update(self, row, task):
row = self.model().update_task(self.horizontalHeader(), row, task)
self.selectRow(row)
self.resizeColumns()
def resizeColumns(self):
for column in range(gui.tasks.table.model.columns):
self.resizeColumnToContents(column)
def keyPressEvent(self, event):
super().keyPressEvent(event)
if event.key() in (Qt.Key_Return, Qt.Key_Enter):
rows = self.get_selected_rows()
if len(rows) == 1:
row = rows[0]
task = self.model().get_at(row)
gui.tasks.dialog.update(
self._database, self, self._update_task_signal, row, task).exec_()
elif event.key() == Qt.Key_Delete:
rows = self.get_selected_rows()
gui.tasks.dialog.show_delete(self._database, self, rows)
def get_selected_rows(self):
return list(set([index.row() for index in self.selectedIndexes()]))
def on_double_click(self, row: int):
task = self.model().get_at(row)
gui.tasks.dialog.update(self._database, self, self._update_task_signal, row, task).exec_()
|
