Python 批量添加和删除前缀

import os

while True:

    print("请选择：\n1.添加前缀\t2.删除前缀")

    action=input("请选择您的操作代码：")

    profix=input("请输入前缀：")

    #获取文件所在路径

    path =input("请输入文件路径：")

    #获取文件路径下的所有文件

    file_names=os.listdir(path)

    #print(file_names)

    #遍历文件

    for iin file_names:

        path_flag=os.path.isfile(os.path.join(path,i))

        if not path_flag:

        #跳过目录

            continue

        #判断执行操作，并对非目录文件

        if action =='1':

        # pass  # 添加前缀

            os.rename(os.path.join(path, i), os.path.join(path,profix + i))# os.path.join(path, profix + i) 连接路径

        elif action =='2':

            # pass  # 删除前缀

            if i.startswith(profix):

                os.rename(os.path.join(path, i),os.path.join(path, i[len(profix):]))

        else:

        print("执行代码错误!!!")

    else:

        print("执行成功!!")