#include <QtGui>#include <QtSql>#include "artistform.h"#include "generateid.h"ArtistForm::ArtistForm(const QString &name, QWidget *parent)    : QDialog(parent){    model = new QSqlTableModel(this);    model->setTable("artist");    model->setSort(Artist_Name, Qt::AscendingOrder);    model->setHeaderData(Artist_Name, Qt::Horizontal, tr("Name"));    model->setHeaderData(Artist_Country, Qt::Horizontal, tr("Country"));    model->select();    connect(model, SIGNAL(beforeInsert(QSqlRecord &)),            this, SLOT(beforeInsertArtist(QSqlRecord &)));    tableView = new QTableView;    tableView->setModel(model);    tableView->setColumnHidden(Artist_Id, true);    tableView->setSelectionBehavior(QAbstractItemView::SelectRows);    tableView->resizeColumnsToContents();    for (int row = 0; row < model->rowCount(); ++row) {        QSqlRecord record = model->record(row);        if (record.value(Artist_Name).toString() == name) {            tableView->selectRow(row);            break;        }    }    addButton = new QPushButton(tr("&Add"));    deleteButton = new QPushButton(tr("&Delete"));    closeButton = new QPushButton(tr("Close"));    connect(addButton, SIGNAL(clicked()), this, SLOT(addArtist()));    connect(deleteButton, SIGNAL(clicked()),            this, SLOT(deleteArtist()));    connect(closeButton, SIGNAL(clicked()), this, SLOT(accept()));    QHBoxLayout *buttonLayout = new QHBoxLayout;    buttonLayout->addWidget(addButton);    buttonLayout->addWidget(deleteButton);    buttonLayout->addStretch();    buttonLayout->addWidget(closeButton);    QVBoxLayout *mainLayout = new QVBoxLayout;    mainLayout->addWidget(tableView);    mainLayout->addLayout(buttonLayout);    setLayout(mainLayout);    setWindowTitle(tr("Edit Artists"));}void ArtistForm::addArtist(){    int row = model->rowCount();    model->insertRow(row);    QModelIndex index = model->index(row, Artist_Name);    tableView->setCurrentIndex(index);    tableView->edit(index);}void ArtistForm::beforeInsertArtist(QSqlRecord &record){    record.setValue("id", generateId("artist"));}void ArtistForm::deleteArtist(){    tableView->setFocus();    QModelIndex index = tableView->currentIndex();    if (!index.isValid())        return;    QSqlRecord record = model->record(index.row());    QSqlTableModel cdModel;    cdModel.setTable("cd");    cdModel.setFilter("artistid = " + record.value("id").toString());    cdModel.select();    if (cdModel.rowCount() == 0) {        model->removeRow(tableView->currentIndex().row());    } else {        QMessageBox::information(this,                tr("Delete Artist"),                tr("Cannot delete %1 because there are CDs associated "                   "with this artist in the collection.")                .arg(record.value("name").toString()));    }}