From 6bc19d4206b34e7710f1bd807a1307e737367183 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Marek=20St=C4=99pniowski?= <marek@stepniowski.com>
Date: Sat, 29 Nov 2008 23:40:20 +0100
Subject: [PATCH] Changed JSONField implementation. Now to get and set decoded
 value you must use get_%s_value and set_%s_value accordingly. Normal field
 access returns jsonified value.

This fixes a bug during saving book info in admin.
---
 apps/catalogue/fields.py                      | 47 ++++++++++---------
 apps/catalogue/models.py                      |  2 +-
 apps/catalogue/views.py                       |  1 +
 .../templates/catalogue/book_detail.html      |  6 +--
 4 files changed, 30 insertions(+), 26 deletions(-)

diff --git a/apps/catalogue/fields.py b/apps/catalogue/fields.py
index d091b77c1..eedff6705 100644
--- a/apps/catalogue/fields.py
+++ b/apps/catalogue/fields.py
@@ -31,36 +31,39 @@ def loads(str):
     return json.loads(str, encoding=settings.DEFAULT_CHARSET)
 
 
+class JSONFormField(forms.CharField):
+    widget = forms.Textarea
+    
+    def clean(self, value):
+        try:
+            loads(value)
+            return value
+        except ValueError, e:
+            raise forms.ValidationError('Enter a valid JSON value. Error: %s' % e)
+
+
 class JSONField(models.TextField):
+    def formfield(self, **kwargs):
+        defaults = {'form_class': JSONFormField}
+        defaults.update(kwargs)
+        return super(JSONField, self).formfield(**defaults)
+
     def db_type(self):
         return 'text'
-    
+
     def get_internal_type(self):
         return 'TextField'
 
-    def pre_save(self, model_instance, add):
-        value = getattr(model_instance, self.attname, None)
-        return dumps(value)
-
     def contribute_to_class(self, cls, name):
         super(JSONField, self).contribute_to_class(cls, name)
-        signals.post_init.connect(self.post_init, sender=cls)
-
-        def get_json(model_instance):
-            return dumps(getattr(model_instance, self.attname, None))
-        setattr(cls, 'get_%s_json' % self.name, get_json)
-
-        def set_json(model_instance, json):
-            return setattr(model_instance, self.attname, loads(json))
-        setattr(cls, 'set_%s_json' % self.name, set_json)
-
-    def post_init(self, **kwargs):
-        instance = kwargs.get('instance', None)
-        value = self.value_from_object(instance)
-        if (value):
-            setattr(instance, self.attname, loads(value))
-        else:
-            setattr(instance, self.attname, None)
+        
+        def get_value(model_instance):
+            return loads(getattr(model_instance, self.attname, None))
+        setattr(cls, 'get_%s_value' % self.name, get_value)
+
+        def set_value(model_instance, json):
+            return setattr(model_instance, self.attname, dumps(json))
+        setattr(cls, 'set_%s_value' % self.name, set_value)
 
 
 class JQueryAutoCompleteWidget(forms.TextInput):
diff --git a/apps/catalogue/models.py b/apps/catalogue/models.py
index 7be0f826d..0e0e98ff6 100644
--- a/apps/catalogue/models.py
+++ b/apps/catalogue/models.py
@@ -173,7 +173,7 @@ class Book(models.Model):
             book_shelves = list(book.tags.filter(category='set'))
         
         book.title = book_info.title
-        book.extra_info = book_info.to_dict()
+        book.set_extra_info_value(book_info.to_dict())
         book._short_html = ''
         book.save()
         
diff --git a/apps/catalogue/views.py b/apps/catalogue/views.py
index 1718d9765..7870cba4f 100644
--- a/apps/catalogue/views.py
+++ b/apps/catalogue/views.py
@@ -115,6 +115,7 @@ def book_detail(request, slug):
     book_children = book.children.all().order_by('parent_number')
     extra_where = 'catalogue_tag.category = "theme"'
     book_themes = models.Tag.objects.related_for_model(book_tag, models.Fragment, counts=True, extra={'where': [extra_where]})
+    extra_info = book.get_extra_info_value()
     
     form = forms.SearchForm()
     return render_to_response('catalogue/book_detail.html', locals(),
diff --git a/wolnelektury/templates/catalogue/book_detail.html b/wolnelektury/templates/catalogue/book_detail.html
index 2f14576d5..07c72bc01 100644
--- a/wolnelektury/templates/catalogue/book_detail.html
+++ b/wolnelektury/templates/catalogue/book_detail.html
@@ -12,7 +12,7 @@
     </form>
     
     <div id="books-list">
-        <p>Na podstawie: {{ book.extra_info.source_name }}</p>
+        <p>Na podstawie: {{ extra_info.source_name }}</p>
         {% if book.has_description %}
             <div id="description">
                 {{ book.description|safe }}
@@ -90,8 +90,8 @@
                     <a href="{{ tag.get_absolute_url }}">{{ tag }}</a>
                     {% endfor %}
                 </li>
-                <li><a href="{{ book.extra_info.about }}">Lektura na wiki projektu</a></li>
-                <li><a href="{{ book.extra_info.source_url }}">Lektura w CBN Polona</a></li>
+                <li><a href="{{ extra_info.about }}">Lektura na wiki projektu</a></li>
+                <li><a href="{{ extra_info.source_url }}">Lektura w CBN Polona</a></li>
                 <li><a href="{{ book.xml_file.url }}">Kod źródłowy utworu (XML)</a></li>
             </ul>
         </div>
-- 
2.20.1