SqlTableModelDialog.zip

#include "mainwindow.h" #include "ui_mainwindow.h" #include "dialog.h" #include <iostream> #include <QSqlDatabase> #include <QSqlError> #include <QSqlQuery> #include <QSqlRecord> #include <QtSql> #include <QDataWidgetMapper> QSqlDatabase DB; // 数据库连接 QSqlQueryModel *qryModel; // 数据库模型 QSqlTableModel *tabModel; // 数据模型 QItemSelectionModel *theSelection; // 选择模型 QDataWidgetMapper *dataMapper; // 数据映射 MainWindow::MainWindow(QWidget *parent) :QMainWindow(parent),ui(new Ui::MainWindow) { ui->setupUi(this); // 打开数据库 DB=QSqlDatabase::addDatabase("QSQLITE"); // 添加 SQL LITE数据库驱动 DB.setDatabaseName("./database.db"); // 设置数据库名称 if (!DB.open()) { return; } // 打开数据表 qryModel=new QSqlQueryModel(this); theSelection=new QItemSelectionModel(qryModel); qryModel->setQuery("SELECT id,name,sex,age,mobile,city FROM Student order by id"); if (qryModel->lastError().isValid()) { return; } // 设置字段 qryModel->setHeaderData(0,Qt::Horizontal,"ID"); qryModel->setHeaderData(1,Qt::Horizontal,"Name"); qryModel->setHeaderData(2,Qt::Horizontal,"Sex"); qryModel->setHeaderData(3,Qt::Horizontal,"Age"); qryModel->setHeaderData(4,Qt::Horizontal,"Mobile"); qryModel->setHeaderData(5,Qt::Horizontal,"City"); ui->tableView->setModel(qryModel); ui->tableView->setSelectionModel(theSelection); ui->tableView->setSelectionBehavior(QAbstractItemView::SelectRows); ui->tableView->setSelectionMode(QAbstractItemView::SingleSelection); ui->tableView->setAlternatingRowColors(true); // this->setCentralWidget(ui->tableView); } MainWindow::~MainWindow() { delete ui; } // 弹出Dialog对话框,并 更新数据 void MainWindow::on_pushButton_update_clicked() { int curRecNo=theSelection->currentIndex().row(); QSqlRecord curRec=qryModel->record(curRecNo); // 获取当前记录 int uid=curRec.value("id").toInt(); // 获取id QSqlQuery query; // 查询出当前记录的所有字段 query.prepare("select * from Student where id = :ID"); query.bindValue(":ID",uid); query.exec(); query.first(); if (!query.isValid()) // 是否为有效记录 return; curRec=query.record(); // 获取当前记录的数据 Dialog *WindowPtr=new Dialog(this); // 创建对话框 Qt::WindowFlags flags=WindowPtr->windowFlags(); WindowPtr->setWindowFlags(flags | Qt::MSWindowsFixedSizeDialogHint); // 设置对话框固定大小 WindowPtr->setUpdateRecord(curRec); // 调用对话框函数更新数据和界面 int ret=WindowPtr->exec(); // 以模态方式显示对话框 if (ret==QDialog::Accepted) // OK键被按下 { QSqlRecord recData=WindowPtr->getRecordData(); // 获得对话框返回的记录 query.prepare("update Student set name=:Name, sex=:Sex,age=:Age, mobile=:Mobile, city=:City where id=:ID"); query.bindValue(":ID",recData.value("id")); query.bindValue(":Name",recData.value("name")); query.bindValue(":Sex",recData.value("sex")); query.bindValue(":Age",recData.value("age")); query.bindValue(":Mobile",recData.value("mobile")); query.bindValue(":City",recData.value("city")); // 数据模型重新查询数据，更新tableView显示 if (query.exec()) { qryModel->query().exec(); } } delete WindowPtr; } // 增加一条新纪录 void MainWindow::on_pushButton_insert_clicked() { QSqlQuery query; query.exec("select * from Student where id =-1"); // 查询字段信息,是否存在 QSqlRecord curRec=query.record(); // 获取当前记录,实际为空记录 curRec.setValue("id",qryModel->rowCount()+1001); Dialog *WindowPtr=new Dialog(this); Qt::WindowFlags flags=WindowPtr->windowFlags(); WindowPtr->setWindowFlags(flags | Qt::MSWindowsFixedSizeDialogHint); // 设置对话框固定大小 WindowPtr->setInsertRecord(curRec); // 插入记录 int ret=WindowPtr->exec(); // 以模态方式显示对话框 if (ret==QDialog::Accepted) // OK键被按下 { QSqlRecord recData=WindowPtr->getRecordData(); query.prepare("INSERT INTO Student(id,name,sex,age,mobile,city)" " VALUES(:Id, :Name, :Sex, :Age, :Mobile, :City)"); query.bindValue(":Id",recData.value("id")); query.bindValue(":Name",recData.value("name")); query.bindValue(":Sex",recData.value("sex")); query.bindValue(":Age",recData.value("age")); query.bindValue(":Mobile",recData.value("mobile")); query.bindValue(":City",recData.value("city")); if (query.exec()) { QString sqlStr=qryModel->query().executedQuery(); // 执行过的SELECT语句 qryModel->setQuery(sqlStr); // 重新查询数据 } } delete WindowPtr; } // 删除选中记录 void MainWindow::on_pushButton_delete_clicked() { int curRecNo=theSelection->currentIndex().row(); QSqlRecord curRec=qryModel->record(curRecNo); // 获取当前记录 if (curRec.isEmpty()) // 当前为空记录 return; int uid=curRec.value("id").toInt(); // 获取员工编号 QSqlQuery query; query.prepare("delete from Student where id = :ID"); query.bindValue(":ID",uid); if (query.exec()) { QString sqlStr=qryModel->query().executedQuery(); // 执行过的SELECT语句 qryModel->setQuery(sqlStr); // 重新查询数据 } } // 将所有人的年龄增加1岁 void MainWindow::on_pushButton_AddAge_clicked() { QSqlQuery qryEmpList; // 员工工资信息列表 qryEmpList.exec("SELECT id,age FROM Student ORDER BY id"); qryEmpList.first(); QSqlQuery qryUpdate; // 临时QSqlQuery qryUpdate.prepare("UPDATE Student SET age=:Age WHERE id = :ID"); while (qryEmpList.isValid()) // 当前记录有效 { int uid=qryEmpList.value("id").toInt(); // 获取uid int uage=qryEmpList.value("age").toInt(); // 获取age uage=uage+1; // 年龄+1 qryUpdate.bindValue(":ID",uid); qryUpdate.bindValue(":Age",uage); // 设置SQL语句参数 qryUpdate.exec(); // 执行update if (!qryEmpList.next()) // 移动到下一条记录，并判断是否到末尾了 break; } // 数据模型重新查询数据，更新tableView的显示 qryModel->query().exec(); }