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

editablesqlmodel.cpp Example File
^{sql/querymodel/editablesqlmodel.cpp}

    /****************************************************************************
    **
    ** Copyright (C) 2004-2006 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();
    }