You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytqt/examples/tablestatistics.py

171 lines
5.2 KiB

#!/usr/bin/env python
#****************************************************************************
#** $Id: tablestatistics.py,v 1.1 2002/06/19 07:56:07 phil Exp $
#**
#** Copyright (C) 1992-1998 Troll Tech AS. All rights reserved.
#**
#** This file is part of an example program for PyTQt. This example
#** program may be used, distributed and modified without limitation.
#**
#*****************************************************************************/
import sys
import os
from PyTQt.tqt import *
from PyTQt.tqttable import *
TRUE = 1
FALSE = 0
# column constants
TB_FILE = 0
TB_SIZE = 1
TB_FLAG = 2
TB_COLS = 3
dirs = (
"kernel",
"tools",
"widgets",
"dialogs",
"xml",
"table",
"network",
"opengl",
"canvas",
)
class Table(TQTable):
def __init__(self):
TQTable.__init__(self, 0, TB_COLS)
self.setSorting(TRUE)
self.horizontalHeader().setLabel(TB_FILE, self.tr("File"))
self.horizontalHeader().setLabel(TB_SIZE, self.tr("Size (bytes)"))
self.horizontalHeader().setLabel(TB_FLAG, self.tr("Use in Sum"))
self.initTable()
self.adjustColumn(TB_FILE)
# if the user edited something we might need to recalculate the sum
self.connect(self, TQ_SIGNAL("valueChanged(int, int)"), self.recalcSum)
def initTable(self):
# read all the TQt source and header files into a list
all = []
tqtdir = os.getenv("TQTDIR")
if tqtdir is None:
raise Exception("The TQTDIR environment variable has not been set.")
for i in dirs:
dir = TQDir(os.path.join(tqtdir, "src", i))
lst = TQStringList(dir.entryList("*.cpp; *.h"))
for f in lst:
if f.contains("moc"):
continue
all.append(os.path.join(i, str(f)))
# set the number of rows we'll need for the table
self.setNumRows(len(all) + 1)
i = 0
sum = 0
# insert the data into the table
for it in all:
self.setText(i, TB_FILE, it)
f = TQFile(os.path.join(tqtdir, "src", str(it)))
self.setText(i, TB_SIZE, str(f.size()))
ci = ComboItem(self, TQTableItem.WhenCurrent)
self.setItem(i, TB_FLAG, ci)
i = i + 1
sum += f.size()
self.displaySum(sum)
def recalcSum(self, dummy, col):
# only recalc if a value in the second or third column changed
if col < TB_SIZE or col > TB_FLAG:
return
sum = 0
for i in range(self.numRows()-1):
if str(self.text(i, TB_FLAG)) == "No":
continue
sum += int(str(self.text(i, TB_SIZE)))
self.displaySum(sum)
def displaySum(self, sum):
# insert calculated data
i1 = TableItem(self, TQTableItem.Never, self.tr("Sum"))
self.setItem(self.numRows()-1, TB_FILE, i1)
i2 = TableItem(self, TQTableItem.Never, str(sum))
self.setItem(self.numRows()-1, TB_SIZE, i2)
def sortColumn(self, col, ascending, wholeRows):
# sum row should not be sorted, so get rid of it for now
self.clearCell(self.numRows()-1, TB_FILE)
self.clearCell(self.numRows()-1, TB_SIZE)
# do sort
TQTable.sortColumn(self, col, ascending, TRUE)
# re-insert sum row
self.recalcSum(0, TB_SIZE)
class TableItem(TQTableItem):
def __init__(self, *args):
TQTableItem.__init__(*(self,) + args)
def paint(self, p, cg, cr, selected):
g = TQColorGroup(cg)
# last row is the sum row - we want to make it more visible by
# using a red background
if self.row() == self.table().numRows()-1:
g.setColor(TQColorGroup.Base, TQColor("red"))
TQTableItem.paint(self, p, g, cr, selected)
class ComboItem(TQTableItem):
def __init__(self, t, et):
TQTableItem.__init__(self, t, et, "Yes")
self.cb = None
# we do not want this item to be replaced
self.setReplaceable(FALSE)
def createEditor(self):
# create an editor - a combobox in our case
self.cb = TQComboBox(self.table().viewport())
TQObject.connect(self.cb, TQ_SIGNAL("activated(int)"),
self.table(), TQ_SLOT("doValueChanged()"))
self.cb.insertItem("Yes")
self.cb.insertItem("No")
# and initialize it
if str(self.text()) == "No":
self.cb.setCurrentItem(1)
else:
self.cb.setCurrentItem(0)
return self.cb
def setContentFromEditor(self, w):
# the user changed the value of the combobox, so synchronize the
# value of the item (its text), with the value of the combobox
if w.inherits("TQComboBox"):
self.setText(w.currentText())
else:
TQTableItem.setContentFromEditor(self, w)
def setText(self, s):
# initialize the combobox from the text
if self.cb:
if str(s) == "No":
self.cb.setCurrentItem(1)
else:
self.cb.setCurrentItem(0)
TQTableItem.setText(self, s)
if __name__ == '__main__':
app = TQApplication(sys.argv)
t = Table()
t.setCaption("Statistics")
t.show()
app.setMainWidget(t)
app.exec_loop()