Filtering books by status in admin.

[redakcja.git] / apps / toolbar / admin.py

diff --git a/apps/toolbar/admin.py b/apps/toolbar/admin.py
index a03f393..283ab78 100644 (file)
--- a/apps/toolbar/admin.py
+++ b/apps/toolbar/admin.py
@@ -1,8 +1,30 @@
 from django.contrib import admin
 from django.utils.translation import ugettext_lazy as _
 from django.contrib import admin
 from django.utils.translation import ugettext_lazy as _

+from django import forms
+from django.utils import simplejson as json

 
 

-from explorer import models
+from toolbar import models

 
 

-admin.site.register(models.Book)
-admin.site.register(models.PullRequest)

 
 

+class ButtonAdminForm(forms.ModelForm):
+    class Meta:
+        model = models.Button
+
+    def clean_params(self):
+        value = self.cleaned_data['params']
+        try:
+            return json.dumps(json.loads(value))
+        except ValueError, e:
+            raise forms.ValidationError(e)
+
+
+class ButtonAdmin(admin.ModelAdmin):
+    form = ButtonAdminForm
+    list_display = ('slug', 'label', 'tooltip', 'accesskey')
+    list_display_links = ('slug',)
+    list_editable = ('label', 'tooltip', 'accesskey')
+    prepopulated_fields = {'slug': ('label',)}
+
+admin.site.register(models.Button, ButtonAdmin)
+admin.site.register(models.ButtonGroup)
+admin.site.register(models.Scriptlet)