Получаем расширение файла при помощи Python

Введение

В данной статье научимся получать расширение файла при помощи 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. Надеюсь Вам понравилась статья, желаю удачи и успехов! 🙂