QSqlTableModel简单例子-学生信息管理系统

1
#include <QtWidgets>
2
#include <QtSql>
3

4
class StudentManager : public QWidget
5
{
6
    Q_OBJECT
7

8
public:
9
    StudentManager(QWidget *parent = nullptr) : QWidget(parent)
10
    {
11
        initializeModel();
12
        setupUI();
13
        setupConnections();
14
        refreshView();
15
    }
16

17
private slots:
18
    // 添加新学生记录
19
    void onAddButtonClicked()
20
    {
21
        // 获取当前行数，新行将添加在末尾
22
        int row = model->rowCount();
23

24
        // 插入新行
25
        if (model->insertRow(row)) {
26
            // 设置新行的默认数据
27
            model->setData(model->index(row, 1), "新学生");     // name列
28
            model->setData(model->index(row, 2), "男");         // sex列
29
            model->setData(model->index(row, 3), 20);           // age列
30

31
            // 提交到数据库
32
            if (!model->submitAll()) {
33
                QMessageBox::warning(this, "错误", "添加记录失败: " +
34
                                   model->lastError().text());
35
                model->revertAll(); // 撤销更改
36
            } else {
37
                refreshView(); // 刷新视图
38
            }
39
        }
40
    }
41

42
    // 删除选中记录
43
    void onDeleteButtonClicked()
44
    {
45
        // 获取当前选中的行
46
        QModelIndex currentIndex = tableView->currentIndex();
47
        if (!currentIndex.isValid()) {
48
            QMessageBox::warning(this, "警告", "请先选择要删除的记录");
49
            return;
50
        }
51

52
        int row = currentIndex.row();
53

54
        // 显示确认对话框
55
        int ret = QMessageBox::question(this, "确认删除",
56
                                      "确定要删除选中的记录吗？",
57
                                      QMessageBox::Yes | QMessageBox::No);
58

59
        if (ret == QMessageBox::Yes) {
60
            // 删除行
61
            if (model->removeRow(row)) {
62
                // 提交到数据库
63
                if (!model->submitAll()) {
64
                    QMessageBox::warning(this, "错误", "删除记录失败: " +
65
                                       model->lastError().text());
66
                    model->revertAll();
67
                } else {
68
                    refreshView();
69
                }
70
            }
71
        }
72
    }
73

74
    // 提交所有更改
75
    void onSubmitButtonClicked()
76
    {
77
        if (model->submitAll()) {
78
            QMessageBox::information(this, "成功", "所有更改已保存到数据库");
79
        } else {
80
            QMessageBox::warning(this, "错误", "提交失败: " +
81
                               model->lastError().text());
82
        }
83
    }
84

85
    // 撤销所有更改
86
    void onRevertButtonClicked()
87
    {
88
        model->revertAll();
89
        refreshView();
90
        QMessageBox::information(this, "信息", "已撤销所有未保存的更改");
91
    }
92

93
    // 筛选数据
94
    void onFilterChanged()
95
    {
96
        QString filter = filterEdit->text();
97
        if (!filter.isEmpty()) {
98
            // 根据姓名筛选
99
            model->setFilter(QString("name LIKE '%%1%'").arg(filter));
100
        } else {
101
            model->setFilter(""); // 清除筛选条件
102
        }
103
        model->select();
104
        refreshView();
105
    }
106

107
private:
108
    void initializeModel()
109
    {
110
        // 初始化数据库连接
111
        QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE");
112
        db.setDatabaseName("students.db");
113

114
        if (!db.open()) {
115
            QMessageBox::critical(nullptr, "错误", "无法打开数据库");
116
            return;
117
        }
118

119
        // 创建学生表（如果不存在）
120
        QSqlQuery query;
121
        query.exec("CREATE TABLE IF NOT EXISTS student ("
122
                  "id INTEGER PRIMARY KEY AUTOINCREMENT, "
123
                  "name VARCHAR(40) NOT NULL, "
124
                  "sex VARCHAR(10) NOT NULL, "
125
                  "age INTEGER NOT NULL, "
126
                  "mobile VARCHAR(20), "
127
                  "city VARCHAR(40))");
128

129
        // 插入一些测试数据（如果表是空的）
130
        query.exec("SELECT COUNT(*) FROM student");
131
        if (query.next() && query.value(0).toInt() == 0) {
132
            query.exec("INSERT INTO student (name, sex, age, mobile, city) "
133
                      "VALUES ('张三', '男', 20, '13800138000', '北京')");
134
            query.exec("INSERT INTO student (name, sex, age, mobile, city) "
135
                      "VALUES ('李四', '女', 22, '13900139000', '上海')");
136
            query.exec("INSERT INTO student (name, sex, age, mobile, city) "
137
                      "VALUES ('王五', '男', 21, '13700137000', '广州')");
138
        }
139

140
        // 创建数据模型
141
        model = new QSqlTableModel(this, db);
142
        model->setTable("student");
143

144
        // 设置编辑策略：手动提交
145
        model->setEditStrategy(QSqlTableModel::OnManualSubmit);
146

147
        // 设置排序：按ID升序
148
        model->setSort(0, Qt::AscendingOrder);
149

150
        // 设置表头标签
151
        model->setHeaderData(0, Qt::Horizontal, tr("ID"));
152
        model->setHeaderData(1, Qt::Horizontal, tr("姓名"));
153
        model->setHeaderData(2, Qt::Horizontal, tr("性别"));
154
        model->setHeaderData(3, Qt::Horizontal, tr("年龄"));
155
        model->setHeaderData(4, Qt::Horizontal, tr("手机"));
156
        model->setHeaderData(5, Qt::Horizontal, tr("城市"));
157

158
        // 加载数据
159
        model->select();
160
    }
161

162
    void setupUI()
163
    {
164
        // 创建表格视图
165
        tableView = new QTableView;
166
        tableView->setModel(model);
167

168
        // 设置表格属性
169
        tableView->setSelectionMode(QAbstractItemView::SingleSelection);
170
        tableView->setSelectionBehavior(QAbstractItemView::SelectRows);
171
        tableView->setEditTriggers(QAbstractItemView::DoubleClicked |
172
                                 QAbstractItemView::EditKeyPressed);
173

174
        // 设置列宽
175
        tableView->setColumnWidth(0, 60);   // ID
176
        tableView->setColumnWidth(1, 100);  // 姓名
177
        tableView->setColumnWidth(2, 60);   // 性别
178
        tableView->setColumnWidth(3, 60);   // 年龄
179
        tableView->setColumnWidth(4, 120);  // 手机
180
        tableView->setColumnWidth(5, 100);  // 城市
181

182
        // 创建按钮
183
        QPushButton *addButton = new QPushButton(tr("添加"));
184
        QPushButton *deleteButton = new QPushButton(tr("删除"));
185
        QPushButton *submitButton = new QPushButton(tr("提交"));
186
        QPushButton *revertButton = new QPushButton(tr("撤销"));
187

188
        // 创建筛选框
189
        filterEdit = new QLineEdit;
190
        filterEdit->setPlaceholderText(tr("输入姓名进行筛选..."));
191

192
        // 创建按钮布局
193
        QHBoxLayout *buttonLayout = new QHBoxLayout;
194
        buttonLayout->addWidget(addButton);
195
        buttonLayout->addWidget(deleteButton);
196
        buttonLayout->addWidget(submitButton);
197
        buttonLayout->addWidget(revertButton);
198
        buttonLayout->addStretch();
199
        buttonLayout->addWidget(new QLabel(tr("筛选:")));
200
        buttonLayout->addWidget(filterEdit);
201

202
        // 创建主布局
203
        QVBoxLayout *mainLayout = new QVBoxLayout;
204
        mainLayout->addLayout(buttonLayout);
205
        mainLayout->addWidget(tableView);
206

207
        setLayout(mainLayout);
208
        setWindowTitle(tr("学生信息管理系统 - QSqlTableModel示例"));
209
        resize(600, 400);
210

211
        // 连接信号槽
212
        connect(addButton, &QPushButton::clicked, this, &StudentManager::onAddButtonClicked);
213
        connect(deleteButton, &QPushButton::clicked, this, &StudentManager::onDeleteButtonClicked);
214
        connect(submitButton, &QPushButton::clicked, this, &StudentManager::onSubmitButtonClicked);
215
        connect(revertButton, &QPushButton::clicked, this, &StudentManager::onRevertButtonClicked);
216
        connect(filterEdit, &QLineEdit::textChanged, this, &StudentManager::onFilterChanged);
217
    }
218

219
    void setupConnections()
220
    {
221
        // 可以在这里添加其他信号连接
222
    }
223

224
    void refreshView()
225
    {
226
        // 刷新表格显示
227
        tableView->viewport()->update();
228
    }
229

230
private:
231
    QSqlTableModel *model;
232
    QTableView *tableView;
233
    QLineEdit *filterEdit;
234
};
235

236
// 主函数
237
int main(int argc, char *argv[])
238
{
239
    QApplication app(argc, argv);
240

241
    StudentManager manager;
242
    manager.show();
243

244
    return app.exec();
245
}
246

247
#include "main.moc"  // 如果使用单独的.cpp文件，不需要这一行，如果以上代码全部放到一个文件中则需要加，例如全部放到main.cpp文件中