qt5base-lts/examples/sql/masterdetail/dialog.cpp

293 lines
9.0 KiB
C++
Raw Normal View History

/****************************************************************************
**
** Copyright (C) 2016 The Qt Company Ltd.
** Contact: https://www.qt.io/licensing/
**
** This file is part of the examples of the Qt Toolkit.
**
** $QT_BEGIN_LICENSE:BSD$
** Commercial License Usage
** Licensees holding valid commercial Qt licenses may use this file in
** accordance with the commercial license agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and The Qt Company. For licensing terms
** and conditions see https://www.qt.io/terms-conditions. For further
** information use the contact form at https://www.qt.io/contact-us.
**
** BSD License Usage
** Alternatively, you may use this file under the terms of the BSD license
** as follows:
**
** "Redistribution and use in source and binary forms, with or without
** modification, are permitted provided that the following conditions are
** met:
** * Redistributions of source code must retain the above copyright
** notice, this list of conditions and the following disclaimer.
** * Redistributions in binary form must reproduce the above copyright
** notice, this list of conditions and the following disclaimer in
** the documentation and/or other materials provided with the
** distribution.
** * Neither the name of The Qt Company Ltd nor the names of its
** contributors may be used to endorse or promote products derived
** from this software without specific prior written permission.
**
**
** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
**
** $QT_END_LICENSE$
**
****************************************************************************/
#include "dialog.h"
int uniqueAlbumId;
int uniqueArtistId;
Dialog::Dialog(QSqlRelationalTableModel *albums, QDomDocument details,
QFile *output, QWidget *parent)
: QDialog(parent)
{
model = albums;
albumDetails = details;
outputFile = output;
QGroupBox *inputWidgetBox = createInputWidgets();
QDialogButtonBox *buttonBox = createButtons();
QVBoxLayout *layout = new QVBoxLayout;
layout->addWidget(inputWidgetBox);
layout->addWidget(buttonBox);
setLayout(layout);
setWindowTitle(tr("Add Album"));
}
void Dialog::submit()
{
QString artist = artistEditor->text();
QString title = titleEditor->text();
if (artist.isEmpty() || title.isEmpty()) {
QString message(tr("Please provide both the name of the artist "
"and the title of the album."));
QMessageBox::information(this, tr("Add Album"), message);
} else {
int artistId = findArtistId(artist);
int albumId = addNewAlbum(title, artistId);
QStringList tracks;
tracks = tracksEditor->text().split(',', QString::SkipEmptyParts);
addTracks(albumId, tracks);
increaseAlbumCount(indexOfArtist(artist));
accept();
}
}
int Dialog::findArtistId(const QString &artist)
{
QSqlTableModel *artistModel = model->relationModel(2);
int row = 0;
while (row < artistModel->rowCount()) {
QSqlRecord record = artistModel->record(row);
if (record.value("artist") == artist)
return record.value("id").toInt();
else
row++;
}
return addNewArtist(artist);
}
int Dialog::addNewArtist(const QString &name)
{
QSqlTableModel *artistModel = model->relationModel(2);
QSqlRecord record;
int id = generateArtistId();
QSqlField f1("id", QVariant::Int);
QSqlField f2("artist", QVariant::String);
QSqlField f3("albumcount", QVariant::Int);
f1.setValue(QVariant(id));
f2.setValue(QVariant(name));
f3.setValue(QVariant(0));
record.append(f1);
record.append(f2);
record.append(f3);
artistModel->insertRecord(-1, record);
return id;
}
int Dialog::addNewAlbum(const QString &title, int artistId)
{
int id = generateAlbumId();
QSqlRecord record;
QSqlField f1("albumid", QVariant::Int);
QSqlField f2("title", QVariant::String);
QSqlField f3("artistid", QVariant::Int);
QSqlField f4("year", QVariant::Int);
f1.setValue(QVariant(id));
f2.setValue(QVariant(title));
f3.setValue(QVariant(artistId));
f4.setValue(QVariant(yearEditor->value()));
record.append(f1);
record.append(f2);
record.append(f3);
record.append(f4);
model->insertRecord(-1, record);
return id;
}
void Dialog::addTracks(int albumId, QStringList tracks)
{
QDomElement albumNode = albumDetails.createElement("album");
albumNode.setAttribute("id", albumId);
for (int i = 0; i < tracks.count(); i++) {
QString trackNumber = QString::number(i);
if (i < 10)
trackNumber.prepend('0');
QDomText textNode = albumDetails.createTextNode(tracks.at(i));
QDomElement trackNode = albumDetails.createElement("track");
trackNode.setAttribute("number", trackNumber);
trackNode.appendChild(textNode);
albumNode.appendChild(trackNode);
}
QDomNodeList archive = albumDetails.elementsByTagName("archive");
archive.item(0).appendChild(albumNode);
/*
The following code is commented out since the example uses an in
memory database, i.e., altering the XML file will bring the data
out of sync.
if (!outputFile->open(QIODevice::WriteOnly)) {
return;
} else {
QTextStream stream(outputFile);
archive.item(0).save(stream, 4);
outputFile->close();
}
*/
}
void Dialog::increaseAlbumCount(QModelIndex artistIndex)
{
QSqlTableModel *artistModel = model->relationModel(2);
QModelIndex albumCountIndex;
albumCountIndex = artistIndex.sibling(artistIndex.row(), 2);
int albumCount = albumCountIndex.data().toInt();
artistModel->setData(albumCountIndex, QVariant(albumCount + 1));
}
void Dialog::revert()
{
artistEditor->clear();
titleEditor->clear();
yearEditor->setValue(QDate::currentDate().year());
tracksEditor->clear();
}
QGroupBox *Dialog::createInputWidgets()
{
QGroupBox *box = new QGroupBox(tr("Add Album"));
QLabel *artistLabel = new QLabel(tr("Artist:"));
QLabel *titleLabel = new QLabel(tr("Title:"));
QLabel *yearLabel = new QLabel(tr("Year:"));
QLabel *tracksLabel = new QLabel(tr("Tracks (separated by comma):"));
artistEditor = new QLineEdit;
titleEditor = new QLineEdit;
yearEditor = new QSpinBox;
yearEditor->setMinimum(1900);
yearEditor->setMaximum(QDate::currentDate().year());
yearEditor->setValue(yearEditor->maximum());
yearEditor->setReadOnly(false);
tracksEditor = new QLineEdit;
QGridLayout *layout = new QGridLayout;
layout->addWidget(artistLabel, 0, 0);
layout->addWidget(artistEditor, 0, 1);
layout->addWidget(titleLabel, 1, 0);
layout->addWidget(titleEditor, 1, 1);
layout->addWidget(yearLabel, 2, 0);
layout->addWidget(yearEditor, 2, 1);
layout->addWidget(tracksLabel, 3, 0, 1, 2);
layout->addWidget(tracksEditor, 4, 0, 1, 2);
box->setLayout(layout);
return box;
}
QDialogButtonBox *Dialog::createButtons()
{
QPushButton *closeButton = new QPushButton(tr("&Close"));
QPushButton *revertButton = new QPushButton(tr("&Revert"));
QPushButton *submitButton = new QPushButton(tr("&Submit"));
closeButton->setDefault(true);
connect(closeButton, SIGNAL(clicked()), this, SLOT(close()));
connect(revertButton, SIGNAL(clicked()), this, SLOT(revert()));
connect(submitButton, SIGNAL(clicked()), this, SLOT(submit()));
QDialogButtonBox *buttonBox = new QDialogButtonBox;
buttonBox->addButton(submitButton, QDialogButtonBox::ResetRole);
buttonBox->addButton(revertButton, QDialogButtonBox::ResetRole);
buttonBox->addButton(closeButton, QDialogButtonBox::RejectRole);
return buttonBox;
}
QModelIndex Dialog::indexOfArtist(const QString &artist)
{
QSqlTableModel *artistModel = model->relationModel(2);
for (int i = 0; i < artistModel->rowCount(); i++) {
QSqlRecord record = artistModel->record(i);
if (record.value("artist") == artist)
return artistModel->index(i, 1);
}
return QModelIndex();
}
int Dialog::generateArtistId()
{
uniqueArtistId += 1;
return uniqueArtistId;
}
int Dialog::generateAlbumId()
{
uniqueAlbumId += 1;
return uniqueAlbumId;
}