143 lines
5.5 KiB
Python
143 lines
5.5 KiB
Python
from django import forms
|
|
from django.contrib import admin
|
|
from django.db.models.functions import Concat, Coalesce
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from orchestra.admin import ExtendedModelAdmin
|
|
from orchestra.admin.utils import admin_link, change_url
|
|
from orchestra.contrib.accounts.admin import AccountAdminMixin
|
|
from orchestra.utils import apps
|
|
|
|
from .actions import view_zone
|
|
from .filters import TopDomainListFilter
|
|
from .forms import RecordInlineFormSet, BatchDomainCreationAdminForm
|
|
from .models import Domain, Record
|
|
|
|
|
|
class RecordInline(admin.TabularInline):
|
|
model = Record
|
|
formset = RecordInlineFormSet
|
|
verbose_name_plural = _("Extra records")
|
|
|
|
def formfield_for_dbfield(self, db_field, **kwargs):
|
|
""" Make value input widget bigger """
|
|
if db_field.name == 'value':
|
|
kwargs['widget'] = forms.TextInput(attrs={'size':'100'})
|
|
if db_field.name == 'ttl':
|
|
kwargs['widget'] = forms.TextInput(attrs={'size':'10'})
|
|
return super(RecordInline, self).formfield_for_dbfield(db_field, **kwargs)
|
|
|
|
|
|
class DomainInline(admin.TabularInline):
|
|
model = Domain
|
|
fields = ('domain_link', 'display_records', 'account_link')
|
|
readonly_fields = ('domain_link', 'display_records', 'account_link')
|
|
extra = 0
|
|
verbose_name_plural = _("Subdomains")
|
|
|
|
domain_link = admin_link('__str__')
|
|
domain_link.short_description = _("Name")
|
|
account_link = admin_link('account')
|
|
|
|
def display_records(self, domain):
|
|
return ', '.join([record.type for record in domain.records.all()])
|
|
display_records.short_description = _("Declared records")
|
|
|
|
def has_add_permission(self, *args, **kwargs):
|
|
return False
|
|
|
|
def get_queryset(self, request):
|
|
""" Order by structured name and imporve performance """
|
|
qs = super(DomainInline, self).get_queryset(request)
|
|
return qs.select_related('account').prefetch_related('records')
|
|
|
|
|
|
class DomainAdmin(AccountAdminMixin, ExtendedModelAdmin):
|
|
list_display = (
|
|
'structured_name', 'display_is_top', 'display_websites', 'account_link'
|
|
)
|
|
add_fields = ('name', 'account')
|
|
fields = ('name', 'account_link')
|
|
inlines = [RecordInline, DomainInline]
|
|
list_filter = [TopDomainListFilter]
|
|
change_readonly_fields = ('name',)
|
|
search_fields = ('name', 'account__username')
|
|
add_form = BatchDomainCreationAdminForm
|
|
change_view_actions = [view_zone]
|
|
|
|
def structured_name(self, domain):
|
|
if domain.is_top:
|
|
return domain.name
|
|
return ' '*4 + domain.name
|
|
structured_name.short_description = _("name")
|
|
structured_name.allow_tags = True
|
|
structured_name.admin_order_field = 'structured_name'
|
|
|
|
def display_is_top(self, domain):
|
|
return domain.is_top
|
|
display_is_top.short_description = _("Is top")
|
|
display_is_top.boolean = True
|
|
display_is_top.admin_order_field = 'top'
|
|
|
|
def display_websites(self, domain):
|
|
if apps.isinstalled('orchestra.contrib.websites'):
|
|
webs = domain.websites.all()
|
|
if webs:
|
|
links = []
|
|
for web in webs:
|
|
url = change_url(web)
|
|
links.append('<a href="%s">%s</a>' % (url, web.name))
|
|
return '<br>'.join(links)
|
|
return _("No website")
|
|
display_websites.admin_order_field = 'websites__name'
|
|
display_websites.short_description = _("Websites")
|
|
display_websites.allow_tags = True
|
|
|
|
def get_queryset(self, request):
|
|
""" Order by structured name and imporve performance """
|
|
qs = super(DomainAdmin, self).get_queryset(request)
|
|
qs = qs.select_related('top', 'account')
|
|
if request.method == 'GET':
|
|
qs = qs.annotate(
|
|
structured_id=Coalesce('top__id', 'id'),
|
|
structured_name=Concat('top__name', 'name')
|
|
).order_by('-structured_id', 'structured_name')
|
|
if apps.isinstalled('orchestra.contrib.websites'):
|
|
qs = qs.prefetch_related('websites')
|
|
return qs
|
|
|
|
def save_model(self, request, obj, form, change):
|
|
""" batch domain creation support """
|
|
super(DomainAdmin, self).save_model(request, obj, form, change)
|
|
self.extra_domains = []
|
|
if not change:
|
|
for name in form.extra_names:
|
|
domain = Domain.objects.create(name=name, account_id=obj.account_id)
|
|
self.extra_domains.append(domain)
|
|
|
|
def save_formset(self, request, form, formset, change):
|
|
"""
|
|
Given an inline formset save it to the database.
|
|
"""
|
|
formset.save()
|
|
|
|
def save_related(self, request, form, formsets, change):
|
|
""" batch domain creation support """
|
|
super(DomainAdmin, self).save_related(request, form, formsets, change)
|
|
if not change:
|
|
# Clone records to extra_domains, if any
|
|
for formset in formsets:
|
|
if formset.model is Record:
|
|
for domain in self.extra_domains:
|
|
# Reset pk value of the record instances to force creation of new ones
|
|
for record_form in formset.forms:
|
|
record = record_form.instance
|
|
if record.pk:
|
|
record.pk = None
|
|
formset.instance = domain
|
|
form.instance = domain
|
|
self.save_formset(request, form, formset, change=change)
|
|
|
|
|
|
admin.site.register(Domain, DomainAdmin)
|