python基础应用：批量自定义修改文件名

# 需求：批量修改文件名，既可添加指定字符串，又能删除指定字符串。

import os
flag = int(input("请输入要执行的操作：1.添加指定字符 2.删除指定字符"))
if flag == 1:
    add = input("请输入想要添加的字符")
elif flag == 2:
    noadd = input("请输入想要删除的字符")
else:
    print("输入错误！")
# 1. 设置添加删除字符串的的标识 如果flag为1则添加指定字符，flag取值为2则删除指定字符
# 2. 获取指定目录的所有文件
    #2.1获取指定目录
dir_name = './'  #当前目录所有文件
    #2.2获取指定目录的文件列表
file_list = os.listdir(dir_name)
#print(file_list)
# 3. 将原有文件名添加/删除指定字符串，构造新名字
for name in file_list:
    #添加指定字符
    if flag == 1:
        new_name = add+name
    #删除指定字符
    if flag == 2:
        num = len(noadd)
        new_name = name[num:]
    #打印新文件名，测试程序正确性

    # 4. os.rename()重命名
    os.rename(name,new_name)