Проверка расширения файла

Вводится имя файла. Требуется проверить, что его расширение входит в список допустимых.

extensions = ['png', 'jpg', 'jpeg', 'gif', 'svg']

file = input().split('.')

if len(file) >= 2:

    fileExtension = file[-1].lower()

    if fileExtension in extensions:

        print("Yes")

    else:

        print("No")

else:

    print("The file doesn't have an extention.") 

С комментариями:

# список допустимых расширений

extensions = ['png', 'jpg', 'jpeg', 'gif', 'svg']

# Введенная строка преобразуется в список.

# Разделение происходит по точке.

file = input().split('.')

# Если длина списка равна двум (или больше),

# то расширение было указано. 

if len(file) >= 2:

    # Оно последнее в списке. Извлекаем его 

    # и преобразуем к нижнему регистру.

    fileExtension = file[-1].lower()

    # Если расширение содержится в списке 

    # допустимых, то выводится "Yes".

    if fileExtension in extensions:

        print("Yes")

    # Когда расширения нет в списке.

    else:

        print("No")

# Длина списка file меньше двух. Значит, у файла 

# нет расширения.

else:

    print("The file doesn't have an extention.")

Пример выполнения программы:

house.JPEG

Yes