以下是一个示例代码,在选择不同的组合框选项时,可以使用该函数将不同的值保存到数据库中:
import sqlite3
def save_to_database(selected_option, value1, value2):
conn = sqlite3.connect('database.db')
c = conn.cursor()
if selected_option == 'Option 1':
table_name = 'table1'
column1_name = 'column1'
column2_name = 'column2'
elif selected_option == 'Option 2':
table_name = 'table2'
column1_name = 'column3'
column2_name = 'column4'
else:
return 'Invalid option selected'
# Insert the values into the database
c.execute(f"INSERT INTO {table_name} ({column1_name}, {column2_name}) VALUES (?, ?)", (value1, value2))
conn.commit()
# Close the connection
conn.close()
在上面的代码中,我们首先导入了sqlite3模块来连接SQLite数据库。我们定义了一个名为“save_to_database”的函数,该函数接受三个参数:选择的选项,值1和值2。选择选项将决定要使用的表名和列名。我们在函数中创建了连接和游标对象,然后使用if-elif语句根据选择的选项确定表名和列名。最后,我们使用execute()方法将值插入到数据库中,并使用commit()方法提交更改。最后,我们关闭了连接对象。
使用该函数的示例代码:
save_to_database('Option 1', 'value1', 'value2')
这将在名为“table1”的表中插入两个值,分别为“value1”和“value2”,并且列名根据选择的选项而不同。
下一篇:编写一个函数将字符c替换为*。