1/****************************************************************************
2**
3** Copyright (C) 2016 The Qt Company Ltd.
4** Contact: https://www.qt.io/licensing/
5**
6** This file is part of the examples of the Qt Toolkit.
7**
8** $QT_BEGIN_LICENSE:BSD$
9** Commercial License Usage
10** Licensees holding valid commercial Qt licenses may use this file in
11** accordance with the commercial license agreement provided with the
12** Software or, alternatively, in accordance with the terms contained in
13** a written agreement between you and The Qt Company. For licensing terms
14** and conditions see https://www.qt.io/terms-conditions. For further
15** information use the contact form at https://www.qt.io/contact-us.
16**
17** BSD License Usage
18** Alternatively, you may use this file under the terms of the BSD license
19** as follows:
20**
21** "Redistribution and use in source and binary forms, with or without
22** modification, are permitted provided that the following conditions are
23** met:
24** * Redistributions of source code must retain the above copyright
25** notice, this list of conditions and the following disclaimer.
26** * Redistributions in binary form must reproduce the above copyright
27** notice, this list of conditions and the following disclaimer in
28** the documentation and/or other materials provided with the
29** distribution.
30** * Neither the name of The Qt Company Ltd nor the names of its
31** contributors may be used to endorse or promote products derived
32** from this software without specific prior written permission.
33**
34**
35** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
36** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
37** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
38** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
39** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
40** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
41** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
42** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
43** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
44** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
45** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
46**
47** $QT_END_LICENSE$
48**
49****************************************************************************/
50
51#include <QtSql>
52
53#include "editablesqlmodel.h"
54
55EditableSqlModel::EditableSqlModel(QObject *parent)
56 : QSqlQueryModel(parent)
57{
58}
59
60//! [0]
61Qt::ItemFlags EditableSqlModel::flags(
62 const QModelIndex &index) const
63{
64 Qt::ItemFlags flags = QSqlQueryModel::flags(index);
65 if (index.column() == 1 || index.column() == 2)
66 flags |= Qt::ItemIsEditable;
67 return flags;
68}
69//! [0]
70
71//! [1]
72bool EditableSqlModel::setData(const QModelIndex &index, const QVariant &value, int /* role */)
73{
74 if (index.column() < 1 || index.column() > 2)
75 return false;
76
77 QModelIndex primaryKeyIndex = QSqlQueryModel::index(index.row(), 0);
78 int id = data(primaryKeyIndex).toInt();
79
80 clear();
81
82 bool ok;
83 if (index.column() == 1) {
84 ok = setFirstName(id, value.toString());
85 } else {
86 ok = setLastName(id, value.toString());
87 }
88 refresh();
89 return ok;
90}
91//! [1]
92
93void EditableSqlModel::refresh()
94{
95 setQuery("select * from person");
96 setHeaderData(0, Qt::Horizontal, QObject::tr("ID"));
97 setHeaderData(1, Qt::Horizontal, QObject::tr("First name"));
98 setHeaderData(2, Qt::Horizontal, QObject::tr("Last name"));
99}
100
101//! [2]
102bool EditableSqlModel::setFirstName(int personId, const QString &firstName)
103{
104 QSqlQuery query;
105 query.prepare("update person set firstname = ? where id = ?");
106 query.addBindValue(firstName);
107 query.addBindValue(personId);
108 return query.exec();
109}
110//! [2]
111
112bool EditableSqlModel::setLastName(int personId, const QString &lastName)
113{
114 QSqlQuery query;
115 query.prepare("update person set lastname = ? where id = ?");
116 query.addBindValue(lastName);
117 query.addBindValue(personId);
118 return query.exec();
119}
120