Python实现按钮切换SQLite数据库连接的方法求助

我明白你想要实现点击按钮切换数据库的需求，咱们一步步来修改你的代码，让它支持这个功能。

首先，你现有代码里的db_name是类属性，这意味着所有griffo实例都会共享同一个数据库名，咱们得把它改成实例属性，这样每个窗口实例都能独立切换数据库。然后添加切换按钮和对应的处理函数，最后确保查询函数始终使用当前选中的数据库名。

以下是修改后的完整代码：

import sqlite3
from tkinter import Tk, LabelFrame, PhotoImage, Button, Label

class griffo: 
    def __init__(self, window): 
        self.db_name = 'database.db'  # 改为实例属性，每个实例独立维护数据库名
        self.wind = window 
        self.wind.title('GRIFFO HISTORYPEDIA') 
        # 窗口图标设置
        icon = PhotoImage(file='griffoicone.png') 
        self.wind.call('wm', 'iconphoto', self.wind._w, icon) 
        self.wind.configure(background='gray15') 

        # 添加当前数据库状态显示标签
        self.db_status_label = Label(
            self.wind, 
            text=f"当前连接数据库: {self.db_name}", 
            foreground="azure", 
            background='gray15'
        )
        self.db_status_label.grid(row=0, column=0, padx=5, pady=5)

        # 数据库切换按钮
        switch_btn = Button(
            self.wind, 
            text="切换数据库", 
            command=self.switch_database, 
            foreground="black", 
            background="azure"
        )
        switch_btn.grid(row=0, column=1, padx=5, pady=5)

        framename = LabelFrame(
            self.wind, 
            text='', 
            foreground="azure", 
            background='gray15'
        ) 
        framename.grid(row=1, column=0, columnspan=3, pady=5) 

    def switch_database(self):
        # 简单的双数据库切换逻辑，可根据需求扩展
        if self.db_name == 'database.db':
            self.db_name = 'data2.db'
        else:
            self.db_name = 'database.db'
        # 更新界面显示的当前数据库
        self.db_status_label.config(text=f"当前连接数据库: {self.db_name}")
        # 可选：切换后提前创建目标数据库的表（run_query里已包含创建逻辑，这里只是提前触发）
        self.run_query("")

    def run_query(self, query, parameters = ()): 
        # 每次连接使用当前实例的db_name，确保切换生效
        with sqlite3.connect(self.db_name) as conn: 
            cursor = conn.cursor() 
            # 每次连接都检查表是否存在，切换新数据库时自动创建表
            cursor.execute("CREATE TABLE IF NOT EXISTS griffo (id INTEGER PRIMARY KEY , name TEXT, geografia TEXT, economico TEXT, social TEXT, tecnologico TEXT, cultural TEXT, colaborador TEXT)") 
            result = cursor.execute(query, parameters) 
            conn.commit() 
            return result

# 运行示例
if __name__ == "__main__":
    root = Tk()
    app = griffo(root)
    root.mainloop()

关键修改说明： #

• 将db_name改为实例属性：原来的类属性会被所有实例共享，改成self.db_name后，每个窗口实例可以独立管理自己的数据库连接目标。
• 添加切换按钮与处理函数：switch_database方法负责切换数据库名，并实时更新界面上的数据库状态标签，让用户清楚当前连接的是哪个库。你可以根据需求修改切换逻辑，比如改成弹窗让用户输入数据库路径，或者在多个预设数据库间循环切换。
• 确保查询使用当前数据库：run_query方法里的sqlite3.connect(self.db_name)会始终使用最新的self.db_name，所以切换后所有后续查询都会自动连接到新数据库。
• 自动创建表：因为代码保留了CREATE TABLE IF NOT EXISTS语句，切换到新数据库时，第一次执行查询就会自动创建所需的表，无需额外操作。

内容的提问来源于stack exchange，提问作者András Pataki