book sorting fix

[redakcja.git] / apps / catalogue / models / chunk.py

diff --git a/apps/catalogue/models/chunk.py b/apps/catalogue/models/chunk.py
index 00e18d9..9b1dcba 100755 (executable)
--- a/apps/catalogue/models/chunk.py
+++ b/apps/catalogue/models/chunk.py
@@ -11,6 +11,7 @@ from django.utils.translation import ugettext_lazy as _
 from catalogue.helpers import cached_in_field
 from catalogue.managers import VisibleManager
 from catalogue.tasks import refresh_instance
+from catalogue.xml_tools import GradedText
 from dvcs import models as dvcs_models
 
 
@@ -20,8 +21,9 @@ class Chunk(dvcs_models.Document):
 
     book = models.ForeignKey('Book', editable=False, verbose_name=_('book'))
     number = models.IntegerField(_('number'))
-    slug = models.SlugField(_('slug'))
     title = models.CharField(_('title'), max_length=255, blank=True)
+    slug = models.SlugField(_('slug'))
+    gallery_start = models.IntegerField(_('gallery start'), null=True, blank=True)
 
     # cache
     _short_html = models.TextField(null=True, blank=True, editable=False)
@@ -62,7 +64,7 @@ class Chunk(dvcs_models.Document):
     # Creating and manipulation
     # =========================
 
-    def split(self, slug, title='', adjust_slug=False, **kwargs):
+    def split(self, slug, title='', **kwargs):
         """ Create an empty chunk after this one """
         self.book.chunk_set.filter(number__gt=self.number).update(
                 number=models.F('number')+1)
@@ -122,3 +124,6 @@ class Chunk(dvcs_models.Document):
         self.changed
         self.hidden
         self.short_html
+
+    def graded(self, master=None):
+        return GradedText(self.materialize(), master)