学习编程离不开存取数据,然后在视觉上表现出来,这就涉及到GUI与数据库的操作。对于大多数的数据库而言,都要单独安装运行,然后在程序中建立连接存取数据。不过初学者练手的话,先从简单的开始应该是个不错的选择。
SQLite是一款轻量型的SQL数据库,Python自带,不用单独安装,今天俺就学学用PyQt5的GUI操作SQLite数据库。
目标:建立一个wallet数据库,包含income、cost、member三个表,可以由三个无线单选按钮选定在QTableWidget控件中显示,每个表只有一列数据subject,然后由增加、删除、修改三个功能按钮对数据操作并保存在数据库中。
步骤一、先搭图形框架
这个在前一篇头条文章中已经练习过,直接拿来就行。
<code> 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()self
.setradio1 = QRadioButton('收入'
)self
.setradio2 = QRadioButton('支出'
)self
.setradio3 = QRadioButton('成员'
)self
.setradio1.setChecked(True)self
.setradioG.addButton(self
.setradio1,0
)self
.setradioG.addButton(self
.setradio2,1
)self
.setradioG.addButton(self
.setradio3,2
)self
.layout41 = QHBoxLayout()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
)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
.setbtSave=QPushButton('修改'
)self
.setbtIns=QPushButton('增加'
)self
.setbtDel=QPushButton('删除'
)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_
())/<code>
上面的代码运行后就会显示图形界面,不过按钮还没生效。
步骤二、编写表格显示模块,涉及从数据库中取出数据,插入到表格控件中,如果表格不存在,会先建立库表。下面的模块都插入到上面的主程序中。
<code>def
initTable
(
self
):self
.tableName=['income'
,'cost'
,'member'
] data_list =self
.getData(self
.tableName[self
.setradioG.checkedId()])self
.tb2.setRowCount(len(data_list))for
indexin
range(len(data_list)):self
.tb2.setItem(index,0
, TableWidgetItem(str(data_list[index][0
])))/<code>
其中self.getData()是从数据库取数据的模块,代码如下:
<code>def getData(name): conn = sqlite3.connect('wallet.db') curs = conn.cursor() command = "CREATE
TABLE
IF
NOT
EXISTS
%s (subjectTEXT
PRIMARYKEY
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/<code>
这段代码需要导入SQLite数据库包了:
<code>import
sqlite3/<code>
这段代码会与数据库wallet建立连接,如果数据库不存在,会自动建立。下面的SQL语句会判断由name传递的表名是否存在,不存在会建立相应的income、cost、member表。
'select ROWID,subject from %s'%(name) 从对应的数据表中取出行号和字段内容,将字段内容赋给列表data_list[]传回给initTable模块,放在表格控件中显示。第一次运行由于数据库中没有内容,表格也没有数据显示。注意:这里的ROWID是数据表中隐含的,不是我们设定的字段,subject是我们建立的字段。取出ROWID在删除、修改字段内容时会用到。
还有一点需要注意的是SQL语句中的参数传递,直接用?占位符很多时候会出问题,特别是数据表名称的传递,绝对不能用?占位符,改用上面的%s进行字符串代换组成SQL命令后再执行就没问题了。
步骤三、给数据表字段增加内容
"增加"按钮是与on_insButton_clicked模块绑定的,因为与"修改"按钮公用一个界面,所以指向同一个模块。
<code>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) 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_button.setDefault(True
) cancel_button.clicked.connect(self
.dialog.reject)self
.dialog.exec_()return
False
/<code>
里面涉及下面两个模块:
<code>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('不能为空'
)/<code>
<code>def
showMsg
(
self
, message): hint_msg = QMessageBox.about(self
,'提示'
,message)/<code>
我这里所有数据表都是只有一个字段,因为是练习做的,所以简单为上,涉及多个字段的话直接加上就行。
今天先做到这里,还有删除模块下次再加上。