python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python文件夹整理工具

Python基于面向对象做一个文件夹整理工具

作者:微小冷

这篇文章主要给大家介绍了Python基于面向对象做一个文件夹整理工具,文中的示例代码讲解详细,具有一定的参考价值,感兴趣的小伙伴可以跟随小编一起学习一下

此前曾经基于函数实现过这个功能:用Python做一个文件夹整理工具。这次基于面向对象的方法,重新实现这个小工具。

本文中提到的FileSystem,来自于这里:Python根据文件后缀整理文件夹

目标

我们的目的是做一个像下面这样的工具,前面两个输入框,用于输入源路径和目标就,下面的图片、视频、音乐表示在目标路径中创建的文件夹,后面的文件后缀,表示将这类文件移动到对应的文件夹中,加减号可以新增或删除文件夹。

文件夹对话框

第一步,实现源路径和目标路径的输入输出,从而需要自定义一个组件,代码如下

import tkinter as tk
import tkinter.ttk as ttk
from tkinter.filedialog import (askopenfilename,
    askopenfilenames, askdirectory, asksaveasfilename)
from tkinter.colorchooser import askcolor

class DialogButton(ttk.Frame):
    def __init__(self, master, 
        height, widthL, widthR, logtype, label=None, text=None, 
        frmDct={}, btnDct={}, enyDct={}, logDct={}):
        w = widthL + widthR
        super().__init__(master, 
            height=height, width = w, **frmDct)
        self.pack(fill=tk.X)

        self.text = tk.StringVar() if not text else text
        ttk.Entry(self, width=widthL, textvariable=self.text, 
            **enyDct).pack(side=tk.LEFT, fill = tk.X, expand=True)
        
        ttk.Button(self, width=widthR, 
            text=self.setLabel(logtype, label),
            command = self.Click, **btnDct).pack(side=tk.RIGHT, padx=5)
        self.logtype = logtype
        self.logDct = logDct

    def setLabel(self, key, label=None):
        if label:
            return label
        labelDct = {
            "文件"   : "选择文件",
            "文件夹" : "选择路径",
            "多文件" : "选择多个文件",
            "保存" : "存储路径",
            "颜色"   : "选择颜色",
        }
        return labelDct[key]

    def Click(self):
        typeDct = {
            "文件"  : askopenfilename,
            "文件夹": askdirectory,
            "多文件": askopenfilenames,
            "保存"  : asksaveasfilename,
            "颜色"  : askcolor,
        }
        text = typeDct[self.logtype](**self.logDct)
        if self.logtype == "颜色":
            text = text[1]
        self.text.set(text)

    def get(self):
        return self.text.get()

    def set(self, txt):
        self.text.set(txt)

文件映射组件

在文件映射列表中,每一行的UI元素是完全相同的,故而可以新建一个文件映射组件,文件映射组件,主要由左右两部分组成,左侧是文件夹名,右侧是对应的文件后缀。从这两个子组件出发,可以为其设置初始化参数以及方法。先看源码,再逐段解析

class ExFolder(ttk.Frame):
    def __init__(self, master,
        folder=None, ex=None, **options):
        super().__init__(master, **options)
        self.pack()

        self.folder = tk.StringVar()
        if folder : self.setFolder(folder)
        self.ex = tk.StringVar()
        if ex : self.setEx(ex)

        self.initWidgets()

    def initWidgets(self):
        ttk.Entry(self, textvariable=self.folder,
            width=10).pack(side=tk.LEFT, padx=2)
        ttk.Entry(self, textvariable=self.ex
            ).pack(side=tk.LEFT, fill=tk.X, expand=True)

    def getFolder(self):
        return self.folder.get()

    def setFolder(self, f):
        self.folder.set(f)

    def getEx(self):
        exs = self.ex.get().split(",")
        return [ex.strip() for ex in exs]

    def setEx(self, exs):
        if type(exs)==list:
            exs = ', '.join([ex.strip() for ex in exs])
        self.ex.set(exs)

首先,folder和ex分别是左右两个组件对应的可变文本,为了便于更新和调用,类中设计了两组、四对便携方法getFolder, setFolder和getEx和setEx。由于Folder中的内容就是文件夹本身,所以设置和读取方法仅仅是StringVar的二次封装,而ex中的内容则不然,需要对文本进行分割,并且实现字符串到列表的转换。

完整组件

最后,是整个组件的布局,源码如下

class FolderSplit(ttk.Frame):
    def __init__(self, master, **options):
        super().__init__(master, **options)
        self.pack()
        self.fs = FileSystem()

        self.initWidgets()

    def initWidgets(self):
        self.exForders = []
        pDct = dict(side=tk.TOP, expand=True, fill=tk.X)
        self.srcFolder = DialogButton(self, 5, 25, 8,
                                      "文件夹", label="源路径")
        self.srcFolder.pack(**pDct)
        self.dstFolder = DialogButton(self, 5, 25, 8,
                                      "文件夹", label="目标路径")
        self.dstFolder.pack(**pDct)

        btns = ttk.Frame(self)
        btns.pack(**pDct)
        ttk.Button(btns, text="➕ ",
            command=self.btnAddFolder).grid(row=0,column=0)
        ttk.Button(btns, text="➖",
            command=self.btnDelFolder).grid(row=0,column=1)
        ttk.Button(btns, text="移动",
            command=self.btnMove).grid(row=0,column=2)
        ttk.Button(btns, text="撤销",
            command=self.btnReMove).grid(row=0,column=3)

        self.exFrm = ttk.LabelFrame(self, text="文件夹映射表")
        self.exFrm.pack(**pDct)

        for k, v in self.fs.fileDct.items():
            self.addFolders(self.exFrm, k, v)

    def btnAddFolder(self):
        self.addFolders(self.exFrm, None, None)

    def addFolders(self, frm, folder, exs):
        ef = ExFolder(frm, folder, exs)
        ef.pack(side=tk.TOP, expand=True, fill=tk.X, padx=2, pady=1)
        self.exForders.append(ef)

    def btnDelFolder(self):
        if len(self.exForders) == 0: return
        self.exForders[-1].pack_forget()
        del self.exForders[-1]

    def setExDct(self):
        fileDct = {ef.getFolder() : ef.getEx()
                        for ef in self.exForders}
        self.fs.update(fileDct)

    def btnMove(self):
        src = self.srcFolder.get()
        dst = self.dstFolder.get()
        self.setExDct()
        if dst=="": dst = src
        self.fs.start(src, dst)

    def btnReMove(self):
        self.fs.restore()

三个已经实现的功能按钮,➕用于新增一个文件夹映射组件,➖则移除最后一个映射组件。移除组件时需要注意,不仅仅要把组件解绑,还要把组件对应的实例删掉。

核心功能被绑定在移动按钮上,首先读取源路径和目标路径,如果目标路径未输入,就在源路径下工作。

接下来,将文件映射组件重新拆解为映射字典,最后调用FileSystem中的移动方法,实现所有文件的移动。

到此这篇关于Python基于面向对象做一个文件夹整理工具的文章就介绍到这了,更多相关Python文件夹整理工具内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文