django-orchestra/orchestra/contrib/databases/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, ChangePasswordAdminMixin
from orchestra.admin.utils import change_url
from orchestra.contrib.accounts.admin import SelectAccountAdminMixin

from .forms import DatabaseCreationForm, DatabaseUserChangeForm, DatabaseUserCreationForm
from .models import Database, DatabaseUser


class DatabaseAdmin(SelectAccountAdminMixin, ExtendedModelAdmin):
    list_display = ('name', 'type', 'display_users', 'account_link')
    list_filter = ('type',)
    search_fields = ('name', 'account__username')
    change_readonly_fields = ('name', 'type')
    extra = 1
    fieldsets = (
        (None, {
            'classes': ('extrapretty',),
            'fields': ('account_link', 'name', 'type', 'users', 'display_users'),
        }),
    )
    add_fieldsets = (
        (None, {
            'classes': ('wide',),
            'fields': ('account_link', 'name', 'type')
        }),
        (_("Create new user"), {
            'classes': ('wide',),
            'fields': ('username', 'password1', 'password2'),
        }),
        (_("Use existing user"), {
            'classes': ('wide',),
            'fields': ('user',)
        }),
    )
    add_form = DatabaseCreationForm
    readonly_fields = ('account_link', 'display_users',)
    filter_horizontal = ['users']
    filter_by_account_fields = ('users',)
    list_prefetch_related = ('users',)
    
    def display_users(self, db):
        links = []
        for user in db.users.all():
            link = '<a href="%s">%s</a>' % (change_url(user), user.username)
            links.append(link)
        return '<br>'.join(links)
    display_users.short_description = _("Users")
    display_users.allow_tags = True
    display_users.admin_order_field = 'users__username'
    
    def save_model(self, request, obj, form, change):
        super(DatabaseAdmin, self).save_model(request, obj, form, change)
        if not change:
            user = form.cleaned_data['user']
            if not user:
                user = DatabaseUser(
                    username=form.cleaned_data['username'],
                    type=obj.type,
                    account_id=obj.account.pk,
                )
                user.set_password(form.cleaned_data["password1"])
            user.save()
            obj.users.add(user)


class DatabaseUserAdmin(SelectAccountAdminMixin, ChangePasswordAdminMixin, ExtendedModelAdmin):
    list_display = ('username', 'type', 'display_databases', 'account_link')
    list_filter = ('type',)
    search_fields = ('username', 'account__username')
    form = DatabaseUserChangeForm
    add_form = DatabaseUserCreationForm
    change_readonly_fields = ('username', 'type')
    fieldsets = (
        (None, {
            'classes': ('extrapretty',),
            'fields': ('account_link', 'username', 'password', 'type', 'display_databases')
        }),
    )
    add_fieldsets = (
        (None, {
            'classes': ('extrapretty',),
            'fields': ('account_link', 'username', 'password1', 'password2', 'type')
        }),
    )
    readonly_fields = ('account_link', 'display_databases',)
    filter_by_account_fields = ('databases',)
    list_prefetch_related = ('databases',)
    
    def display_databases(self, user):
        links = []
        for db in user.databases.all():
            link = '<a href="%s">%s</a>' % (change_url(db), db.name)
            links.append(link)
        return '<br>'.join(links)
    display_databases.short_description = _("Databases")
    display_databases.allow_tags = True
    display_databases.admin_order_field = 'databases__name'
    
    def get_urls(self):
        useradmin = UserAdmin(DatabaseUser, self.admin_site)
        return [
            url(r'^(\d+)/password/$',
                self.admin_site.admin_view(useradmin.user_change_password))
        ] + super(DatabaseUserAdmin, self).get_urls()
    
    def save_model(self, request, obj, form, change):
        """ set password """
        if not change:
            obj.set_password(form.cleaned_data["password1"])
        super(DatabaseUserAdmin, self).save_model(request, obj, form, change)


admin.site.register(Database, DatabaseAdmin)
admin.site.register(DatabaseUser, DatabaseUserAdmin)