django-orchestra-test/orchestra/apps/webapps/admin.py

from django import forms
from django.contrib import admin
from django.core.urlresolvers import reverse
from django.utils.translation import ugettext, ugettext_lazy as _

from orchestra.admin import ExtendedModelAdmin
from orchestra.admin.utils import change_url
from orchestra.apps.accounts.admin import AccountAdminMixin
from orchestra.forms.widgets import DynamicHelpTextSelect

from . import settings
from .models import WebApp, WebAppOption


class WebAppOptionInline(admin.TabularInline):
    model = WebAppOption
    extra = 1
    
    OPTIONS_HELP_TEXT = {
        k: str(unicode(v[1])) if len(v) == 3 else ''
            for k, v in settings.WEBAPPS_OPTIONS.iteritems()
    }
    
    class Media:
        css = {
            'all': ('orchestra/css/hide-inline-id.css',)
        }
    
    def formfield_for_dbfield(self, db_field, **kwargs):
        if db_field.name == 'value':
            kwargs['widget'] = forms.TextInput(attrs={'size':'100'})
        if db_field.name == 'name':
            # Help text based on select widget
            kwargs['widget'] = DynamicHelpTextSelect(
                'this.id.replace("name", "value")', self.OPTIONS_HELP_TEXT
            )
        return super(WebAppOptionInline, self).formfield_for_dbfield(db_field, **kwargs)


class WebAppAdmin(AccountAdminMixin, ExtendedModelAdmin):
    list_display = ('display_name', 'type', 'display_websites', 'account_link')
    list_filter = ('type',)
    add_fields = ('account', 'name', 'type')
    fields = ('account_link', 'name', 'type')
    inlines = [WebAppOptionInline]
    readonly_fields = ('account_link',)
    change_readonly_fields = ('name', 'type')
    prefetch_related = ('content_set__website',)
    
    TYPE_HELP_TEXT = {
        k: str(unicode(v.get('help_text', '')))
            for k, v in settings.WEBAPPS_TYPES.iteritems()
    }
    
    def display_name(self, webapp):
        return webapp.get_name()
    display_name.short_description = _("Name")
    display_name.admin_order_field = 'name'
    
    def display_websites(self, webapp):
        websites = []
        for content in webapp.content_set.all():
            website = content.website
            url = change_url(website)
            name = "%s on %s" % (website.name, content.path)
            websites.append('<a href="%s">%s</a>' % (url, name))
        add_url = reverse('admin:websites_website_add')
        # TODO support for preselecting related web app on website
        add_url += '?account=%s' % webapp.account_id
        plus = '<strong style="color:green; font-size:12px">+</strong>'
        websites.append('<a href="%s">%s%s</a>' % (add_url, plus, ugettext("Add website")))
        return '<br>'.join(websites)
    display_websites.short_description = _("web sites")
    display_websites.allow_tags = True
    
    def formfield_for_dbfield(self, db_field, **kwargs):
        """ Make value input widget bigger """
        if db_field.name == 'type':
            # Help text based on select widget
            kwargs['widget'] = DynamicHelpTextSelect(
                'this.id.replace("name", "value")', self.TYPE_HELP_TEXT
            )
            kwargs['help_text'] = self.TYPE_HELP_TEXT.get(db_field.default, '')
        return super(WebAppAdmin, self).formfield_for_dbfield(db_field, **kwargs)

admin.site.register(WebApp, WebAppAdmin)