python中获取文件和图片类型的方法

在 Python 中获取文件和图片类型的方法有几种。下面我将介绍三种常见的方法：

一. 使用第三方库 filetype

filetype 是一个常用的 Python 库，可以用于检测文件类型，包括图片类型。你可以使用它来确定文件的 MIME 类型和文件扩展名，从而判断文件类型。

安装 filetype 库：

pip install filetype

示例代码：

import filetype

# 要检测的文件路径
file_path = 'path/to/your/file'

# 检测文件类型
kind = filetype.guess(file_path)

if kind is None:
    print('无法确定文件类型！')
else:
    print('文件类型:', kind.mime)
    print('文件扩展名:', kind.extension)

二. 使用第三方库 Pillow（针对图片）

Pillow 是一个 Python 图像处理库，它可以帮助你处理图片，并提供了获取图片类型的功能。

安装 Pillow 库：

pip install Pillow

示例代码：

from PIL import Image

# 要检测的图片路径
image_path = 'path/to/your/image.jpg'

# 打开图片
image = Image.open(image_path)

# 获取图片的格式
image_format = image.format

print('图片格式:', image_format)

三. 使用Python标准库imghdr（针对图片）

没错！imghdr 是 Python 标准库中的一个模块，专门用于检测图像文件的类型。虽然 imghdr 只能用于图像文件，但在某些情况下，它可能是一个更轻量级的选择。

示例代码：

import imghdr

# 要检测的图片路径
image_path = 'path/to/your/image.jpg'

# 获取图像文件类型
image_type = imghdr.what(image_path)

if image_type is None:
    print('无法确定图像类型！')
else:
    print('图像类型:', image_type)

import requests
import imghdr

img_url = "https://example.com/test"
response = requests.get(img_url)

# 从响应内容中读取图像类型
ext = imghdr.what(None, h=response.content)
print(ext)

imghdr.what() 函数将返回文件的图像类型，例如 'jpeg', 'png', 'gif' ,webp等。如果无法确定类型，则返回 None。

这种方法比较简单，如果你只需要检测图像文件的类型，而不需要关注其他类型的文件，则 imghdr 是一个很好的选择。