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

Пользователь вводит имя файла. Проверить, находится ли расширение файла в списке допустимых.



# список допустимых расширений
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("Некорректное название файла")


Добавить комментарий

;-) :| :x :twisted: :smile: :shock: :sad: :roll: :razz: :oops: :o :mrgreen: :lol: :idea: :grin: :evil: :cry: :cool: :arrow: :???: :?: :!: