8bfebaa22f
Misc code cleanup for the sql examples: - don't include global Qt headers but only needed ones - use proper tr() where possible - pass parameters by const ref - style fixes Change-Id: I4fd4293948918b9d7b373b6d1e8eeecf6f25a622 Reviewed-by: Samuel Gaist <samuel.gaist@idiap.ch>
71 lines
1.8 KiB
C++
71 lines
1.8 KiB
C++
// Copyright (C) 2016 The Qt Company Ltd.
|
|
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
|
|
|
|
#include "editablesqlmodel.h"
|
|
|
|
#include <QSqlQuery>
|
|
|
|
EditableSqlModel::EditableSqlModel(QObject *parent)
|
|
: QSqlQueryModel(parent)
|
|
{
|
|
}
|
|
|
|
//! [0]
|
|
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;
|
|
}
|
|
//! [0]
|
|
|
|
//! [1]
|
|
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;
|
|
}
|
|
//! [1]
|
|
|
|
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"));
|
|
}
|
|
|
|
//! [2]
|
|
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();
|
|
}
|
|
//! [2]
|
|
|
|
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();
|
|
}
|