Qt SQL示例：Books（图书评级）

这个示例展示将 Qt 的 SQL 类与模型/视图框架一起使用的方法。

书籍的信息、作者的信息、图书类型的信息保存在数据库的不同表中。

自定义委托实现鼠标按下为图书评级。

构建在内存中的数据库表并插入数据：

#ifndef INITDB_H
#define INITDB_H
 
#include 
 
void addBook(QSqlQuery &q, const QString & 书名, int 年份,
             const QVariant & 作者ID, const QVariant & 类型ID, int 星级)
{
    q.addBindValue(书名);
    q.addBindValue(年份);
    q.addBindValue(作者ID);
    q.addBindValue(类型ID);
    q.addBindValue(星级);
    q.exec();
}
 
QVariant addGenre(QSqlQuery &q, const QString &作品类型名称)
{
    q.addBindValue(作品类型名称);
    q.exec();
    return q.lastInsertId();
}
 
QVariant addAuthor(QSqlQuery &q, const QString &作者名称, QDate 作者生日)
{
    q.addBindValue(作者名称);
    q.addBindValue(作者生日);
    q.exec();
    return q.lastInsertId();
}
 
const auto BOOKS_SQL = R"(create table 书籍(id integer primary key,
                                           书名 varchar,
                                           作者 integer,
                                           类型 integer,
                                           年份 integer,
                                           星级 integer))";
 
const auto AUTHORS_SQL =  R"(create table 作家(id integer primary key,
                                               姓名 varchar,
                                               生日 date))";
 
const auto GENRES_SQL = R"(create table 作品类型(id integer primary key,
                                                类型名称 varchar))";
 
const auto INSERT_AUTHOR_SQL = R"(insert into 作家(姓名, 生日)
                                              values(?, ?))";
 
const auto INSERT_BOOK_SQL = R"(insert into 书籍(书名, 年份, 作者, 类型, 星级)
                                            values(?, ?, ?, ?, ?))";
 
const auto INSERT_GENRE_SQL = R"(insert into 作品类型(类型名称)
                                             values(?))";
 
QSqlError initDb()
{
    QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE");
    db.setDatabaseName(":memory:");
 
    if (!db.open())
        return db.lastError();
 
    QStringList tables = db.tables();
    if (tables.contains("书籍", Qt::CaseInsensitive)
        && tables.contains("作家", Qt::CaseInsensitive))
        return QSqlError();
 
    QSqlQuery q;
    if (!q.exec(BOOKS_SQL))
        return q.lastError();
    if (!q.exec(AUTHORS_SQL))
        return q.lastError();
    if (!q.exec(GENRES_SQL))
        return q.lastError();
 
    if (!q.prepare(INSERT_AUTHOR_SQL))
        return q.lastError();
 
    QVariant asimovId = addAuthor(q, QLatin1String("艾萨克.阿西莫夫").data(), QDate(1920, 2, 1));
    QVariant greeneId = addAuthor(q, QLatin1String("格雷厄姆·格林").data(), QDate(1904, 10, 2));
    QVariant pratchettId = addAuthor(q, QLatin1String("特里·普拉切特").data(), QDate(1948, 4, 28));
 
    if (!q.prepare(INSERT_GENRE_SQL))
        return q.lastError();
 
    QVariant sfiction = addGenre(q, QLatin1String("科幻小说").data());
    QVariant fiction = addGenre(q, QLatin1String("幽默讽刺小说").data());
    QVariant fantasy = addGenre(q, QLatin1String("奇幻小说").data());
 
    if (!q.prepare(INSERT_BOOK_SQL))
        return q.lastError();
 
    addBook(q, QLatin1String("基地").data(), 1951, asimovId, sfiction, 3);
    addBook(q, QLatin1String("基地与帝国").data(), 1952, asimovId, sfiction, 4);
    addBook(q, QLatin1String("第二基地").data(), 1953, asimovId, sfiction, 3);
    addBook(q, QLatin1String("基金边缘").data(), 1982, asimovId, sfiction, 3);
    addBook(q, QLatin1String("基地与地球").data(), 1986, asimovId, sfiction, 4);
    addBook(q, QLatin1String("基地前奏").data(), 1988, asimovId, sfiction, 3);
    addBook(q, QLatin1String("迈向基地").data(), 1993, asimovId, sfiction, 3);
    addBook(q, QLatin1String("权力与荣耀").data(), 1940, greeneId, fiction, 4);
    addBook(q, QLatin1String("第三个人").data(), 1950, greeneId, fiction, 5);
    addBook(q, QLatin1String("我们在哈瓦那的人").data(), 1958, greeneId, fiction, 4);
    addBook(q, QLatin1String("卫兵！卫兵！").data(), 1989, pratchettId, fantasy, 3);
    addBook(q, QLatin1String("守夜人").data(), 2002, pratchettId, fantasy, 3);
    addBook(q, QLatin1String("邮局奇遇记").data(), 2004, pratchettId, fantasy, 3);
 
    return QSqlError();
}
 
#endif

构建了书籍、作家、作品类型三个数据表并插入相应的数据。

简单的UI界面：

使用模型显示数据库中的数据：

BookWindow::BookWindow()
{
    ui.setupUi(this);
 
    if (!QSqlDatabase::drivers().contains("QSQLITE"))
        QMessageBox::critical(this,"无法加载数据库","此示例需要 SQLITE 驱动程序");
 
    QSqlError err = initDb();
    if (err.type() != QSqlError::NoError)
    {
        QMessageBox::critical(this, "无法初始化数据库","错误信息：" + err.text());
        return;
    }
 
    model = new QSqlRelationalTableModel(ui.bookTable);
    model->setEditStrategy(QSqlTableModel::OnManualSubmit);
    model->setTable("书籍");
 
    //“书籍”表中相关索引
    authorIdx = model->fieldIndex("作者");
    genreIdx = model->fieldIndex("类型");
 
    //设置与其他数据库表的关系：
    model->setRelation(authorIdx, QSqlRelation("作家", "id", "姓名"));
    model->setRelation(genreIdx, QSqlRelation("作品类型", "id", "类型名称"));
 
    //设置本地化的标题
    model->setHeaderData(authorIdx, Qt::Horizontal, tr("_作者_"));
    model->setHeaderData(genreIdx, Qt::Horizontal, tr("_类型_"));
    model->setHeaderData(model->fieldIndex("书名"),Qt::Horizontal, tr("_书名_"));
    model->setHeaderData(model->fieldIndex("年份"), Qt::Horizontal, tr("_年份_"));
    model->setHeaderData(model->fieldIndex("星级"),Qt::Horizontal, tr("_星级_"));
 
    if (!model->select())
    {
        QMessageBox::critical(this, "无法初始化数据库","错误信息：" + model->lastError().text());
        return;
    }
 
    //设置模型并隐藏 ID 列：
    ui.bookTable->setModel(model);
    ui.bookTable->setItemDelegate(new BookDelegate(ui.bookTable));
    ui.bookTable->setColumnHidden(model->fieldIndex("id"), true);
    ui.bookTable->setSelectionMode(QAbstractItemView::SingleSelection);
 
    //使用作者姓名填充QComboBox
    ui.authorEdit->setModel(model->relationModel(authorIdx));
    ui.authorEdit->setModelColumn(model->relationModel(authorIdx)->fieldIndex("姓名"));
 
    //使用小说类型填充QComboBox
    ui.genreEdit->setModel(model->relationModel(genreIdx));
    ui.genreEdit->setModelColumn(model->relationModel(genreIdx)->fieldIndex("类型名称"));
 
    //此列拉伸
    ui.bookTable->horizontalHeader()->setSectionResizeMode(model->fieldIndex("星级"),QHeaderView::Stretch);
 
    auto mapper = new QDataWidgetMapper(this);
    mapper->setModel(model);
    mapper->setItemDelegate(new BookDelegate(this));
    mapper->addMapping(ui.titleEdit, model->fieldIndex("书名"));
    mapper->addMapping(ui.yearEdit, model->fieldIndex("年份"));
    mapper->addMapping(ui.authorEdit, authorIdx);
    mapper->addMapping(ui.genreEdit, genreIdx);
    mapper->addMapping(ui.ratingEdit, model->fieldIndex("星级"));
 
    ui.ratingEdit->setSuffix(" 星");
 
    connect(ui.bookTable->selectionModel(),&QItemSelectionModel::currentRowChanged,
            mapper,&QDataWidgetMapper::setCurrentModelIndex);
 
    ui.bookTable->setCurrentIndex(model->index(0, 0));
}

使用的是 QSqlRelationalTableModel 模型，它可为单个数据库表提供可编辑的数据模型，并允许将列设置为其他数据库表的外键。

设置此模型显示 “书籍” 表中的数据，并设置：

• “书籍” 表中的 “作者” 列（作家的id）对应 “作家” 表中的（id），显示为作家的姓名
• “书籍” 表中的 “类型” 列（小说类型id）对应 “作品类型” 表中的（id），显示为作品的类型名称

    //“书籍”表中相关索引
    authorIdx = model->fieldIndex("作者");
    genreIdx = model->fieldIndex("类型");
 
    //设置与其他数据库表的关系：
    model->setRelation(authorIdx, QSqlRelation("作家", "id", "姓名"));
    model->setRelation(genreIdx, QSqlRelation("作品类型", "id", "类型名称"));

最后使用 QDataWidgetMapper 使模型和小部件之间建立对应关系，它可以用于显示模型中的数据并将小部件中数据的更改反馈到模型。

关于第五列的星形委托见：自定义委托（星形委托）

涉及到的知识点和类：

• QSqlQuery
• QSqlDatabase
• QSqlRelationalTableModel
• QSqlRelation
• QDataWidgetMapper
• QSqlRelationalDelegate
• 自定义委托