X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/dba809feb44cd1c4d155f3e3254a1cca5323f95f..f73ecfb6b870b18ed88303ff71a06da55001274b:/apps/toolbar/models.py

diff --git a/apps/toolbar/models.py b/apps/toolbar/models.py
index 04146223..a23e3463 100644
--- a/apps/toolbar/models.py
+++ b/apps/toolbar/models.py
@@ -1,3 +1,8 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
 
@@ -6,79 +11,65 @@ class ButtonGroup(models.Model):
     name = models.CharField(max_length=32)
     slug = models.SlugField()
     position = models.IntegerField(default=0)
-    
+
     class Meta:
         ordering = ('position', 'name',)
         verbose_name, verbose_name_plural = _('button group'), _('button groups')
-    
+
     def __unicode__(self):
         return self.name
 
+    def to_dict(self, with_buttons=False):
+        d = {'name': self.name, 'position': self.position}
+
+        if with_buttons:
+            d['buttons'] = [b.to_dict() for b in self.button_set.all()]
+
+        return d
 
-#class ButtonGroupManager(models.Manager):
-#
-#    def with_buttons(self):
-#        from django.db import connection
-#        cursor = connection.cursor()
-#        cursor.execute("""
-#            SELECT g.name, g.slug, CONCAT(b.slug),
-#            FROM toolbar_buttongroup as g LEFT JOIN toolbar_button as b
-#
-#            WHERE p.id = r.poll_id
-#            GROUP BY 1, 2, 3
-#            ORDER BY 3 DESC""")
-#        result_list = []
-#        for row in cursor.fetchall():
-#            p = self.model(id=row[0], question=row[1], poll_date=row[2])
-#            p.num_responses = row[3]
-#            result_list.append(p)
-#        return result_list
 
 class Button(models.Model):
     label = models.CharField(max_length=32)
-    slug = models.SlugField(unique=True) #unused
+    slug = models.SlugField(unique=True)  # unused
 
     # behaviour
-    params = models.TextField(default='[]') # TODO: should be a JSON field
+    params = models.TextField(default='[]')  # TODO: should be a JSON field
     scriptlet = models.ForeignKey('Scriptlet', null=True, blank=True)
     link = models.CharField(max_length=256, blank=True, default='')
 
     # ui related stuff
-    key = models.CharField(blank=True, max_length=1)
-    key_mod = models.PositiveIntegerField(blank=True, default=1)
+    accesskey = models.CharField(blank=True, max_length=1)
+
     tooltip = models.CharField(blank=True, max_length=120)
 
     # Why the button is restricted to have the same position in each group ?
-    # position = models.IntegerField(default=0)   
+    # position = models.IntegerField(default=0)
     group = models.ManyToManyField(ButtonGroup)
-    
+
     class Meta:
         ordering = ('slug',)
         verbose_name, verbose_name_plural = _('button'), _('buttons')
 
-    def hotkey_code(self):
-        return ord(self.key.upper()) | (self.key_mod << 8)
+    @property
+    def full_tooltip(self):
+        return u"%s %s" % (self.tooltip, "[Alt+%s]" % self.accesskey if self.accesskey else "")
 
-    def hotkey_name(self):
-        if not self.key:
-            return ''
+    def to_dict(self):
+        return {
+            'label': self.label,
+            'tooltip': self.tooltip,
+            'accesskey': self.accesskey,
+            'params': self.params,
+            'scriptlet_id': self.scriptlet_id,
+        }
 
-        mods = []
-        if self.key_mod & 0x01: mods.append('Alt')
-        if self.key_mod & 0x02: mods.append('Ctrl')
-        if self.key_mod & 0x04: mods.append('Shift')
-        mods.append('"'+self.key+'"')
-        return '+'.join(mods)
-    
     def __unicode__(self):
         return self.label
 
+
 class Scriptlet(models.Model):
     name = models.CharField(max_length=64, primary_key=True)
     code = models.TextField()
 
-    # TODO: add this later and remap code property to this
-    # code_min = models.TextField()
-
     def __unicode__(self):
-        return _(u'javascript')+u':'+self.name
+        return _(u'javascript') + u':' + self.name