python给一列数据增加前缀_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()