Add support for image/webp to mimetypes package

python/cog/__init__.py

@@ -1,5 +1,8 @@
+import mimetypes
+
 from pydantic import BaseModel
 
+from .mimetypes_ext import install_mime_extensions
 from .predictor import BasePredictor
 from .server.worker import emit_metric
 from .types import (
@@ -11,6 +14,8 @@
     Secret,
 )
 
+install_mime_extensions(mimetypes)
+
 try:
     from ._version import __version__
 except ImportError:

python/cog/mimetypes_ext.py

@@ -0,0 +1,16 @@
+from typing import Protocol
+
+
+class IMimeTypes(Protocol):
+    def add_type(self, type: str, ext: str, strict: bool = True) -> None: ...
+
+
+def install_mime_extensions(mimetypes: IMimeTypes) -> None:
+    """
+    Older versions of Python are missing the MIME types for more recent file formats
+    this function adds the missing MIME types to the mimetypes module.
+    """
+
+    # This could also be done by loading a mime.types file from disk using
+    # mimetypes.read_mime_types().
+    mimetypes.add_type("image/webp", ".webp")

python/tests/test_mimetypes_ext.py

@@ -0,0 +1,17 @@
+from mimetypes import MimeTypes
+from cog.mimetypes_ext import install_mime_extensions
+
+
+def test_webp_ext_support():
+    # Assert on empty database.
+    mt = MimeTypes(filenames=tuple())
+    assert mt.guess_type("image.webp") == (None, None)
+    install_mime_extensions(mt)
+    assert mt.guess_type("image.webp") == ("image/webp", None)
+
+    # Assert global override
+    import cog
+
+    import mimetypes
+
+    assert mimetypes.guess_type("image.webp") == ("image/webp", None)