116 lines
5.1 KiB
Plaintext
116 lines
5.1 KiB
Plaintext
./apps/vps/admin.pyfrom django.db import models from django.conf.urls import patterns from django.contrib import admin from django.contrib.auth.admin import
|
|
UserAdmin from django.core.urlresolvers import reverse from django.utils.safestring import mark_safe from django.utils.translation import ugettext_lazy as _
|
|
./apps/issues/admin.py from orchestra.admin import ExtendedModelAdmin from orchestra.admin.utils import link from orchestra.apps.accounts.admin import
|
|
AccountAdminMixin, SelectAccountAdminMixin ./apps/contacts/admin.py from .forms import (DatabaseUserChangeForm, DatabaseUserCreationForm, ./apps/webapps/admin.py
|
|
DatabaseCreationForm) from .models import Database, Role, DatabaseUser ./apps/prices/admin.py ./apps/websites/admin.py class UserInline(admin.TabularInline):
|
|
./apps/users/roles/posix/admin.py model = Role ./apps/users/roles/admin.py verbose_name_plural = _("Users") ./apps/users/roles/jabber/admin.py readonly_fields =
|
|
('user_link',) ./apps/users/roles/mail/admin.py extra = 0 ./apps/users/admin.py ./apps/orchestration/admin.py user_link = link('user') ./apps/orders/admin.py
|
|
./apps/domains/admin.py def formfield_for_dbfield(self, db_field, **kwargs): ./apps/accounts/admin.py """ Make value input widget bigger """ ./apps/lists/admin.py if
|
|
db_field.name == 'user': ./apps/resources/admin.py users = db_field.rel.to.objects.filter(type=self.parent_object.type)
|
|
kwargs['queryset'] = users.filter(account=self.account)
|
|
return super(UserInline, self).formfield_for_dbfield(db_field, **kwargs)
|
|
|
|
|
|
class PermissionInline(AccountAdminMixin, admin.TabularInline):
|
|
model = Role
|
|
verbose_name_plural = _("Permissions")
|
|
readonly_fields = ('database_link',)
|
|
extra = 0
|
|
filter_by_account_fields = ['database']
|
|
|
|
database_link = link('database', popup=True)
|
|
|
|
def formfield_for_dbfield(self, db_field, **kwargs):
|
|
""" Make value input widget bigger """
|
|
formfield = super(PermissionInline, self).formfield_for_dbfield(db_field, **kwargs)
|
|
if db_field.name == 'database':
|
|
# Hack widget render in order to append ?account=id to the add url
|
|
db_type = self.parent_object.type
|
|
old_render = formfield.widget.render
|
|
def render(*args, **kwargs):
|
|
output = old_render(*args, **kwargs)
|
|
output = output.replace('/add/?', '/add/?type=%s&' % db_type)
|
|
return mark_safe(output)
|
|
formfield.widget.render = render
|
|
formfield.queryset = formfield.queryset.filter(type=db_type)
|
|
return formfield
|
|
|
|
|
|
class DatabaseAdmin(SelectAccountAdminMixin, ExtendedModelAdmin):
|
|
list_display = ('name', 'type', 'account_link')
|
|
list_filter = ('type',)
|
|
search_fields = ['name', 'account__user__username']
|
|
inlines = [UserInline]
|
|
add_inlines = []
|
|
change_readonly_fields = ('name', 'type')
|
|
extra = 1
|
|
fieldsets = (
|
|
(None, {
|
|
'classes': ('extrapretty',),
|
|
'fields': ('account_link', 'name', 'type'),
|
|
}),
|
|
)
|
|
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
|
|
|
|
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.objects.create(
|
|
username=form.cleaned_data['username'],
|
|
type=obj.type,
|
|
account_id = obj.account.pk,
|
|
)
|
|
user.set_password(form.cleaned_data["password1"])
|
|
user.save()
|
|
Role.objects.create(database=obj, user=user, is_owner=True)
|
|
|
|
|
|
class DatabaseUserAdmin(SelectAccountAdminMixin, ExtendedModelAdmin):
|
|
list_display = ('username', 'type', 'account_link')
|
|
list_filter = ('type',)
|
|
search_fields = ['username', 'account__user__username']
|
|
form = DatabaseUserChangeForm
|
|
add_form = DatabaseUserCreationForm
|
|
change_readonly_fields = ('username', 'type')
|
|
inlines = [PermissionInline]
|
|
add_inlines = []
|
|
fieldsets = (
|
|
(None, {
|
|
'classes': ('extrapretty',),
|
|
'fields': ('account_link', 'username', 'password', 'type')
|
|
}),
|
|
)
|
|
add_fieldsets = (
|
|
(None, {
|
|
'classes': ('extrapretty',),
|
|
'fields': ('account_link', 'username', 'password1', 'password2', 'type')
|
|
}),
|
|
)
|
|
|
|
def get_urls(self):
|
|
useradmin = UserAdmin(DatabaseUser, self.admin_site)
|
|
return patterns('',
|
|
(r'^(\d+)/password/$',
|
|
self.admin_site.admin_view(useradmin.user_change_password))
|
|
) + super(DatabaseUserAdmin, self).get_urls()
|
|
|
|
|
|
admin.site.register(Database, DatabaseAdmin)
|
|
admin.site.register(DatabaseUser, DatabaseUserAdmin)
|