Select columns from a QTableWidget

[toglez]

Hi all!

I use PyQt4 and Python to create a program to display numbers from a file in a QTable Widget, select various columns from the table and then display the selected columns in a new table.

Well, I don't seem to understand how to do this. Here is my code:

main.py:
-------------
-----------------------------------code start --------------------------------------------------------

from PyQt4 import QtGui, QtCore
from tstMain import Ui_tstMain
 
import sys, string
from os.path import isfile
from os import chdir
 
import variousFuncs
 
 
## Create a dictionary of updateable variables. These will be updated by the various
## functions of Class startGui
updVars = {"InitialFile": [], "SelectedColumns": []}
 
## the Ui_tstMain module is imported and is a file generated by pyuic4, out of
## a "layout" file tstMain.ui created by Qt4
class startGui(QtGui.QMainWindow):
    def __init__(self, parent = None):
        QtGui.QWidget.__init__(self, parent)
        self.ui = Ui_tstMain()
        self.ui.setupUi(self)
 
        ## create our slots here
        QtCore.QObject.connect(self.ui.btn_SetWorkingDir, QtCore.SIGNAL("clicked()"), self.setWorkingDir)
        QtCore.QObject.connect(self.ui.btn_OpenFile, QtCore.SIGNAL("clicked()"), self.openFile)
        QtCore.QObject.connect(self.ui.btn_ListShow, QtCore.SIGNAL("clicked()"), self.showList)
 
    def setWorkingDir(self):
        wd = QtGui.QFileDialog(self)
        currentDir = QtCore.QString(wd.getExistingDirectory())
        self.ui.lnEdt_WorkingDir.setText(currentDir)
        chdir(currentDir)
 
    def openFile(self):
        fd = QtGui.QFileDialog(self)
        self.file = fd.getOpenFileName()
 
        if isfile(self.file):
            fileToOpen = open(self.file)
            initialFile = [string.split(line) for line in fileToOpen]
            variousFuncs.dbl_str2num(initialFile)
 
            updVars["InitialFile"] = initialFile
            variousFuncs.tablePainter(initialFile, self.ui.tbl_OpenFile)
 
            updVars["SelectedColumns"] = self.ui.tbl_OpenFile.selectedItems
 
    def showList(self):
        tempSelected = updVars["SelectedColumns"]
        variousFuncs.tablePainter(tempSelected, self.ui.tbl_ListShow)
 
if __name__ == "__main__":
    app = QtGui.QApplication(sys.argv)
    appStart = startGui()
    appStart.show()
    sys.exit(app.exec_())

--------------------------------------code end--------------------------------------------------------------

the various function script is this:
varousFuncs.py:
-----------------------------------code start --------------------------------------------------------

#!/usr/bin/python
 
import string, random, math
from PyQt4 import QtGui, QtCore
 
##
def dbl_str2num(lst):
    """ This function transforms the items of the list "lst" from strings to numbers.
    The list must be a nested list. """
    for i in range(len(lst)):
        for j in range(len(lst[0])):
            lst[i][j] = eval(lst[i][j])
 
def tablePainter(someTwoDimLst, someTbl):
    """ This function creates a QTableWidgetItem out of a 2-dimension nested list which contains numbers,
    and then uses it to "paint" the someTbl object, which is a QTableWidget """
    someTbl.setColumnCount(len(someTwoDimLst[0]))
    someTbl.setRowCount(len(someTwoDimLst))
 
    for i in range(len(someTwoDimLst)):
        for j in range(len(someTwoDimLst[0])):
            itemListShow = QtGui.QTableWidgetItem(QtCore.QString.number(someTwoDimLst[i][j]), QtGui.QTableWidgetItem.Type)
            someTbl.setItem(i, j, itemListShow)

--------------------------------------code end--------------------------------------------------------------

as shown, the critical line in main.py is the line:

updVars["SelectedColumns"] = self.ui.tbl_OpenFile.selectedItems

which updates the global dictionary updVars with the selected columns (does it???) from the table tbl_OpenFile. The dictionary data is then used by the function showList() to display the selected items in a new table.

The whole thing ends up with the following error error:

Traceback (most recent call last):
File "D:\Tutorial\main.py", line 48, in showList
variousFuncs.tablePainter(tempSelected, self.ui.tbl_ListShow)
File "D:\Tutorial\variousFuncs.py", line 17, in tablePainter
someTbl.setColumnCount(len(someTwoDimLst[0]))
TypeError: unsubscriptable object

Any help would be much appreciated
Thanx all in advance,

Thomas

[bartonc]

Should be:

#
            initialFile = [line.split() for line in fileToOpen.readlines()]

on line #40

[toglez]

Thank you bartonc for the suggestion.

initialFile = [line.split() for line in fileToOpen.readlines()]

works,
as well as my version:

initialFile = [string.split(line) for line in fileToOpen]

but this doesn't cure the problem. I am still not able to select the columns and display them into another table.

I also forgot to mention the python script (tstMain.py) created by pyuic4 out of the initial tstMain.ui file. So, this is it:

# -*- coding: utf-8 -*-
 
# Form implementation generated from reading ui file 'tstMain.ui'
#
# Created: Sun Sep 30 21:39:12 2007
#      by: PyQt4 UI code generator 4.3
#
# WARNING! All changes made in this file will be lost!
 
from PyQt4 import QtCore, QtGui
 
class Ui_tstMain(object):
    def setupUi(self, tstMain):
        tstMain.setObjectName("tstMain")
        tstMain.setEnabled(True)
        tstMain.resize(QtCore.QSize(QtCore.QRect(0,0,951,394).size()).expandedTo(tstMain.minimumSizeHint()))
 
        sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Preferred,QtGui.QSizePolicy.Preferred)
        sizePolicy.setHorizontalStretch(0)
        sizePolicy.setVerticalStretch(0)
        sizePolicy.setHeightForWidth(tstMain.sizePolicy().hasHeightForWidth())
        tstMain.setSizePolicy(sizePolicy)
        tstMain.setLayoutDirection(QtCore.Qt.LeftToRight)
 
        self.centralwidget = QtGui.QWidget(tstMain)
        self.centralwidget.setObjectName("centralwidget")
 
        self.tbl_OpenFile = QtGui.QTableWidget(self.centralwidget)
        self.tbl_OpenFile.setGeometry(QtCore.QRect(110,80,811,121))
 
        sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding,QtGui.QSizePolicy.Expanding)
        sizePolicy.setHorizontalStretch(0)
        sizePolicy.setVerticalStretch(0)
        sizePolicy.setHeightForWidth(self.tbl_OpenFile.sizePolicy().hasHeightForWidth())
        self.tbl_OpenFile.setSizePolicy(sizePolicy)
        self.tbl_OpenFile.setMaximumSize(QtCore.QSize(16777215,16777215))
        self.tbl_OpenFile.setLineWidth(1)
        self.tbl_OpenFile.setObjectName("tbl_OpenFile")
 
        self.btn_OpenFile = QtGui.QPushButton(self.centralwidget)
        self.btn_OpenFile.setGeometry(QtCore.QRect(20,80,81,23))
        self.btn_OpenFile.setObjectName("btn_OpenFile")
 
        self.tbl_ListShow = QtGui.QTableWidget(self.centralwidget)
        self.tbl_ListShow.setGeometry(QtCore.QRect(20,260,901,81))
        self.tbl_ListShow.setObjectName("tbl_ListShow")
 
        self.btn_SetWorkingDir = QtGui.QPushButton(self.centralwidget)
        self.btn_SetWorkingDir.setGeometry(QtCore.QRect(20,50,82,23))
        self.btn_SetWorkingDir.setObjectName("btn_SetWorkingDir")
 
        self.lnEdt_WorkingDir = QtGui.QLineEdit(self.centralwidget)
        self.lnEdt_WorkingDir.setEnabled(True)
        self.lnEdt_WorkingDir.setGeometry(QtCore.QRect(110,50,811,20))
        self.lnEdt_WorkingDir.setAutoFillBackground(False)
        self.lnEdt_WorkingDir.setReadOnly(True)
        self.lnEdt_WorkingDir.setObjectName("lnEdt_WorkingDir")
 
        self.btn_ListShow = QtGui.QPushButton(self.centralwidget)
        self.btn_ListShow.setGeometry(QtCore.QRect(730,230,191,23))
        self.btn_ListShow.setObjectName("btn_ListShow")
        tstMain.setCentralWidget(self.centralwidget)
 
        self.menubar = QtGui.QMenuBar(tstMain)
        self.menubar.setGeometry(QtCore.QRect(0,0,951,21))
        self.menubar.setObjectName("menubar")
        tstMain.setMenuBar(self.menubar)
 
        self.statusbar = QtGui.QStatusBar(tstMain)
        self.statusbar.setObjectName("statusbar")
        tstMain.setStatusBar(self.statusbar)
 
        self.retranslateUi(tstMain)
        QtCore.QMetaObject.connectSlotsByName(tstMain)
 
    def retranslateUi(self, tstMain):
        tstMain.setWindowTitle(QtGui.QApplication.translate("tstMain", "My Test Application", None, QtGui.QApplication.UnicodeUTF8))
        self.tbl_OpenFile.setRowCount(0)
        self.tbl_OpenFile.clear()
        self.tbl_OpenFile.setColumnCount(0)
        self.tbl_OpenFile.setRowCount(0)
        self.btn_OpenFile.setText(QtGui.QApplication.translate("tstMain", "Open File", None, QtGui.QApplication.UnicodeUTF8))
        self.tbl_ListShow.clear()
        self.tbl_ListShow.setColumnCount(0)
        self.tbl_ListShow.setRowCount(0)
        self.btn_SetWorkingDir.setText(QtGui.QApplication.translate("tstMain", "Set Working Dir", None, QtGui.QApplication.UnicodeUTF8))
        self.lnEdt_WorkingDir.setText(QtGui.QApplication.translate("tstMain", "Working Directory NOT yet set !", None, QtGui.QApplication.UnicodeUTF8))
        self.btn_ListShow.setText(QtGui.QApplication.translate("tstMain", "Show List", None, QtGui.QApplication.UnicodeUTF8))
 
 

Thomas

[bartonc]

Thank you bartonc for the suggestion.

initialFile = [line.split() for line in fileToOpen.readlines()]

works,
as well as my version:

initialFile = [string.split(line) for line in fileToOpen]

but this doesn't cure the problem. I am still not able to select the columns and display them into another table.

Thomas

Yep, sorry to have missed that.

Traceback (most recent call last):
File "D:\Tutorial\main.py", line 48, in showList
variousFuncs.tablePainter(tempSelected, self.ui.tbl_ListShow)
File "D:\Tutorial\variousFuncs.py", line 17, in tablePainter
someTbl.setColumnCount(len(someTwoDimLst[0]))
TypeError: unsubscriptable object

Looking at it again, it seems to me that if there are no columns selected, then you well get this error, but I'm not familiar with QT.

[toglez]

But I do select columns during the runtime by clicking at the top of each one of them.

Thanx anyway, you are very kind...

Thomas.