Initial commit

env/lib/python3.10/site-packages/wagtail/images/__init__.py

@@ -0,0 +1,34 @@
+from django.conf import settings
+from django.core.exceptions import ImproperlyConfigured
+
+
+def get_image_model_string():
+    """
+    Get the dotted ``app.Model`` name for the image model as a string.
+    Useful for developers making Wagtail plugins that need to refer to the
+    image model, such as in foreign keys, but the model itself is not required.
+    """
+    return getattr(settings, "WAGTAILIMAGES_IMAGE_MODEL", "wagtailimages.Image")
+
+
+def get_image_model():
+    """
+    Get the image model from the ``WAGTAILIMAGES_IMAGE_MODEL`` setting.
+    Useful for developers making Wagtail plugins that need the image model.
+    Defaults to the standard ``wagtail.images.models.Image`` model
+    if no custom model is defined.
+    """
+    from django.apps import apps
+
+    model_string = get_image_model_string()
+    try:
+        return apps.get_model(model_string, require_ready=False)
+    except ValueError:
+        raise ImproperlyConfigured(
+            "WAGTAILIMAGES_IMAGE_MODEL must be of the form 'app_label.model_name'"
+        )
+    except LookupError:
+        raise ImproperlyConfigured(
+            "WAGTAILIMAGES_IMAGE_MODEL refers to model '%s' that has not been installed"
+            % model_string
+        )