django-orchestra/orchestra/contrib/vps/admin.py

from django.conf.urls import url
from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from django.utils.translation import ugettext_lazy as _

from orchestra.admin import ExtendedModelAdmin
from orchestra.contrib.accounts.actions import list_accounts
from orchestra.contrib.accounts.admin import AccountAdminMixin
from orchestra.forms import UserCreationForm, NonStoredUserChangeForm

from .models import VPS


class VPSAdmin(AccountAdminMixin, ExtendedModelAdmin):
    list_display = ('hostname', 'type', 'template', 'account_link')
    list_filter = ('type', 'template')
    form = NonStoredUserChangeForm
    add_form = UserCreationForm
    readonly_fields = ('account_link',)
    change_readonly_fields = ('account', 'hostname', 'type', 'template')
    fieldsets = (
        (None, {
            'classes': ('wide',),
            'fields': ('account_link', 'hostname', 'type', 'template')
        }),
        (_("Login"), {
            'classes': ('wide',),
            'fields': ('password',)
        })
    )
    add_fieldsets = (
        (None, {
            'classes': ('wide',),
            'fields': ('account', 'hostname', 'type', 'template')
        }),
        (_("Login"), {
            'classes': ('wide',),
            'fields': ('password1', 'password2',)
        }),
    )
    actions = (list_accounts,)
    
    def get_urls(self):
        useradmin = UserAdmin(VPS, self.admin_site)
        return [
            url(r'^(\d+)/password/$',
                self.admin_site.admin_view(useradmin.user_change_password))
        ] + super(VPSAdmin, self).get_urls()


admin.site.register(VPS, VPSAdmin)