From ed5dc65579a2f982ae9c86032a384bf3a0929b69 Mon Sep 17 00:00:00 2001
From: Jean-Baptiste <bleme@pm.me>
Date: Wed, 26 Jun 2019 09:22:47 +0200
Subject: [PATCH] update: automaticaly call admin.site.register for all
sublclasses of djangoldp.Model
---
djangoldp/admin.py | 18 +++++++++++++++++-
1 file changed, 17 insertions(+), 1 deletion(-)
diff --git a/djangoldp/admin.py b/djangoldp/admin.py
index 13370eb9..b385c6c7 100644
--- a/djangoldp/admin.py
+++ b/djangoldp/admin.py
@@ -1,4 +1,20 @@
+from importlib import import_module
+
+from django.conf import settings
from django.contrib import admin
-from .models import LDPSource
+from .models import LDPSource, Model
+
+for package in settings.DJANGOLDP_PACKAGES:
+ try:
+ import_module('{}.models'.format(package))
+ except ModuleNotFoundError:
+ pass
+
+model_classes = {cls.__name__: cls for cls in Model.__subclasses__()}
+
+for class_name in model_classes:
+ model_class = model_classes[class_name]
+ if not admin.site.is_registered(model_class):
+ admin.site.register(model_class)
admin.site.register(LDPSource)
--
GitLab