27 lines
933 B
Python
27 lines
933 B
Python
from admin_extra_buttons.api import ExtraButtonsMixin, button
|
|
from admin_extra_buttons.utils import HttpResponseRedirectToReferrer
|
|
from django.contrib import admin
|
|
from .models import TLD, RootZone
|
|
import tldtester.sorter as sorter
|
|
import threading
|
|
|
|
|
|
class tlds(ExtraButtonsMixin, admin.ModelAdmin):
|
|
list_display = ('tld', 'nsamount', 'v4nsamount', 'v6nsamount', 'dnssec', 'atlasv4', 'atlasv6', 'lastEdition')
|
|
|
|
@button(change_form=True, html_attrs={'style': 'background-color:#88FF88;color:black'})
|
|
def refresh(self, request):
|
|
self.message_user(request, 'refresh called')
|
|
t1 = threading.Thread(target=sorter.main())
|
|
t1.start()
|
|
# Optional: returns HttpResponse
|
|
return HttpResponseRedirectToReferrer(request)
|
|
|
|
|
|
class RootZones(admin.ModelAdmin):
|
|
list_display = ('name', 'rectype', 'value', 'lastEdition')
|
|
|
|
|
|
admin.site.register(TLD, tlds)
|
|
admin.site.register(RootZone, RootZones)
|