При работе с файлами в Python обычно требуется способ проверить расширение файла. Знание расширения файла может помочь вам определить тип файла и выполнить соответствующие операции. В этой статье мы рассмотрим несколько методов с примерами кода для проверки расширений файлов в Python.

Метод 1: использование метода splittext()из модуля os.path

import os
def get_file_extension(filepath):
    _, extension = os.path.splitext(filepath)
    return extension
# Usage example
file_path = "path/to/file.txt"
extension = get_file_extension(file_path)
print(extension)  # Output: ".txt"

Метод 2: использование метода endswith()

def check_file_extension(filepath, extension):
    return filepath.endswith(extension)
# Usage example
file_path = "path/to/file.txt"
desired_extension = ".txt"
is_valid_extension = check_file_extension(file_path, desired_extension)
print(is_valid_extension)  # Output: True

Метод 3. Использование регулярных выражений (модуль re)

import re
def check_file_extension_regex(filepath, extension):
    pattern = r"\{}$".format(extension)
    match = re.search(pattern, filepath)
    return match is not None
# Usage example
file_path = "path/to/file.txt"
desired_extension = ".txt"
is_valid_extension = check_file_extension_regex(file_path, desired_extension)
print(is_valid_extension)  # Output: True

Метод 4. Использование модуля pathlib

from pathlib import Path
def check_file_extension_pathlib(filepath, extension):
    path = Path(filepath)
    return path.suffix == extension
# Usage example
file_path = "path/to/file.txt"
desired_extension = ".txt"
is_valid_extension = check_file_extension_pathlib(file_path, desired_extension)
print(is_valid_extension)  # Output: True

Метод 5. Сравнение расширений с помощью простого сравнения строк

def check_file_extension_string_comparison(filepath, extension):
    return filepath[-len(extension):] == extension
# Usage example
file_path = "path/to/file.txt"
desired_extension = ".txt"
is_valid_extension = check_file_extension_string_comparison(file_path, desired_extension)
print(is_valid_extension)  # Output: True

В этой статье мы рассмотрели несколько методов проверки расширений файлов в Python. Предпочитаете ли вы использовать встроенные модули, такие как os.pathили pathlib, регулярные выражения или простое сравнение строк, теперь у вас есть множество вариантов, соответствующих вашим потребностям. Используя эти методы, вы можете эффективно обрабатывать файлы разных типов и выполнять определенные операции в зависимости от их расширений.

Не забудьте выбрать метод, который лучше всего соответствует вашим требованиям и стилю кодирования. Приятного кодирования!