python自制工具----给文件添加前后缀。

PreffixAndSuffix 类：

import os

class PreffixAndSuffix:

    def __init__(self):
        pass

    @staticmethod
    def add_preffix(file_path, preffix):  # 为file_path添加preffix前缀 并返回文件名绝对路径
        dir_name, filename, extension = PreffixAndSuffix.get_names(file_path)

        new_name = dir_name + preffix + filename + extension
        os.rename(file_path, new_name)
        return new_name

    @staticmethod
    def del_preffix(file_path, preffix):        # 为file_path删除preffix前缀 并返回文件名绝对路径
        dir_name, filename, extension = PreffixAndSuffix.get_names(file_path)

        if filename.startswith(preffix):            # 判断文件名是否以preffix开头
            filename = filename.partition(preffix)[2]  # ('', preffix, 去掉前缀文件名)[2]
            new_name = dir_name + filename + extension
            os.rename(file_path, new_name)
            return new_name
        else:
            return file_path

    @staticmethod
    def add_suffix(file_path, suffix):  # 为file_path添加preffix后缀 并返回文件名绝对路径
        dir_name, filename, extension = PreffixAndSuffix.get_names(file_path)

        new_name = dir_name + filename + suffix + extension
        os.rename(file_path, new_name)
        return new_name


    @staticmethod
    def del_suffix(file_path, suffix):  # 为file_path删除preffix后缀 并返回文件名绝对路径
        dir_name, filename, extension = PreffixAndSuffix.get_names(file_path)

        if filename.endswith(suffix):  # 判断文件名是否以preffix开头
            filename = filename.rpartition(suffix)[0]  # (文件名, suffix, 扩展名)[0]

            new_name = dir_name + filename + extension
            os.rename(file_path, new_name)
            return new_name
        else:
            return file_path

    @staticmethod
    def get_names(file_path):
        file_path = os.path.abspath(file_path)  # 获取这个文件/文件夹的绝对路径
        dir_name = os.path.dirname(file_path)  # 获取所在目录
        dir_name = dir_name + os.sep  # 为拼接做准备
        filename, extension = os.path.splitext(file_path)  #: 分离文件名与扩展名结果为（filename，扩展名） 如果参数为一个路径则返回（路径，''）
        name = filename.rpartition(os.sep)[2]  # (文件目录名 ,目录分隔符, 文件名/目录名)
        names = (dir_name, name, extension)    #(文件所在目录名, 文件名, 文件扩展名)
        # print(names)
        return names

　　

 

测试类：

import os
from PreffixAndSuffix import PreffixAndSuffix


def change_name(file_path):
    pass




def main():

    y_or_n = {'Y': True, 'y': True, 'N': False, 'n': False, '': False}
    method = {'0': PreffixAndSuffix.add_preffix, '1': PreffixAndSuffix.add_suffix, '2': PreffixAndSuffix.del_preffix, '3': PreffixAndSuffix.del_suffix}

    file_path = str(input('请输入要求改的文件路径：'))
    designated_suffix_or_preffix = str(input('请输入指定的文件名前缀或后缀：'))

    while True:
        ecursive_modify = str(input('是否递归修改？((Y/N)默认为N)'))
        if ecursive_modify in y_or_n:
            isecursive_modify = y_or_n[ecursive_modify]
            break

    while True:
        designated_suffix_modify = str(input('是否只修改特定后缀文件？((Y/N)默认为N)'))
        if designated_suffix_modify in y_or_n:
            isdesignated_suffix_modify = y_or_n[designated_suffix_modify]
            if isdesignated_suffix_modify:  # 只修改指定后缀的文件
                designated_suffix =  str(input('请输入指定的后缀(如：.txt、.jpg)'))
                break
            else:
                designated_suffix = ''
                break

    while True:
        tip = '''
        0：添加指定前缀
        1：添加指定后缀
        2：删除指定前缀
        3：删除指定后缀
        请输入： '''
        change_method_num = input(tip)
        if change_method_num in method:     # 判断是否在方法字典中
            change_method = method[change_method_num]   # 获取选择的方法入口地址。
            break

    print('文件名：', file_path)
    print('指定的前缀或后缀：', designated_suffix_or_preffix)
    print('是否递归修改：', isecursive_modify)
    print('是否只修改特定后缀文件：', isdesignated_suffix_modify)
    print('指定修改的后缀：', designated_suffix)
    print('选择方法：',method[change_method_num])


    if not isecursive_modify:   # 如果不递归修改
        print('当前执行目录：', file_path)
        os.chdir(file_path)
        for file in os.listdir(file_path):  # 遍历当前目录下所有文件
            abs_path = file_path + os.sep + file    # 拼接文件的绝对路径
            if os.path.isfile(abs_path):    # 如果是文件
                if abs_path.endswith(designated_suffix):    # 判断是否为指定后缀。
                    new_name = change_method(abs_path, designated_suffix_or_preffix) # 调用方法字典中的选择的方法
                    print('已成功改名：', new_name)
        print('-' * 80)

    else:
        for root, dirs, files in os.walk(file_path):    # 递归遍历 file_path
            print('当前执行目录：', root)
            os.chdir(root)
            print(files)    # 当前目录下所有文件

            for file in files:  # 遍历所有文件
                abs_path = root + os.sep + file  # 拼接绝对地址
                if abs_path.endswith(designated_suffix): # 判断是否为指定后缀
                    new_name = change_method(abs_path, designated_suffix_or_preffix)    # 调用方法字典中的选择的方法
                    print('已成功改名：', new_name)
            print('-' * 80)



if __name__ == '__main__':
    main()