批量修改文件后缀名

import os
def look_name():
    os.chdir(path)
    print('路径：',os.getcwd())
    for i in os.listdir(os.getcwd()):
        print(i)
def all_name():
    for i in os.listdir(os.getcwd()):
        end=i.rfind('.')
        i1=i.replace(i[end:],new_name)
        print(i,i1)
        os.rename(i,i1)
def part_name():
    worn_name = input('请选择需要修改的后缀名')
    if worn_name.find('.') == -1 :
        worn_name='.'+worn_name
    for i in os.listdir(os.getcwd()):
        i1=i.replace(worn_name,new_name)
        print(i,i1)
        os.rename(i,i1)

path = input('请将需要改回名字的目录粘贴进来（空格加回车结束）：')
new_name = input('请将需要改成的后缀名填写进来（空格加回车结束）：')
if new_name.find('.')==-1:
    new_name='.'+new_name
look_name()

while True:
    xg = input('修改部分请输入1,修改全部请输入2,退出请按Y：')
    if xg =='1' or xg == '2':
        judeg = input('是否确认修改：确认请按Y')
        if judeg == 'y' or 'Y':
            if xg == '1':
                part_name()
                break
            else:
                all_name()
                break
        else:
            continue

    elif xg =='Y' or xg=='y':
        break
    else:
        print('输入错误，退出请按Y')
print('谢谢使用，再见。')