mirror of
https://github.com/django-q2/django-q2.git
synced 2026-09-15 13:37:56 +08:00
84 lines
2.0 KiB
Python
84 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'
|
|
)
|
|
|
|
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')
|
|
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',
|
|
'func',
|
|
'schedule_type',
|
|
'repeats',
|
|
'next_run',
|
|
'last_run',
|
|
'success'
|
|
)
|
|
|
|
list_filter = ('next_run', 'schedule_type')
|
|
search_fields = ('func',)
|
|
list_display_links = ('id', 'func')
|
|
|
|
|
|
admin.site.register(Schedule, ScheduleAdmin)
|
|
admin.site.register(Success, TaskAdmin)
|
|
admin.site.register(Failure, FailAdmin)
|