python批量删除文件名后缀

最近用工具转了一些mp3，名称被自动加了后缀，手工修改太麻烦，写了个简单的python脚本，如下：

import os
import sys

# 打印所有命令行参数
print(sys.argv)

root_dir = sys.argv[1]
suffix = sys.argv[2]

files = os.listdir(root_dir)
os.chdir(root_dir)
for file in files:
    # 分割文件名中的目录、文件名和扩展名
    directory, file_name_ext = os.path.split(file)
    file_name, extension = os.path.splitext(file_name_ext)
    # print((directory, file_name, extension))
    # new_file = file_name.rstrip(suffix) + extension
    if file_name.endswith(suffix):
        new_file = file_name[:-len(suffix)] + extension
        os.rename(file, new_file)
        print(new_file)