Python之文件操作实例：批量修改文件的前缀名

# 导入包，很重要的
import os

"""
# 创建文件夹及文件
os.mkdir("./test")
for i in range(1, 6):
    os.mkdir("./test/test"+str(i))
    fp = open("./test/test"+str(i)+".txt", 'w',encoding='utf-8')
"""
while True:
    print("功能代码:\n1.添加前缀\t2.删除前缀")
    sn = input("请选择您的操作代码:")
    profix = input("请输入前缀:")
    path = input("请输入您的文件路径:")

    # 获取路径下的所有文件夹及文件
    file_list = os.listdir(path)
    # print(file_list)
    # 遍历所有的文件
    for i in file_list:
        path_flag = os.path.isfile(os.path.join(path, i))

        if not path_flag:  # 跳过目录
            continue

        # 判断执行的操作.并对非目录的文件进行操作
        if sn == '1':
            # pass  # 添加前缀
            os.rename(os.path.join(path, i), os.path.join(path, profix + i))  # os.path.join(path, profix + i) 连接路径
        elif sn == '2':
            # pass  # 删除前缀
            if i.startswith(profix):
                os.rename(os.path.join(path, i), os.path.join(path, i[len(profix):]))
        else:
            print("执行代码错误!!!")
    else:
        print("执行成功!!")