Qt数据库模型

Qt中使用了自己的机制来避免使用SQL语句，为我们提供了更简单的数据库操作及数据显示模型，分别是

• 只读的QSqlQueryModel
• 操作单表的QSqlTableModel
• 可以支持外键的QSqlRelationalTableModel

QSqlTableModel案例

QSqlTableModel项目截图：

         

 部分代码如下：

void consumer::on_saleBtu_clicked()
{
     QString id=ui->countEdit->text();
     int num=id.toInt();
     model2 =new QSqlQueryModel(ui->saletableView);
     model2->setQuery(QString("select tableid,name,num,price from menu,orderToAll where orderToAll.id = menu.id and flag=1 and orderToAll.tableid ='%1'").arg(num));
     ui->saletableView->setModel(model2);
     ui->saletableView->setEditTriggers(QAbstractItemView::NoEditTriggers);
     ui->saletableView->verticalHeader()->setVisible(false);
     ui->saletableView->horizontalHeader()->setResizeMode(QHeaderView::Stretch);
     model2->setHeaderData(0,Qt::Horizontal,"桌号");
     model2->setHeaderData(1,Qt::Horizontal,"食物");
     model2->setHeaderData(2,Qt::Horizontal,"数量");
     model2->setHeaderData(3,Qt::Horizontal,"单价");
     //获取行数
     int sum=0;
     int row=model2->rowCount();   //获取行数
     for(int i=0;i<row;i++)
     {
         int num = model2->record(i).value("num").toInt();
         int price = model2->record(i).value("price").toInt();
         sum+=num*price;
     }
     QString txt=QString::number(sum);
     ui->saleEdit->setText(txt);
}

 创建自定的QSqlQueryModel

      由于QSqlQueryModel默认是只读的，所以在窗口上并不能对表格中的内容进行修改。要想使其可读写，需要自己的类继承自QSqlQueryModel，并且重写setData() 和 flags() 两个函数。如果我们要改变数据的显示，就要重写data() 函数。 

 下面的例子中我们让student表查询结果的id属性列显示红色，name属性列可编辑。
1.向项目中添加新的C++类，类名为MySqlQueryModel，基类为QSqlQueryModel，类型信息选择“继承自QObject”。

2.完成后打开mysqlquerymodel.h文件，在public中添加函数声明：
Qt::ItemFlags flags(const QModelIndex &index) const;
bool setData(const QModelIndex &index, const QVariant &value, int role);
QVariant data(const QModelIndex &item, int role=Qt::DisplayRole) const;
然后添加私有函数声明：
private:
    bool setName(int studentId, const QString &name);
    void refresh();
3.到mysqlquerymodel.cpp文件中，更改如下：

#include "mysqlquerymodel.h"
#include <QSqlQuery>
#include <QColor>
MySqlQueryModel::MySqlQueryModel(QObject *parent) :
    QSqlQueryModel(parent)
{

}
Qt::ItemFlags MySqlQueryModel::flags(
       const QModelIndex &index) const //返回表格是否可更改的标志
{
    Qt::ItemFlags flags = QSqlQueryModel::flags(index);
    if (index.column() == 1) //第二个属性可更改
       flags |= Qt::ItemIsEditable;
    return flags;
}
bool MySqlQueryModel::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(); //获取id号
    clear();
    bool ok;
    if (index.column() == 1) //第二个属性可更改
       ok = setName(id, value.toString());
    refresh();
    return ok;
}
void MySqlQueryModel::refresh() //更新显示
{
    setQuery("select * from student");
    setHeaderData(0, Qt::Horizontal, QObject::tr("id"));
    setHeaderData(1, Qt::Horizontal, QObject::tr("name"));
}

//添加name属性的值
bool MySqlQueryModel::setName(int studentId, const QString &name)
{
    QSqlQuery query;
    query.prepare("update student set name = ? where id = ?");
    query.addBindValue(name);
    query.addBindValue(studentId);
    return query.exec();
}

//更改数据显示样式
QVariant MySqlQueryModel::data(const QModelIndex &index, int role) const
{
      QVariant value = QSqlQueryModel::data(index, role);

      //第一个属性的字体颜色为红色
      if (role == Qt::TextColorRole && index.column() == 0)
      { 
          return qVariantFromValue(QColor(Qt::red));   
      }    
      return value;
}

4.到mainwindow.cpp文件中先添加头文件包含：
#include "mysqlquerymodel.h"

5.更改查询按钮槽内容如下：
void MainWindow::on_pushButton_clicked()
{
    QSqlQueryModel *model = new QSqlQueryModel;
    model->setQuery("select * from student");
    model->setHeaderData(0, Qt::Horizontal, tr("id"));
    model->setHeaderData(1, Qt::Horizontal, tr("name"));
    QTableView *view = new QTableView;
    view->setModel(model);
    view->show(); 
    //创建自己模型的对象
    MySqlQueryModel *myModel = new MySqlQueryModel;    myModel->setQuery("select * from student");
    myModel->setHeaderData(0, Qt::Horizontal, tr("id"));
    myModel->setHeaderData(1, Qt::Horizontal, tr("name"));
    QTableView *view1 = new QTableView;
    view1->setWindowTitle("mySqlQueryModel"); //修改窗口标题
    view1->setModel(myModel);
    view1->show();
}

  自定义QSqlQueryModel实战