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()

转载于:https://www.cnblogs.com/zbuter/p/9171567.html

• 点赞
• 收藏
• 分享
• 文章举报