fastapi-file-validator 0.0.0__py3-none-any.whl

fastapi_file_validator-0.0.0.dist-info/METADATA

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: fastapi-file-validator
+Version: 0.0.0
+Summary: A lightweight, flexible file extension validator and MIME type checker for fastapi
+Author-email: Seungkyu-Han <trust1204@gmail.com>
+License: MIT
+Project-URL: Homepage, https://github.com/Seungkyu-Han/fastapi-file-validator
+Project-URL: Bug Tracker, https://github.com/Seungkyu-Han/fastapi-file-validator
+Project-URL: Documentation, https://github.com/Seungkyu-Han/fastapi-file-validator
+Project-URL: Source, https://github.com/Seungkyu-Han/fastapi-file-validator
+Keywords: validator
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.110.1

fastapi_file_validator-0.0.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+file_validator/__init__.py,sha256=45cYSsnJB7JHfv2j3_8TZLEJmOhoXw97e_WiGMya3Lc,189
+file_validator/file_extension_validator.py,sha256=6JHv4L81fbE0aQFoIejr4EdFEUveQfpxbJJDo15K9lo,1249
+file_validator/file_size_validator.py,sha256=4hm_cx-MwnoQT9-_t0LMrBXTrfVRlp_Dg_2si-fbYNk,1028
+fastapi_file_validator-0.0.0.dist-info/METADATA,sha256=Nzb9bi9APuM4XoBPxM4dMuK_oRJ924ff-Dnl0BUwIRU,654
+fastapi_file_validator-0.0.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+fastapi_file_validator-0.0.0.dist-info/top_level.txt,sha256=eyZ1S8dWz5evkVYPVPaEUqBRn5nlw1QYV0sp0h72aBY,15
+fastapi_file_validator-0.0.0.dist-info/RECORD,,

fastapi_file_validator-0.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

fastapi_file_validator-0.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+file_validator

file_validator/__init__.py

@@ -0,0 +1,7 @@
+from .file_extension_validator import file_extension_validator
+from .file_size_validator import file_size_validator
+
+__all__ = [
+    "file_extension_validator",
+    "file_size_validator",
+]

file_validator/file_extension_validator.py

@@ -0,0 +1,42 @@
+import os
+
+from fastapi import UploadFile, HTTPException, status
+from functools import wraps
+
+
+def file_extension_validator(file_arg_name: str, file_extensions: set[str]):
+    def decorator(func):
+        @wraps(func)
+        async def wrapper(*args, **kwargs):
+            if file_arg_name in kwargs:
+
+                file_arg = kwargs.get(file_arg_name)
+
+                if not isinstance(file_arg, UploadFile):
+                    return await func(*args, **kwargs)
+
+                file: UploadFile = file_arg
+
+                filename: str | None = file.filename
+
+                if not filename:
+                    raise HTTPException(
+                        status_code=status.HTTP_400_BAD_REQUEST,
+                        detail=f"filename is empty",
+                    )
+
+                _, ext = os.path.splitext(filename)
+
+                file_extension = ext[1:].lower()
+
+                if file_extension not in file_extensions:
+                    raise HTTPException(
+                        status_code=status.HTTP_415_UNSUPPORTED_MEDIA_TYPE,
+                        detail=f"{file_extension} type is unavailable",
+                    )
+
+            return await func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator

file_validator/file_size_validator.py

@@ -0,0 +1,34 @@
+from fastapi import UploadFile, HTTPException, status
+from functools import wraps
+
+
+def file_size_validator(file_arg_name: str, max_size_mb: int):
+    max_size_bytes = max_size_mb * 1024 * 1024
+
+    def decorator(func):
+        @wraps(func)
+        async def wrapper(*args, **kwargs):
+            if file_arg_name in kwargs:
+                file_arg = kwargs.get(file_arg_name)
+
+                if not isinstance(file_arg, UploadFile):
+                    return await func(*args, **kwargs)
+
+                file: UploadFile = file_arg
+
+                file_size: int | None = file.size
+
+                if not file_size:
+                    return await func(*args, **kwargs)
+
+                if file_size > max_size_bytes:
+                    raise HTTPException(
+                        status_code=status.HTTP_413_CONTENT_TOO_LARGE,
+                        detail=f"file size is bigger than {max_size_mb}MB",
+                    )
+
+            return await func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator