Merge pull request #7 from wedevelopnl/fix/widget-admin-checks

Update WidgetAdmin.php
wedevelopnl · Apr 24, 2024 · b76f709 · b76f709
2 parents 1b0640d + cb7326e
commit b76f709
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/Admin/WidgetAdmin.php b/src/Admin/WidgetAdmin.php
@@ -48,6 +48,11 @@ class WidgetAdmin extends ModelAdmin
     public function getGridField(): GridField
     {
         $gridField = parent::getGridField();
+
+        if (!in_array($gridField->getModelClass(), ClassInfo::subclassesFor(Widget::class), true)) {
+            return $gridField;
+        }
+
         $dataList = $gridField->getList();
 
         if ($dataList instanceof DataList) {
@@ -72,6 +77,11 @@ public function getGridField(): GridField
     public function getEditForm($id = null, $fields = null): Form
     {
         $form = parent::getEditForm();
+
+        if (!in_array($this->getModelClass(), ClassInfo::subclassesFor(Widget::class), true)) {
+            return $form;
+        }
+
         $formFields = $form->Fields();
         $currentModelClass = $this->getModelClass();
         $dataList = $currentModelClass::get()->filter(['IsPartOfCollection' => false]);