Search code examples
pythonqtprintingpysideqtableview

How to print a QTableView


I have a Python and PySide app that connects to a mysql database and displays the results of a query in a QTableView. I need to print the contents of the table view. Here's some code:

    self.db_table = QtGui.QTableView(self)
    self.model =  QtSql.QSqlQueryModel()
    self.model.setQuery("SELECT * FROM simpsons")
    self.model.setHeaderData(1, QtCore.Qt.Horizontal, self.tr("First Name"))
    self.model.setHeaderData(2, QtCore.Qt.Horizontal, self.tr("Last Name"))
    self.db_table.setModel(self.model) 

    self.print_btn = QtGui.QPushButton("Print")
    self.print_btn.clicked.connect(self.print_btn_clicked)

    def print_btn_clicked(self):
        printDialog = QtGui.QPrintDialog(self.printer, self)
        if printDialog.exec_() == QtGui.QDialog.Accepted:
         #printing code

I can't find an example for this and I don't understand much from the documentation so I'd appreciate some help


Solution

  • One way to do it is to dump the table contents into a QTextDocument, and then print that.

    The following demo uses a simple text-table, but html could be used to get more sophisticated formatting:

    from PyQt4 import QtGui, QtCore
    
    class Window(QtGui.QWidget):
        def __init__(self, rows, columns):
            QtGui.QWidget.__init__(self)
            self.table = QtGui.QTableView(self)
            model =  QtGui.QStandardItemModel(rows, columns, self.table)
            for row in range(rows):
                for column in range(columns):
                    item = QtGui.QStandardItem('(%d, %d)' % (row, column))
                    item.setTextAlignment(QtCore.Qt.AlignCenter)
                    model.setItem(row, column, item)
            self.table.setModel(model)
            self.buttonPrint = QtGui.QPushButton('Print', self)
            self.buttonPrint.clicked.connect(self.handlePrint)
            self.buttonPreview = QtGui.QPushButton('Preview', self)
            self.buttonPreview.clicked.connect(self.handlePreview)
            layout = QtGui.QGridLayout(self)
            layout.addWidget(self.table, 0, 0, 1, 2)
            layout.addWidget(self.buttonPrint, 1, 0)
            layout.addWidget(self.buttonPreview, 1, 1)
    
        def handlePrint(self):
            dialog = QtGui.QPrintDialog()
            if dialog.exec_() == QtGui.QDialog.Accepted:
                self.handlePaintRequest(dialog.printer())
    
        def handlePreview(self):
            dialog = QtGui.QPrintPreviewDialog()
            dialog.paintRequested.connect(self.handlePaintRequest)
            dialog.exec_()
    
        def handlePaintRequest(self, printer):
            document = QtGui.QTextDocument()
            cursor = QtGui.QTextCursor(document)
            model = self.table.model()
            table = cursor.insertTable(
                model.rowCount(), model.columnCount())
            for row in range(table.rows()):
                for column in range(table.columns()):
                    cursor.insertText(model.item(row, column).text())
                    cursor.movePosition(QtGui.QTextCursor.NextCell)
            document.print_(printer)
    
    if __name__ == '__main__':
    
        import sys
        app = QtGui.QApplication(sys.argv)
        window = Window(25, 2)
        window.resize(300, 400)
        window.show()
        sys.exit(app.exec_())