从 QComboBox 中选择字符串并将 ID 插入 SQLite db
Posted
技术标签:
【中文标题】从 QComboBox 中选择字符串并将 ID 插入 SQLite db【英文标题】:Select string from QComboBox and insert ID into SQLite db 【发布时间】:2020-03-06 01:54:01 【问题描述】:我有一个简单的表单,其中包含一个组合框和一个“保存”按钮。组合框的下拉列表显示:“选择颜色”、“红色”、“绿色”、“蓝色”。 “保存”按钮将用户从组合框中的选择保存到基础数据库表 tbl_palette
。
我还有一个查找表tlkp_colors
,其中包含颜色列表及其各自的 ID。我使用这个查找表来填充组合框。
这个过于简化的例子的完整代码如下所示(可以按原样运行):
import sys
import os.path
import sqlite3
######### Qt DESIGNER AUTO GENERATED CODE ##########
from PyQt5 import QtCore, QtGui, QtWidgets
class Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(407, 240)
self.centralwidget = QtWidgets.QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
self.comboBox = QtWidgets.QComboBox(self.centralwidget)
self.comboBox.setGeometry(QtCore.QRect(100, 60, 201, 22))
self.comboBox.setObjectName("comboBox")
self.pushButton = QtWidgets.QPushButton(self.centralwidget)
self.pushButton.setGeometry(QtCore.QRect(150, 130, 101, 28))
self.pushButton.setObjectName("pushButton")
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtWidgets.QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 407, 26))
self.menubar.setObjectName("menubar")
MainWindow.setMenuBar(self.menubar)
self.statusbar = QtWidgets.QStatusBar(MainWindow)
self.statusbar.setObjectName("statusbar")
MainWindow.setStatusBar(self.statusbar)
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate
MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
self.pushButton.setText(_translate("MainWindow", "Save"))
######### END OF AUTO GENERATED CODE ###############
DIR_PATH = os.path.dirname(os.path.abspath(__file__))
DB_NAME = 'test.db'
target_db = os.path.join(DIR_PATH, DB_NAME)
class Database:
def __init__(self):
# create database 'test.db', add schema, then close it
cnn = sqlite3.connect(target_db)
c = cnn.cursor()
c.executescript("""
DROP TABLE IF EXISTS tlkp_colors;
CREATE TABLE tlkp_colors
(
color_id INTEGER PRIMARY KEY,
color_name TEXT
);
DROP TABLE IF EXISTS tbl_palette;
CREATE TABLE tbl_palette
(
id INTEGER PRIMARY KEY AUTOINCREMENT,
color NUMBER
);
INSERT INTO tlkp_colors VALUES (0, 'Choose color');
INSERT INTO tlkp_colors VALUES (10, 'Red');
INSERT INTO tlkp_colors VALUES (20, 'Green');
INSERT INTO tlkp_colors VALUES (30, 'Blue');
""")
cnn.commit()
cnn.close()
def save(self, your_choice):
data_tuple = (your_choice,)
sqlite_insert_query = """INSERT INTO tbl_palette (color) VALUES (?)"""
db = sqlite3.connect(target_db)
c = db.cursor()
c.execute(sqlite_insert_query, data_tuple )
db.commit()
db.close()
def close_db(self):
target_db.close()
def populate_cbo_from_tlkp_table():
cnn = sqlite3.connect(target_db)
c = cnn.cursor()
c.execute("SELECT color_name FROM tlkp_colors")
list_of_strings = [item[0] for item in c.fetchall()]
cnn.commit()
cnn.close()
return list_of_strings
# comment the following line after the first run (db created already)
db = Database() # create a new instance Database
app = QtWidgets.QApplication(sys.argv)
MainWindow = QtWidgets.QMainWindow()
ui = Ui_MainWindow()
ui.setupUi(MainWindow)
ui.comboBox.addItems(populate_cbo_from_tlkp_table())
ui.pushButton.clicked.connect(lambda : db.save(ui.comboBox.currentIndex()))
MainWindow.show()
sys.exit(app.exec_())
if __name__ == "__main__":
import sys
app = QtWidgets.QApplication(sys.argv)
MainWindow = QtWidgets.QMainWindow()
ui = Ui_MainWindow()
ui.setupUi(MainWindow)
MainWindow.show()
sys.exit(app.exec_())
当我点击“保存”按钮时,我希望将颜色的 id 保存在基础表 tbl_palette
中,而不是颜色本身 - 即我想要 10
而不是 "Red"
。
如果您能解释如何更改代码来做到这一点,我将不胜感激。我在网上搜索过,但到目前为止我所看到的任何东西都没有帮助我解决我的问题。我是 Qt Designer 的新手,我似乎遗漏了一些东西。
【问题讨论】:
从表中获取(color_name, color_id)
并在 for 循环中使用 addItem 来填充组合。然后可以使用currentData 从组合中检索color_id
。
【参考方案1】:
我按照ekhumoro的建议修改了代码:
我换了
def populate_cbo_from_tlkp_table():
cnn = sqlite3.connect(target_db)
c = cnn.cursor()
c.execute("SELECT color_name FROM tlkp_colors")
list_of_strings = [item[0] for item in c.fetchall()]
cnn.commit()
cnn.close()
return list_of_strings
与
def populate_cbo_from_tlkp_table():
cnn = sqlite3.connect(target_db)
c = cnn.cursor()
c.execute("SELECT color_name, color_id FROM tlkp_colors")
rows = c.fetchall()
for row in rows:
ui.comboBox.addItem(str(row[0]), row[1])
print(row)
cnn.commit()
cnn.close()
和
ui.comboBox.addItems(populate_cbo_from_tlkp_table())
ui.pushButton.clicked.connect(lambda : db.save(ui.comboBox.currentIndex()))
与
populate_cbo_from_tlkp_table()
ui.pushButton.clicked.connect(lambda : db.save(ui.comboBox.currentData()))
现在它完全符合我的要求。
【讨论】:
以上是关于从 QComboBox 中选择字符串并将 ID 插入 SQLite db的主要内容,如果未能解决你的问题,请参考以下文章
在 QStyledItemDelegate 中显示 QComboBox 文本而不是索引值