Tkinter应用中如何将已打开的CSV文件名传递给保存函数？

核心问题分析 #

你的代码主要问题在于全局变量使用不当以及缺少对窗口实例变量的正确引用。嵌套函数command_open和command_new中对database的赋值未正确修改全局变量，同时MainPage中的savetodb函数无法获取正确的文件路径。此外，ent_track的创建方式导致无法获取输入值，保存时使用w模式会覆盖整个文件而非追加数据。

解决方案步骤 #

1. 使用窗口实例变量存储文件路径 #

将全局变量database替换为windows类的实例变量self.database，让所有组件通过控制器（即windows实例）访问当前数据库路径。

2. 修正菜单命令中的路径赋值 #

在command_open和command_new中直接设置self.database，并处理用户取消对话框的情况。

3. 修复输入框的引用问题 #

将ent_track的创建与布局分离，确保能获取输入值。

4. 在保存函数中通过控制器访问路径 #

在MainPage的savetodb函数中，通过controller.database获取文件路径，添加错误检查（如未选择数据库、输入为空），同时使用追加模式(a)保存数据。

修正后的完整代码 #

import tkinter as tk
from tkinter import *
from tkinter import ttk
from math import *
import csv
from tkinter import filedialog as fd
from tkinter.filedialog import asksaveasfile
from tkinter.messagebox import showinfo
from datetime import datetime

class windows(tk.Tk):
    def __init__(self, *args, **kwargs):
        tk.Tk.__init__(self, *args, **kwargs)
        self.wm_title("calculator and database")
        self.database = None  # 用实例变量存储数据库路径
        
        tire_menu = Menu(self)
        self.config(menu=tire_menu)

        def command_open():
            filename = fd.askopenfilename(
                title='Select database',
                filetypes=[("CSV files", "*.csv")])
            if filename:  # 用户未取消选择
                showinfo(
                    title='You have selected',
                    message=filename)
                self.database = filename

        def command_new():
            extensions = [("csv file(*.csv)", "*.csv")]
            file = asksaveasfile(filetypes=extensions,
                                defaultextension=extensions)
            if file:  # 用户未取消创建
                headerlist = [
                    "Date",
                    "Track"
                ]
                self.database = file.name
                file.close()  # 关闭由asksaveasfile打开的文件
                with open(self.database, "w", newline='') as f:
                    writer = csv.writer(f)
                    writer.writerow(headerlist)
                showinfo(title="Success", message="New database created!")

        # 创建菜单
        file_menu = Menu(tire_menu)
        db_menu = Menu(tire_menu)

        tire_menu.add_cascade(label="File", menu=file_menu)
        file_menu.add_command(label="Exit", command=self.quit)

        tire_menu.add_cascade(label="Database", menu=db_menu)
        db_menu.add_command(label="New DB", command=command_new)
        db_menu.add_command(label="Open DB", command=command_open)

        # 创建容器框架
        container = tk.Frame(self, height=600, width=800)
        container.pack(side="top", fill="both", expand=True)
        container.grid_rowconfigure(0, weight=1)
        container.grid_columnconfigure(0, weight=1)

        # 初始化所有页面
        self.frames = {}
        for F in (MainPage, SidePage, CompletionScreen):
            frame = F(container, self)
            self.frames[F] = frame
            frame.grid(row=0, column=0, sticky="nsew")

        self.show_frame(MainPage)

    def show_frame(self, cont):
        frame = self.frames[cont]
        frame.tkraise()


class MainPage(tk.Frame):
    def __init__(self, parent, controller):
        tk.Frame.__init__(self, parent)
        self.controller = controller  # 保存控制器引用
        
        title_main = tk.Label(
            self,
            text="calculator")
        title_main.config(font=("Times 18 bold"), bg="white")
        title_main.grid(row=0, column=0, columnspan=2, pady=5)

        def savetodb():
            # 检查是否已选择/创建数据库
            if not self.controller.database:
                showinfo(title="Error", message="Please create or open a database first!")
                return
            
            # 获取输入值
            track = ent_track.get().strip()
            if not track:
                showinfo(title="Error", message="Please enter a track name!")
                return
            
            # 准备数据
            data = [
                datetime.today().strftime('%Y-%m-%d'),
                track
            ]
            
            # 追加写入数据
            try:
                with open(self.controller.database, "a", newline='') as f:
                    writer = csv.writer(f)
                    writer.writerow(data)
                showinfo(title="Success", message="Data saved successfully!")
            except Exception as e:
                showinfo(title="Error", message=f"Failed to save data: {str(e)}")

        # 创建参考数据框架
        frame_reference = tk.LabelFrame(
            self,
            text="1: Enter reference data: ")
        frame_reference.grid(row=1, column=0, padx=5, pady=5, columnspan=2)

        lbl_track = Label(
                    frame_reference,
                    text="Track: ",
                    font='Times 11',
                    justify=tk.CENTER)
        lbl_track.grid(row=0, column=0)

        # 分离Entry创建与布局，以便获取输入值
        ent_track = Entry(
            frame_reference,
            justify=tk.CENTER,
            width=23)
        ent_track.grid(row=0, column=1)

        btn_savetodb = Button(
            frame_reference,
            text="Save to DB!",
            font='Times 11 bold',
            command=savetodb)
        btn_savetodb.grid(row=2, column=0, columnspan=2, padx=10, pady=5)

class SidePage(tk.Frame):
    def __init__(self, parent, controller):
        tk.Frame.__init__(self, parent)
        label = tk.Label(self, text="This is the Side Page")
        label.pack(padx=10, pady=10)

        switch_window_button = tk.Button(
            self,
            text="Go to the Completion Screen",
            command=lambda: controller.show_frame(CompletionScreen),
        )
        switch_window_button.pack(side="bottom", fill=tk.X)


class CompletionScreen(tk.Frame):
    def __init__(self, parent, controller):
        tk.Frame.__init__(self, parent)
        label = tk.Label(self, text="Completion Screen, we did it!")
        label.pack(padx=10, pady=10)
        switch_window_button = ttk.Button(
            self, text="Return to menu", command=lambda: controller.show_frame(MainPage)
        )
        switch_window_button.pack(side="bottom", fill=tk.X)

if __name__ == "__main__":
    testObj = windows()
    testObj.mainloop()

关键改进点说明 #

• 实例变量替代全局变量：使用self.database存储文件路径，避免全局变量混乱，确保所有组件正确访问当前数据库。
• 错误处理：添加未选择数据库、输入为空的检查，以及文件写入异常捕获，提升用户体验。
• 输入值获取：修正ent_track创建方式，确保能获取用户输入的赛道名称。
• 追加写入：使用"a"模式保存数据，避免每次保存覆盖整个文件。
• 对话框取消处理：在command_open和command_new中检查用户是否取消操作，避免空路径错误。

内容的提问来源于stack exchange，提问作者Andrei Filep