fix: use same base ALLOWED_FORMATS list as thumbor (#25)

theatlantic · Jul 8, 2024 · 2b4595b · 2b4595b
1 parent 5e44c0c
commit 2b4595b
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/src/thumbor_video_engine/filters/format.py b/src/thumbor_video_engine/filters/format.py
@@ -1,24 +1,31 @@
 from thumbor.filters import BaseFilter, filter_method, PHASE_PRE_LOAD
+import thumbor.filters.format
 from thumbor.utils import logger
 
 from thumbor_video_engine.compat import filter_retval
 
 
-ALLOWED_FORMATS = [
-    'png', 'jpeg', 'jpg', 'gif', 'webp', 'webm', 'mp4', 'hevc', 'h264', 'h265', 'vp9']
+ALLOWED_FORMATS = list(thumbor.filters.format.ALLOWED_FORMATS) + [
+    "webm",
+    "mp4",
+    "hevc",
+    "h264",
+    "h265",
+    "vp9",
+]
 
 
 class Filter(BaseFilter):
     phase = PHASE_PRE_LOAD
 
     @filter_method(BaseFilter.String)
     def format(self, format):
-        logger.debug('Setting format to %s' % format)
+        logger.debug("Setting format to %s" % format)
         if format.lower() not in ALLOWED_FORMATS:
-            logger.warning('Format not allowed: %s' % format.lower())
+            logger.warning("Format not allowed: %s" % format.lower())
             self.context.request.format = None
         else:
-            logger.debug('Format specified: %s' % format.lower())
+            logger.debug("Format specified: %s" % format.lower())
             self.context.request.format = format.lower()
 
         return filter_retval()