1 /****************************************************************************

     2 **

     3 ** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).

     4 ** All rights reserved.

     5 ** Contact: Nokia Corporation (qt-info@nokia.com)

     6 **

     7 ** This file is part of the examples of the Qt Toolkit.

     8 **

     9 ** $QT_BEGIN_LICENSE:LGPL$

    10 ** No Commercial Usage

    11 ** This file contains pre-release code and may not be distributed.

    12 ** You may use this file in accordance with the terms and conditions

    13 ** contained in the Technology Preview License Agreement accompanying

    14 ** this package.

    15 **

    16 ** GNU Lesser General Public License Usage

    17 ** Alternatively, this file may be used under the terms of the GNU Lesser

    18 ** General Public License version 2.1 as published by the Free Software

    19 ** Foundation and appearing in the file LICENSE.LGPL included in the

    20 ** packaging of this file.  Please review the following information to

    21 ** ensure the GNU Lesser General Public License version 2.1 requirements

    22 ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

    23 **

    24 ** In addition, as a special exception, Nokia gives you certain additional

    25 ** rights.  These rights are described in the Nokia Qt LGPL Exception

    26 ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

    27 **

    28 ** If you have questions regarding the use of this file, please contact

    29 ** Nokia at qt-info@nokia.com.

    30 **

    31 **

    32 **

    33 **

    34 **

    35 **

    36 **

    37 **

    38 ** $QT_END_LICENSE$

    39 **

    40 ****************************************************************************/

    41 

    42 #include <QtSql>

    43 

    44 #include "editablesqlmodel.h"

    45 

    46 EditableSqlModel::EditableSqlModel(QObject *parent)

    47     : QSqlQueryModel(parent)

    48 {

    49 }

    50 

    51 //! [0]

    52 Qt::ItemFlags EditableSqlModel::flags(

    53         const QModelIndex &index) const

    54 {

    55     Qt::ItemFlags flags = QSqlQueryModel::flags(index);

    56     if (index.column() == 1 || index.column() == 2)

    57         flags |= Qt::ItemIsEditable;

    58     return flags;

    59 }

    60 //! [0]

    61 

    62 //! [1]

    63 bool EditableSqlModel::setData(const QModelIndex &index, const QVariant &value, int /* role */)

    64 {

    65     if (index.column() < 1 || index.column() > 2)

    66         return false;

    67 

    68     QModelIndex primaryKeyIndex = QSqlQueryModel::index(index.row(), 0);

    69     int id = data(primaryKeyIndex).toInt();

    70 

    71     clear();

    72 

    73     bool ok;

    74     if (index.column() == 1) {

    75         ok = setFirstName(id, value.toString());

    76     } else {

    77         ok = setLastName(id, value.toString());

    78     }

    79     refresh();

    80     return ok;

    81 }

    82 //! [1]

    83 

    84 void EditableSqlModel::refresh()

    85 {

    86     setQuery("select * from person");

    87     setHeaderData(0, Qt::Horizontal, QObject::tr("ID"));

    88     setHeaderData(1, Qt::Horizontal, QObject::tr("First name"));

    89     setHeaderData(2, Qt::Horizontal, QObject::tr("Last name"));

    90 }

    91 

    92 //! [2]

    93 bool EditableSqlModel::setFirstName(int personId, const QString &firstName)

    94 {

    95     QSqlQuery query;

    96     query.prepare("update person set firstname = ? where id = ?");

    97     query.addBindValue(firstName);

    98     query.addBindValue(personId);

    99     return query.exec();

   100 }

   101 //! [2]

   102 

   103 bool EditableSqlModel::setLastName(int personId, const QString &lastName)

   104 {

   105     QSqlQuery query;

   106     query.prepare("update person set lastname = ? where id = ?");

   107     query.addBindValue(lastName);

   108     query.addBindValue(personId);

   109     return query.exec();

   110 }