首页 > 解决方案 > 如何使用 PIL 检查 .webp 文件是否是 python 中的 gif 或图像?

问题描述

我试图在 python 中创建一个“.webp 提取器脚本”,在其中我会循环通过一个预先确定的文件目录搜索 .webp,然后确定它们是否是保存为 .webp 的 .gif 文件类型,所以我可以将 .webp 保存为 .gif 或者如果它不是 .gif 为 .webp,那么它将是一个图像,例如保存为 .webp 的 .jpg/.png 所以我可以将 .webp 保存为一个图像。

但是,检测 .webp 是否存储 gif 或图像有点令人困惑,这是我在检查之前的代码:

import os
from PIL import Image

for entry in os.scandir(Directory):
    Searched += 1
    if entry.path.endswith(".webp") and entry.is_file():
        print("--------------")
        print("Found a : .webp")
        print(entry.path)

        IsAGif = False

        MediaFile = Image.open(entry.path)

然后我被困在那里,不得不尝试找到一些解决方案。但我必须改变我的心态,并从逻辑上思考。我认为 .gif 有多个图像帧,而 .png 之类的图像只有一个。因此,如果我可以读取 .webp 文件并找出它,我就可以区分图像和 .gif...

标签: python-3.ximagepngwebp

解决方案


最终,我发现了如何读取图像/.gif 中的帧:

import os
from PIL import Image, ImageSequence

MediaFile = Image.open(entry.path)
for frame in ImageSequence.Iterator(MediaFile):
     #Rest of code here

因此,要查找 .webp 是 .gif 还是图像,我只需要读取有多少帧。如果有一帧,.webp 是一个图像,否则如果它超过一帧,它是一个 .gif,因为它在帧之间转换。因此,我为此提出了一个功能:

from PIL import Image, ImageSequence

def IsAGif(FilePath):
    try:
        MediaFile = Image.open(FilePath)
        Index = 0
        for Frame in ImageSequence.Iterator(MediaFile):
            Index += 1
        if Index > 1: #Must mean the the .webp is a gif because it has more than one frame
            return True
        else: #Must mean that the .webp has 1 frame which makes it a image
            return False
    except: #If the .webp can't be opened then it is most likely not a .gif or .webp
        return False

推荐阅读