Qt 4.0: editablesqlmodel.cpp Example File (sql/querymodel/editablesqlmodel.cpp)

|  | Home · All Classes · Main Classes · Grouped Classes · Modules · Functions |

# editablesqlmodel.cpp Example File
sql/querymodel/editablesqlmodel.cpp

        /****************************************************************************
        **
        ** Copyright (C) 2004-2005 Trolltech AS. All rights reserved.
        **
        ** This file is part of the documentation of the Qt Toolkit.
        **
        ** This file may be used under the terms of the GNU General Public
    ** License version 2.0 as published by the Free Software Foundation
    ** and appearing in the file LICENSE.GPL included in the packaging of
    ** this file.  Please review the following information to ensure GNU
    ** General Public Licensing requirements will be met:
    ** http://www.trolltech.com/products/qt/opensource.html
    **
    ** If you are unsure which license is appropriate for your use, please
    ** review the following information:
    ** http://www.trolltech.com/products/qt/licensing.html or contact the
    ** sales department at sales@trolltech.com.
        **
        ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
        ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
        **
        ****************************************************************************/

        #include <QtSql>

        #include "editablesqlmodel.h"

        EditableSqlModel::EditableSqlModel(QObject *parent)
            : QSqlQueryModel(parent)
        {
        }

        Qt::ItemFlags EditableSqlModel::flags(
                const QModelIndex &index) const
        {
            Qt::ItemFlags flags = QSqlQueryModel::flags(index);
            if (index.column() == 1 || index.column() == 2)
                flags |= Qt::ItemIsEditable;
            return flags;
        }

        bool EditableSqlModel::setData(const QModelIndex &index, const QVariant &value, int /* role */)
        {
            if (index.column() < 1 || index.column() > 2)
                return false;

            QModelIndex primaryKeyIndex = QSqlQueryModel::index(index.row(), 0);
            int id = data(primaryKeyIndex).toInt();

            clear();

            bool ok;
            if (index.column() == 1) {
                ok = setFirstName(id, value.toString());
            } else {
                ok = setLastName(id, value.toString());
            }
            refresh();
            return ok;
        }

        void EditableSqlModel::refresh()
        {
            setQuery("select * from person");
            setHeaderData(0, Qt::Horizontal, QObject::tr("ID"));
            setHeaderData(1, Qt::Horizontal, QObject::tr("First name"));
            setHeaderData(2, Qt::Horizontal, QObject::tr("Last name"));
        }

        bool EditableSqlModel::setFirstName(int personId, const QString &firstName)
        {
            QSqlQuery query;
            query.prepare("update person set firstname = ? where id = ?");
            query.addBindValue(firstName);
            query.addBindValue(personId);
            return query.exec();
        }

        bool EditableSqlModel::setLastName(int personId, const QString &lastName)
        {
            QSqlQuery query;
            query.prepare("update person set lastname = ? where id = ?");
            query.addBindValue(lastName);
            query.addBindValue(personId);
            return query.exec();
        }