File: table.py

package info (click to toggle)
openstructure 2.9.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 205,228 kB
  • sloc: cpp: 188,129; python: 35,361; ansic: 34,298; fortran: 3,275; sh: 286; xml: 146; makefile: 29
file content (59 lines) | stat: -rw-r--r-- 1,883 bytes parent folder | download | duplicates (4)
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
from PyQt5.QtGui import *
from PyQt5.QtCore import *

from ost import table
__all__=('Table', )

class TableModel(QAbstractTableModel):
  def __init__(self, table, parent=None):
    QAbstractTableModel.__init__(self, parent)
    self.table=table
    
  def rowCount(self, index):
    return len(self.table.rows)

  def headerData(self, section, orientation, role):
    if role!=Qt.DisplayRole or orientation!=Qt.Horizontal:
      return QVariant()
    return self.table.col_names[section]
    
  def columnCount(self, index):
    return len(self.table.col_names)

  def sort(self, column, order):
    o='+'
    if order!=Qt.AscendingOrder:
      o='-'
    self.table.Sort(by=self.table.col_names[column], order=o)
    self.reset()
    
  def data(self, index, role):
    if not index.isValid() or role!=Qt.DisplayRole:
      return QVariant()
    row=self.table.rows[index.row()]
    return QVariant(row[index.column()])

class Table(QTableView):
  def __init__(self, table):
     QTableView.__init__(self)
     self._model=TableModel(table)
     self.setFrameShape(QFrame.NoFrame)    
     self.setAttribute(Qt.WA_MacSmallSize)
     self.setShowGrid(False)
     self.double_click = None
     self.horizontalHeader().setStretchLastSection(True)
     self.setContextMenuPolicy(Qt.CustomContextMenu)
     self.setSelectionBehavior(QAbstractItemView.SelectRows)
     self.setSizePolicy(QSizePolicy.MinimumExpanding, 
                        QSizePolicy.MinimumExpanding)
     self.setSortingEnabled(True)
     self.setModel(self._model)
     QObject.connect(self, SIGNAL('doubleClicked(QModelIndex)'), 
                     self.OnDoubleClick)
  def OnDoubleClick(self, model_index):
    print('DOUBLE')
    if not self.double_click:
      return
    row = table.TableRow(self._model.table.rows[model_index.row()],
                         self._model.table)
    self.double_click(row)