Вводится имя файла. Требуется проверить, что его расширение входит в список допустимых.
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