|  | Home · All Classes · Main Classes · Annotated · Grouped Classes · 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 distributed under the terms of the Q Public License
    ** as defined by Trolltech AS of Norway and appearing in the file
    ** LICENSE.QPL included in the packaging of this file.
    **
    ** This file may be distributed and/or modified under the terms of the
    ** GNU General Public License version 2 as published by the Free Software
    ** Foundation and appearing in the file LICENSE.GPL included in the
    ** packaging of this file.
    **
    ** See http://www.trolltech.com/pricing.html or email sales@trolltech.com for
    **   information about Qt Commercial License Agreements.
    ** See http://www.trolltech.com/qpl/ for QPL licensing information.
    ** See http://www.trolltech.com/gpl/ for GPL licensing information.
    **
    ** Contact info@trolltech.com if any conditions of this licensing are
    ** not clear to you.
        **
        ** 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();
        }