Merge pull request h2non#172 from isatsam/master

Add support for DDS

README.rst

@@ -80,6 +80,7 @@ Image
 -  **heic** - ``image/heic``
 -  **avif** - ``image/avif``
 -  **qoi** - ``image/qoi``
+-  **dds** - ``image/dds``
 
 Video
 ^^^^^

filetype/types/__init__.py

@@ -32,6 +32,7 @@
     image.Dcm(),
     image.Avif(),
     image.Qoi(),
+    image.Dds(),
 )
 
 # Supported video types

filetype/types/image.py

@@ -434,3 +434,20 @@ def match(self, buf):
                 buf[1] == 0x6F and
                 buf[2] == 0x69 and
                 buf[3] == 0x66)
+
+
+class Dds(Type):
+    """
+    Implements the DDS image type matcher.
+    """
+    MIME = 'image/dds'
+    EXTENSION = 'dds'
+
+    def __init__(self):
+        super(Dds, self).__init__(
+            mime=Dds.MIME,
+            extension=Dds.EXTENSION
+        )
+
+    def match(self, buf):
+        return buf.startswith(b'\x44\x44\x53\x20')

tests/test_types.py

@@ -56,6 +56,12 @@ def test_guess_avif(self):
         self.assertEqual(kind.mime, 'image/avif')
         self.assertEqual(kind.extension, 'avif')
 
+    def test_guess_dds(self):
+        kind = filetype.guess(FIXTURES + '/sample.dds')
+        self.assertTrue(kind is not None)
+        self.assertEqual(kind.mime, 'image/dds')
+        self.assertEqual(kind.extension, 'dds')
+
     def test_guess_m4a(self):
         kind = filetype.guess(FIXTURES + '/sample.m4a')
         self.assertTrue(kind is not None)