update requirements

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

diff --git a/apps/toolbar/admin.py b/apps/toolbar/admin.py
index a03f393..654480c 100644 (file)
--- a/apps/toolbar/admin.py
+++ b/apps/toolbar/admin.py
@@ -1,8 +1,31 @@
 from django.contrib import admin
 from django.utils.translation import ugettext_lazy as _
+from django import forms
+import 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
+        exclude = []
+
+    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)