Initial commit

env/lib/python3.10/site-packages/treebeard/tests/admin.py

@@ -0,0 +1,18 @@
+import itertools
+
+from django.contrib import admin
+from treebeard.admin import admin_factory
+from treebeard.forms import movenodeform_factory
+
+from treebeard.tests.models import BASE_MODELS, UNICODE_MODELS, DEP_MODELS
+
+
+def register(admin_site, model):
+    form_class = movenodeform_factory(model)
+    admin_class = admin_factory(form_class)
+    admin_site.register(model, admin_class)
+
+
+def register_all(admin_site=admin.site):
+    for model in itertools.chain(BASE_MODELS, UNICODE_MODELS, DEP_MODELS):
+        register(admin_site, model)