|
1 # -*- coding: utf-8 -*- |
|
2 |
|
3 # Copyright (c) 2009 - 2021 Detlev Offenbach <detlev@die-offenbachs.de> |
|
4 # |
|
5 |
|
6 """ |
|
7 Module implementing specialized table views. |
|
8 """ |
|
9 |
|
10 from PyQt5.QtCore import Qt, QItemSelectionModel |
|
11 from PyQt5.QtWidgets import QTableView |
|
12 |
|
13 |
|
14 class E5TableView(QTableView): |
|
15 """ |
|
16 Class implementing a table view supporting removal of entries. |
|
17 """ |
|
18 def keyPressEvent(self, evt): |
|
19 """ |
|
20 Protected method implementing special key handling. |
|
21 |
|
22 @param evt reference to the event (QKeyEvent) |
|
23 """ |
|
24 if ( |
|
25 evt.key() in [Qt.Key.Key_Delete, Qt.Key.Key_Backspace] and |
|
26 self.model() is not None |
|
27 ): |
|
28 self.removeSelected() |
|
29 evt.setAccepted(True) |
|
30 else: |
|
31 super().keyPressEvent(evt) |
|
32 |
|
33 def removeSelected(self): |
|
34 """ |
|
35 Public method to remove the selected entries. |
|
36 """ |
|
37 if self.model() is None or self.selectionModel() is None: |
|
38 # no models available |
|
39 return |
|
40 |
|
41 row = 0 |
|
42 selectedRows = self.selectionModel().selectedRows() |
|
43 for selectedRow in reversed(selectedRows): |
|
44 row = selectedRow.row() |
|
45 self.model().removeRow(row, self.rootIndex()) |
|
46 |
|
47 idx = self.model().index(row, 0, self.rootIndex()) |
|
48 if not idx.isValid(): |
|
49 idx = self.model().index(row - 1, 0, self.rootIndex()) |
|
50 self.selectionModel().select( |
|
51 idx, |
|
52 QItemSelectionModel.SelectionFlag.SelectCurrent | |
|
53 QItemSelectionModel.SelectionFlag.Rows) |
|
54 self.setCurrentIndex(idx) |
|
55 |
|
56 def removeAll(self): |
|
57 """ |
|
58 Public method to clear the view. |
|
59 """ |
|
60 if self.model() is not None: |
|
61 self.model().removeRows(0, self.model().rowCount(self.rootIndex()), |
|
62 self.rootIndex()) |