从QComboBox中选择字符串,然后将ID插入SQLite数据库

我有一个简单的表单,其中包含一个组合框和一个“保存”按钮。组合框的下拉列表显示:“选择颜色”,“红色”,“绿色”,“蓝色”。 “保存”按钮将用户从组合框中的选择保存到基础数据库表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,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 NOT {{1 }}。

如果您能解释如何更改代码来做到这一点,我将非常感谢。我已经在网上搜索过,但是到目前为止,没有任何东西可以帮助我解决问题。我是Qt Designer的新手,似乎丢失了一些东西。

apple123450 回答:从QComboBox中选择字符串,然后将ID插入SQLite数据库

我根据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()))

现在它完全可以执行我想要的操作了。

本文链接:https://www.f2er.com/3131332.html

大家都在问