学习编程离不开存取数据，然后在视觉上表现出来，这就涉及到GUI与数据库的操作。对于大多数的数据库而言，都要单独安装运行，然后在程序中建立连接存取数据。不过初学者练手的话，先从简单的开始应该是个不错的选择。

SQLite是一款轻量型的SQL数据库，Python自带，不用单独安装，今天俺就学学用PyQt5的GUI操作SQLite数据库。

目标：建立一个wallet数据库，包含income、cost、member三个表，可以由三个无线单选按钮选定在QTableWidget控件中显示，每个表只有一列数据subject，然后由增加、删除、修改三个功能按钮对数据操作并保存在数据库中。

步骤一、先搭图形框架

这个在前一篇头条文章中已经练习过，直接拿来就行。

# 通过QTableWidget表格控件操作SQLite数据库

import sys
from PyQt5.QtCore import *
from PyQt5.QtWidgets import *

class Table(QWidget):
????def __init__(self,parent=None):
????????super(Table, self).__init__(parent)
????????self.setGeometry(100,100,160,260)
????????self.setGUI()

????def setGUI(self):
????????self.setradioG = QButtonGroup() ???# 第1项选择按钮组
????????self.setradio1 = QRadioButton('收入')
????????self.setradio2 = QRadioButton('支出')
????????self.setradio3 = QRadioButton('成员')

????????# 按钮点击事件与表格刷新功能建立联系
? ? ? ? #self.setradioG.buttonClicked.connect(self.initTable)
????????self.setradio1.setChecked(True)?# 默认按钮1选中
????????self.setradioG.addButton(self.setradio1,0)
????????self.setradioG.addButton(self.setradio2,1)
????????self.setradioG.addButton(self.setradio3,2)
????????self.layout41 = QHBoxLayout()?# 3个按钮水平布局
????????self.layout41.addStretch()
????????self.layout41.addWidget(self.setradio1)
????????self.layout41.addWidget(self.setradio2)
????????self.layout41.addWidget(self.setradio3)
????????self.layout41.addStretch()

????????self.tb2 = QTableWidget(5,1) ?# 创建表格(第2项内容)
????????self.tb2.setStyleSheet('background-color:LightSeaGreen;color:white;')
? ? ? ? self.tb2.verticalHeader().setStyleSheet('background-color:white;color:LightSeaGreen;')
????????# 水平方向拉伸填满表格
????????self.tb2.horizontalHeader().setStretchLastSection(True)
????????# 隐藏水平表头
????????self.tb2.horizontalHeader().setVisible(False)
????????# 设置表格单行选择
????????self.tb2.setSelectionBehavior(QAbstractItemView.SelectRows)
????????#self.initTable()??# 下一步使用
?????????self.setbtSave=QPushButton('修改') ???# 第3项
????????self.setbtIns=QPushButton('增加') ????# 第3项
????????self.setbtDel=QPushButton('删除') ????# 第3项
????????self.setbtSave.clicked.connect(self.on_editButton_clicked) ?# 按钮绑定
????????self.setbtIns.clicked.connect(self.on_insButton_clicked)
????????self.setbtDel.clicked.connect(self.on_delButton_clicked)
????????self.layout43=QHBoxLayout() ????# 按钮水平布局
????????self.layout43.addWidget(self.setbtIns)
????????self.layout43.addWidget(self.setbtDel)
????????self.layout43.addWidget(self.setbtSave)

????????self.layout4 = QVBoxLayout() ???????# 布局垂直整合
????????self.layout4.addLayout(self.layout41)
????????self.layout4.addWidget(self.tb2)
????????self.layout4.addStretch()
????????self.layout4.addLayout(self.layout43)
????????self.layout4.setContentsMargins(0, 0, 0, 0)
????????self.setLayout(self.layout4)
????????self.setWindowFlags(Qt.FramelessWindowHint) ?# 无边框

if __name__ == '__main__':
????app=QApplication(sys.argv)
????table=Table()
????table.show()
????sys.exit(app.exec_())

上面的代码运行后就会显示图形界面，不过按钮还没生效。

步骤二、编写表格显示模块，涉及从数据库中取出数据，插入到表格控件中，如果表格不存在，会先建立库表。下面的模块都插入到上面的主程序中。

def initTable(self): ???# 取出数据库数据填入表格
????????self.tableName=['income','cost','member']
????# 获取数据库数据
????????data_list = self.getData(self.tableName[self.setradioG.checkedId()])
? ? ? ? self.tb2.setRowCount(len(data_list)) ???# 设置表格行数
????????for index in range(len(data_list)): # 数据填入表格
????????????self.tb2.setItem(index, 0, TableWidgetItem(str(data_list[index][0])))

其中self.getData()是从数据库取数据的模块，代码如下：

def getData(name):
????????conn = sqlite3.connect('wallet.db')?# 连接数据库
????????curs = conn.cursor()
????????command = "CREATE TABLE IF NOT EXISTS %s (subject ?TEXT ?PRIMARY KEY ?NOT NULL);"%(name)
????????c = curs.execute(command)
????????command = 'select ROWID,subject from %s'%(name)
????????c = curs.execute(command)
????????data_list = []
????????for row in c :
????????????temp_list = []
????????????temp_list.append(row[1])
????????????data_list.append(temp_list)
????????conn.close()
????????return data_list

这段代码需要导入SQLite数据库包了:

 import sqlite3

这段代码会与数据库wallet建立连接，如果数据库不存在，会自动建立。下面的SQL语句会判断由name传递的表名是否存在，不存在会建立相应的income、cost、member表。

'select ROWID,subject from %s'%(name) 从对应的数据表中取出行号和字段内容，将字段内容赋给列表data_list[]传回给initTable模块，放在表格控件中显示。第一次运行由于数据库中没有内容，表格也没有数据显示。注意：这里的ROWID是数据表中隐含的，不是我们设定的字段，subject是我们建立的字段。取出ROWID在删除、修改字段内容时会用到。

还有一点需要注意的是SQL语句中的参数传递，直接用?占位符很多时候会出问题，特别是数据表名称的传递，绝对不能用?占位符，改用上面的%s进行字符串代换组成SQL命令后再执行就没问题了。

步骤三、给数据表字段增加内容

"增加"按钮是与on_insButton_clicked模块绑定的，因为与"修改"按钮公用一个界面，所以指向同一个模块。

def showDialog(self, status, subject=''): ??# 增加、修改功能对话框
????self.dialog = QDialog(self)
????if status == 1:
????????self.dialog.setWindowTitle('新增')
????else:
????????self.dialog.setWindowTitle('修改')
????group = QGroupBox(self.dialog)
????# 标签和输入框
????lb1 = QLabel('名称:', group)
????self.enter = QLineEdit(group)
????ok_button = QPushButton('确定', self.dialog)
????cancel_button = QPushButton('取消', self.dialog)
????# 标签和按钮控件加到垂直布局
????group_layout = QVBoxLayout()
????group_item = [lb1, self.enter]
????for item in group_item:
????????group_layout.addWidget(item)
????# 将垂直布局添加到groupbox中
????group.setLayout(group_layout)
????group.setFixedSize(group.sizeHint())

????# 两个按钮添加到水平布局中
????button_layout = QHBoxLayout()
????button_layout.addWidget(ok_button)
????button_layout.addWidget(cancel_button)
????# 上面内容加入到垂直布局
????dialog_layout = QVBoxLayout()
????dialog_layout.addWidget(group)
????dialog_layout.addLayout(button_layout)
????# 应用整体布局
????self.dialog.setLayout(dialog_layout)
????# 按传入的状态绑定确定按钮的功能
????if status == 1:
????????ok_button.clicked.connect(self.addAccept)
????else:
????????ok_button.clicked.connect(self.editAccept)
????# 默认选中ok按钮
????ok_button.setDefault(True)
????cancel_button.clicked.connect(self.dialog.reject)
????self.dialog.exec_()
????return False

里面涉及下面两个模块：

def addAccept(self):
????# 输入项不为空允许增加
????if self.enter.text() != '':
????????self.dialog.close() ???????????# 关闭窗口
????????# 在数据库中新增记录
????????self.addData(self.tableName[self.setradioG.checkedId()], self.enter.text())
????????# 刷新表格
????????self.initTable()
????????self.showMsg('新增成功')
????else:
????????self.showMsg('不能为空')

def showMsg(self, message): ???# 消息框
????hint_msg = QMessageBox.about(self,'提示',message)

我这里所有数据表都是只有一个字段，因为是练习做的，所以简单为上，涉及多个字段的话直接加上就行。

今天先做到这里，还有删除模块下次再加上。