feat: Allow for multiple admin sites

Author: fsbraun

djangocms_versioning/helpers.py

@@ -4,6 +4,7 @@
 import warnings
 from collections.abc import Iterable
 from contextlib import contextmanager
+from functools import cache
 
 from cms.models import Page, PageContent, Placeholder
 from cms.toolbar.utils import get_object_edit_url, get_object_preview_url
@@ -17,6 +18,7 @@
 from django.db import models
 from django.http import HttpRequest
 from django.template.loader import render_to_string
+from django.urls import URLResolver, get_resolver
 from django.utils.encoding import force_str
 from django.utils.translation import get_language, override as force_language
 
@@ -54,6 +56,21 @@ def is_editable(content_obj: models.Model, request: HttpRequest) -> bool:
     )
 
 
+@cache
+def get_admin_sites():
+    resolver = get_resolver()
+
+    def find_admin_sites(patterns):
+        for pattern in patterns:
+            if isinstance(pattern, URLResolver):
+                yield from find_admin_sites(pattern.url_patterns)
+            instance = getattr(pattern.callback, "__self__", None)
+            if isinstance(instance, admin.AdminSite):
+                yield instance
+
+    return list(find_admin_sites(resolver.url_patterns))
+
+
 def versioning_admin_factory(admin_class: type[admin.ModelAdmin], mixin: type) -> type[admin.ModelAdmin]:
     """A class factory returning admin class with overriden
     versioning functionality.
@@ -93,13 +110,17 @@ def replace_admin_for_models(pairs: tuple[type[models.Model], type], admin_site:
     :param admin_site: AdminSite instance
     """
     if admin_site is None:
-        admin_site = admin.site
-    for model, mixin in pairs:
-        try:
-            modeladmin = admin_site._registry[model]
-        except KeyError:
-            continue
-        _replace_admin_for_model(modeladmin, mixin, admin_site)
+        admin_sites = get_admin_sites() or [admin.site]
+    else:
+        admin_sites = [admin_site]
+
+    for _admin in admin_sites:
+        for model, mixin in pairs:
+            try:
+                modeladmin = admin_site._registry[model]
+            except KeyError:
+                continue
+            _replace_admin_for_model(modeladmin, mixin, _admin)
 
 
 def register_versionadmin_proxy(versionable, admin_site: admin.AdminSite | None = None):