Введение
В данной статье научимся получать расширение файла при помощи Python.
Функция os.path.splitext()
Для получения расширения файла мы воспользуемся функцией splitext() из модуля os.path, который входит в стандартную библиотеку Python.
Функция os.path.splitext() предназначена для разделения имени пути на корневую часть и расширение. Корневая часть содержит всё, кроме расширения, а расширение представляет собой часть пути после последней точки.
Написание кода
Импортируем библиотеку os:
import os
Создадим переменную file_path, в которую передадим путь до файла, расширение которого нам необходимо получить:
import os
file_path = "C:/путь/к/файлу/file.txt"
В переменные file_name и file_extension сохраним путь к файлу с именем и его расширение соответственно:
import os
file_path = "C:/путь/к/файлу/file.txt"
file_name, file_extension = os.path.splitext(file_path)
Осталось добавить вывод полученных данных:
import os
file_path = "C:/путь/к/файлу/file.txt"
file_name, file_extension = os.path.splitext(file_path)
print(f"Путь к файлу: {file_name}\n"
f"Расширение файла: {file_extension}")
# Вывод:
# Путь к файлу: C:/путь/к/файлу/file
# Расширение файла: .txt
Функция для проверки расширения
Напишем функцию для проверки необходимого расширения файла. Назовём её check_file_extension() и в качестве параметров укажем file_path, в который будет передаваться путь до файла, и expected_extension, в который будет передаваться необходимое расширение.
import os
def check_file_extension(file_path, expected_extension):
Внутри функции сначала имя файла и расширение будут сохраняться в соответствующие переменные:
import os
def check_file_extension(file_path, expected_extension):
file_name, file_extension = os.path.splitext(file_path)
Далее будет идти условие, что если расширение переданного файла схоже с указанным в expected_extension, то будет выведен текст «Расширение файла верное.», иначе — «Расширение файла неверное.»:
import os
def check_file_extension(file_path, expected_extension):
file_name, file_extension = os.path.splitext(file_path)
if file_extension == expected_extension:
print("Расширение файла верное.")
else:
print("Расширение файла неверное.")
Пример использования функции
import os
def check_file_extension(file_path, expected_extension):
file_name, file_extension = os.path.splitext(file_path)
if file_extension == expected_extension:
print("Расширение файла верное.")
else:
print("Расширение файла неверное.")
# Пример использования функции
file_path = "file.txt"
expected_extension = ".txt"
check_file_extension(file_path, expected_extension)
# Вывод: Расширение файла верное.
Заключение
В ходе статьи мы с Вами научились получать расширение файлов при помощи Python. Надеюсь Вам понравилась статья, желаю удачи и успехов! 🙂
Мой Telegram канал
Мой YouTube канал
Мой курс по Python (50 видоуроков + дополнительные уроки)
Курс по созданию телеграм-ботов на Python с фреймворком Aiogram