Cleanup code for older versions of Wagtail (< 4.1)

Author: katdom13

tests/testapp/models.py

@@ -1,6 +1,5 @@
 from django.db import models
 from wagtail.admin.panels import FieldPanel
-from wagtail.admin.panels import FieldPanel as StreamFieldPanel
 from wagtail.fields import StreamField
 from wagtail.models import Page
 
@@ -42,5 +41,5 @@ class PageWithStreamfield(Page):
     )
 
     content_panels = Page.content_panels + [
-        StreamFieldPanel("body"),
+        FieldPanel("body"),
     ]

wagtail_color_panel/edit_handlers.py

@@ -4,27 +4,18 @@
 
 
 class NativeColorPanel(FieldPanel):
-    def widget_overrides(self):
-        # For Wagtail<3.0 we use widget_overrides
-        return {
-            self.field_name: ColorInputWidget(),
-        }
-
     def get_form_options(self):
-        # For Wagtail 3.0 we use get_form_options
-        # So we can mix them to provide supports to Wagtail 2,3
         opts = super().get_form_options()
-        opts["widgets"] = self.widget_overrides()
+        opts["widgets"] = {
+            self.field_name: ColorInputWidget(),
+        }
         return opts
 
 
 class PolyfillColorPanel(FieldPanel):
-    def widget_overrides(self):
-        return {
-            self.field_name: PolyfillColorInputWidget(),
-        }
-
     def get_form_options(self):
         opts = super().get_form_options()
-        opts["widgets"] = self.widget_overrides()
+        opts["widgets"] = {
+            self.field_name: PolyfillColorInputWidget(),
+        }
         return opts