Python中如何编写辅助模块module?

美女程序员鼓励师

必须编写一个辅助整理模块,打开名为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的方法,希望能对大家有所帮助!

以上是 Python中如何编写辅助模块module? 的全部内容, 来源链接: utcz.com/z/543342.html

回到顶部