fix: 添加文件后缀判断以避免假的 zip

Author: NoNormalCreeper

byrdocs/__init__.py

@@ -74,12 +74,12 @@ def wrapper(*args, **kwargs):
 
 def get_file_type(file) -> str:
     # https://en.wikipedia.org/wiki/List_of_file_signatures
-    # use magic number to check file type
+    # use magic number to check file type, together with suffix
     with open(file, "rb") as f:
         magic_number = f.read(4)
-        if magic_number == b"%PDF":
+        if magic_number == b"%PDF" and file.endswith(".pdf"):
             return "pdf"
-        elif magic_number == b"PK\x03\x04":
+        elif magic_number == b"PK\x03\x04" and file.endswith(".zip"):
             return "zip"
         else:
             return "unsupported"

pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools>=61.0"]
 build-backend = "setuptools.build_meta"
 [project]
 name = "byrdocs_cli"
-version = "0.5.2"
+version = "0.5.3"
 authors = [
   { name="Rikka", email="[email protected]" },
 ]