使用 filetype 替代 imghdr

Author: MeetWq

nonebot_plugin_memes_api/matchers/utils.py

@@ -1,11 +1,11 @@
-import imghdr
 import tempfile
 from datetime import datetime
 from io import BytesIO
 from pathlib import Path
 from typing import Annotated
 from zipfile import ZIP_BZIP2, ZipFile
 
+import filetype
 from nonebot.adapters import Bot, Event
 from nonebot.matcher import Matcher
 from nonebot.params import Depends
@@ -94,7 +94,7 @@ def zip_images(files: list[bytes]):
     output = BytesIO()
     with ZipFile(output, "w", ZIP_BZIP2) as zip_file:
         for i, file in enumerate(files):
-            ext = imghdr.what(None, h=file)
+            ext = filetype.guess_extension(file)
             zip_file.writestr(f"{i}.{ext}", file)
     return output
 

poetry.lock

@@ -18,6 +18,7 @@ nonebot-plugin-orm = ">=0.7.0,<1.0.0"
 httpx = ">=0.20.0,<1.0.0"
 pyyaml = "^6.0"
 python-dateutil = "^2.8.2"
+filetype = "^1.2.0"
 
 [tool.poetry.group.dev.dependencies]
 nonebot2 = { extras = ["fastapi"], version = "^2.3.0" }