Пользователь вводит имя файла. Проверить, находится ли расширение файла в списке допустимых.
# список допустимых расширений extensions = ['png', 'jpg', 'jpeg', 'gif', 'svg'] fname = input() # Введенная строка преобразуется в список. # Разделение происходит по точке. fname = fname.split('.') # Если длина списка равна двум (или больше), # то расширение было указано. if len(fname) >= 2: # Оно последнее в списке. Извлекаем его # и преобразуем к нижнему регистру. fname_ext = fname[-1].lower() # Если расширение содержится в списке # допустимых, то выводится "Да". if fname_ext in extensions: print("Yes") # Когда расширения нет в списке. else: print("Нет") # Длина списка fname меньше двух. # Значит, у файла нет расширения. else: print("Некорректное название файла")