将python filedialog限制为特定的文件类型

时间:2017-09-20 20:40:08

标签: python-3.x tkinter

我创建了以下试用代码,以制作一个程序,将Excel工作表加载到数据框中,然后将该数据框保存到另一个Excel工作表(所有用户都选择)

import os
import pandas
from pandas import ExcelWriter
import tkinter as tk
from tkinter import filedialog

class Load_Save_Program():

    def __init__(self,master):
        self.master = master

        self.button1=tk.Button(self.master,text="Load",command=self.Load_file)
        self.button1.grid(row=0,column=0)

        self.button2=tk.Button(self.master,text="Save",command=self.Save_file)
        self.button2.grid(row=0,column=1)

        self.text=tk.Text(master)
        self.text.grid(row=1,column=0,columnspan=2)

    def Load_file(self):
        self.df_import=pandas.read_excel(filedialog.askopenfilename(initialdir = os.getcwd()),
                                            filetypes=("excel files","*.xlsx"))
        self.text.insert(tk.END,self.df_import)

    def Save_file(self):
        self.writer = ExcelWriter(filedialog.asksaveasfilename(initialdir = os.getcwd()),
                                            filetypes=("Excel files", "*.xlsx"))
        self.df_import.to_excel(self.writer,'sheet1')
        self.writer.save()

root=tk.Tk()
Load_Save_Program(root)
root.mainloop()

我想要做的是扩展它,以便当程序弹出文件目录窗口时,它只显示.xlsx文件类型的文件,以避免用户打开不兼容的文件类型时出错。到目前为止,我还没有提出任何可以解释如何正确设置的信息。

2 个答案:

答案 0 :(得分:2)

结构记录有一个文件类型选项,可以完全按照您的要求进行操作。一般语法是filetypes=[(label1, ext1), (label2, ext2), ...]

在你的情况下会给出:

filedialog.askopenfilename(filetypes=[("Excel files", "*.xlsx")])

答案 1 :(得分:0)

事实证明,订单很重要。此外,答案必须包括文件类型周围的[]。有问题的两行必须是

self.df_import=pandas.read_excel(filedialog.askopenfilename(filetypes=[("Excel files","*.xlsx")],initialdir = os.getcwd()))
self.writer = ExcelWriter(filedialog.asksaveasfilename(filetypes=[("Excel files", "*.xlsx")],initialdir = os.getcwd()))