mirror of
https://github.com/django-q2/django-q2.git
synced 2026-09-17 14:37:54 +08:00
c127604f10
* task now takes an optional keyword `group` * added result_group() and fetch_group() * added group column to Success admin * group column in searchable * added `name` field to Schedules * Scheduled tasks take schedule name as group
86 lines
2.0 KiB
Python
86 lines
2.0 KiB
Python
from django.contrib import admin
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from .tasks import async
|
|
from .models import Success, Failure, Schedule
|
|
|
|
|
|
class TaskAdmin(admin.ModelAdmin):
|
|
list_display = (
|
|
u'name',
|
|
'func',
|
|
'started',
|
|
'stopped',
|
|
'time_taken',
|
|
'group'
|
|
)
|
|
|
|
def has_add_permission(self, request, obj=None):
|
|
"""Don't allow adds"""
|
|
return False
|
|
|
|
def get_queryset(self, request):
|
|
"""Only show successes"""
|
|
qs = super(TaskAdmin, self).get_queryset(request)
|
|
return qs.filter(success=True)
|
|
|
|
search_fields = ('name', 'func', 'group')
|
|
readonly_fields = []
|
|
|
|
def get_readonly_fields(self, request, obj=None):
|
|
return list(self.readonly_fields) + \
|
|
[field.name for field in obj._meta.fields]
|
|
|
|
|
|
def retry_failed(FailAdmin, request, queryset):
|
|
for task in queryset:
|
|
async(task.func, *task.args or (), hook=task.hook, **task.kwargs or {})
|
|
task.delete()
|
|
|
|
|
|
retry_failed.short_description = _("Resubmit selected tasks to queue")
|
|
|
|
|
|
class FailAdmin(admin.ModelAdmin):
|
|
list_display = (
|
|
'name',
|
|
'func',
|
|
'started',
|
|
'stopped',
|
|
'result'
|
|
)
|
|
|
|
def has_add_permission(self, request, obj=None):
|
|
"""Don't allow adds"""
|
|
return False
|
|
|
|
actions = [retry_failed]
|
|
search_fields = ('name', 'func')
|
|
readonly_fields = []
|
|
|
|
def get_readonly_fields(self, request, obj=None):
|
|
return list(self.readonly_fields) + \
|
|
[field.name for field in obj._meta.fields]
|
|
|
|
|
|
class ScheduleAdmin(admin.ModelAdmin):
|
|
list_display = (
|
|
'id',
|
|
'name',
|
|
'func',
|
|
'schedule_type',
|
|
'repeats',
|
|
'next_run',
|
|
'last_run',
|
|
'success'
|
|
)
|
|
|
|
list_filter = ('next_run', 'schedule_type')
|
|
search_fields = ('func',)
|
|
list_display_links = ('id', 'name')
|
|
|
|
|
|
admin.site.register(Schedule, ScheduleAdmin)
|
|
admin.site.register(Success, TaskAdmin)
|
|
admin.site.register(Failure, FailAdmin)
|