add support for Django 4.2 storages

auto-mat · Nov 14, 2023 · 90a34d5 · 90a34d5
1 parent d8357e7
commit 90a34d5
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/import_export_celery/fields.py b/import_export_celery/fields.py
@@ -1,13 +1,17 @@
 from django.conf import settings
-from django.core.files.storage import get_storage_class
+from django.core.files.storage import get_storage_class, storages
 from django.db import models
 
 
 def lazy_initialize_storage_class():
     # If the user has specified a custom storage backend, use it.
     if getattr(settings, "IMPORT_EXPORT_CELERY_STORAGE", None):
-        storage_class = get_storage_class(settings.IMPORT_EXPORT_CELERY_STORAGE)
-        return storage_class()
+        try:
+            storage_class = get_storage_class(settings.IMPORT_EXPORT_CELERY_STORAGE)
+            return storage_class()
+        except (ImportError, TypeError):
+            return storages[settings.IMPORT_EXPORT_CELERY_STORAGE]
+
     return None