必须编写一个辅助整理模块,打开名为module.py的Python文件。
1、Directory类
在module.py中打开,输入:
# 导入所需的模块 import os import shutil # 先把路径设置为你需要整理的文件夹的路径 path = r"C:\Python\Python整理文件\我的文件夹" # 定义Directory类 class Directory(): def __init__(self, directory_name, file_suffix): # 参数:文件夹名 文件后缀名 self.file_suffix = file_suffix self.directory_name = directory_name # 要保存的文件 self.file_list = [] def add_file(self, file_name): # 获取文件后缀名 并判断是否在file_suffix列表中 if file_name.split('.')[-1] in self.file_suffix: # 如果有 则添加到file_list self.file_list.append(file_name) return True else: return False def save_files(self): # 这个方法用于保存已经添加的文件 # 首先使用mkdir函数创建文件夹 os.mkdir(path + "\\" + self.directory_name) for file in self.file_list: # 再使用move函数一个一个移动文件 shutil.move(path + "\\" + file, path + "\\" + self.directory_name)
定义Directory类是为了便于统一管理文件。第一个,获取参数:要保存的文件夹的名称,文件后缀名称(列表)。文件后缀名这个参数是为了判断保存哪些后缀名符合要求的文件。
在add_file(fille_name)中传入一个文件名,该文件的后缀名决定是否进行存储,如果合格,则加入存储列表,返回True,否则返回False。
save_files()用来保存添加到file_list中的文件,首先自动创建一个名为directory_name的属性的文件夹,然后使用shutil.move函数将文件移动到已创建的文件夹中。
2、Directories类
为了使管理更加方便,在module.py中添加了Directories类,使Directory类的管理更加方便。
class Directories(): def __init__(self, *directories): # 传入需要管理的Directory类 参数数量制 self.directories = directories # 传入文件名 def add_file(self, file_name): for directory in self.directories: # 挨个执行Directory类中的add_file函数 # 直至符合存储条件 if directory.add_file(file_name): break # 遍历每个Directory类 # 并且执行Directory类的save_files函数保存 def save_files(self): for directory in self.directories: directory.save_files()
以上就是Python中编写辅助模块module的方法,希望能对大家有所帮助!