@@ -842,26 +842,41 @@ from django_celery_beat.models import (
842842 PeriodicTask,
843843 SolarSchedule,
844844)
845-
845+ from django_celery_beat.admin import ClockedScheduleAdmin as BaseClockedScheduleAdmin
846+ from django_celery_beat.admin import CrontabScheduleAdmin as BaseCrontabScheduleAdmin
847+ from django_celery_beat.admin import PeriodicTaskAdmin as BasePeriodicTaskAdmin
848+ from django_celery_beat.admin import PeriodicTaskForm, TaskSelectWidget
846849
847850admin.site.unregister(PeriodicTask)
848851admin.site.unregister(IntervalSchedule)
849852admin.site.unregister(CrontabSchedule)
850853admin.site.unregister(SolarSchedule)
851854admin.site.unregister(ClockedSchedule)
852855
853- @admin.register (PeriodicTask)
854- class PeriodicTaskAdmin ( ModelAdmin ):
856+
857+ class UnfoldTaskSelectWidget ( UnfoldAdminSelectWidget , TaskSelectWidget ):
855858 pass
856859
857860
861+ class UnfoldPeriodicTaskForm (PeriodicTaskForm ):
862+ def __init__ (self , * args , ** kwargs ):
863+ super ().__init__ (* args, ** kwargs)
864+ self .fields[" task" ].widget = UnfoldAdminTextInputWidget()
865+ self .fields[" regtask" ].widget = UnfoldTaskSelectWidget()
866+
867+
868+ @admin.register (PeriodicTask)
869+ class PeriodicTaskAdmin (BasePeriodicTaskAdmin , ModelAdmin ):
870+ form = UnfoldPeriodicTaskForm
871+
872+
858873@admin.register (IntervalSchedule)
859874class IntervalScheduleAdmin (ModelAdmin ):
860875 pass
861876
862877
863878@admin.register (CrontabSchedule)
864- class CrontabScheduleAdmin (ModelAdmin ):
879+ class CrontabScheduleAdmin (BaseCrontabScheduleAdmin , ModelAdmin ):
865880 pass
866881
867882
@@ -870,7 +885,7 @@ class SolarScheduleAdmin(ModelAdmin):
870885 pass
871886
872887@admin.register (ClockedSchedule)
873- class ClockedScheduleAdmin (ModelAdmin ):
888+ class ClockedScheduleAdmin (BaseClockedScheduleAdmin , ModelAdmin ):
874889 pass
875890```
876891
0 commit comments