import re from django import forms from django.contrib import admin from django.utils.translation import ugettext_lazy as _ from orchestra.admin import ChangeListDefaultFilter, ExtendedModelAdmin from orchestra.admin.utils import admin_link, change_url from orchestra.apps.accounts.admin import AccountAdminMixin from orchestra.utils import apps from .actions import view_zone from .forms import RecordInlineFormSet, DomainAdminForm from .filters import TopDomainListFilter from .models import Domain, Record class RecordInline(admin.TabularInline): model = Record formset = RecordInlineFormSet verbose_name_plural = _("Extra records") class Media: css = { 'all': ('orchestra/css/hide-inline-id.css',) } 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'}) return super(RecordInline, self).formfield_for_dbfield(db_field, **kwargs) class DomainInline(admin.TabularInline): model = Domain fields = ('domain_link',) readonly_fields = ('domain_link',) extra = 0 verbose_name_plural = _("Subdomains") domain_link = admin_link() domain_link.short_description = _("Name") def has_add_permission(self, *args, **kwargs): return False class DomainAdmin(ChangeListDefaultFilter, AccountAdminMixin, ExtendedModelAdmin): fields = ('name', 'account') list_display = ( 'structured_name', 'display_is_top', 'websites', 'account_link' ) inlines = [RecordInline, DomainInline] list_filter = [TopDomainListFilter] change_readonly_fields = ('name',) search_fields = ['name',] default_changelist_filters = ( ('top_domain', 'True'), ) form = DomainAdminForm change_view_actions = [view_zone] def structured_name(self, domain): if not domain.is_top: return ' '*4 + domain.name return 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.boolean = True display_is_top.admin_order_field = 'top' def websites(self, domain): if apps.isinstalled('orchestra.apps.websites'): webs = domain.websites.all() if webs: links = [] for web in webs: url = change_url(web) links.append('%s' % (url, web.name)) return '
'.join(links) return _("No website") websites.admin_order_field = 'websites__name' websites.short_description = _("Websites") 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') # For some reason if we do this we know for sure that join table will be called T4 query = str(qs.query) table = re.findall(r'(T\d+)\."account_id"', query)[0] qs = qs.extra( select={ 'structured_name': 'CONCAT({table}.name, domains_domain.name)'.format(table=table) }, ).order_by('structured_name') if apps.isinstalled('orchestra.apps.websites'): qs = qs.prefetch_related('websites') return qs admin.site.register(Domain, DomainAdmin)