Ver Fonte

fix: validation for upload methods of non-image files within the work… (#15932)

Arcaner há 1 mês atrás
pai
commit
e6a8800f66
1 ficheiros alterados com 8 adições e 2 exclusões
  1. 8 2
      api/factories/file_factory.py

+ 8 - 2
api/factories/file_factory.py

@@ -289,9 +289,15 @@ def _is_file_valid_with_config(
     ):
         return False
 
-    if input_file_type == FileType.IMAGE and config.image_config:
-        if config.image_config.transfer_methods and file_transfer_method not in config.image_config.transfer_methods:
+    if input_file_type == FileType.IMAGE:
+        if (
+            config.image_config
+            and config.image_config.transfer_methods
+            and file_transfer_method not in config.image_config.transfer_methods
+        ):
             return False
+    elif config.allowed_file_upload_methods and file_transfer_method not in config.allowed_file_upload_methods:
+        return False
 
     return True