Fix existing tests.

[redakcja.git] / src / documents / models / book.py

diff --git a/src/documents/models/book.py b/src/documents/models/book.py
index e649180..23d3344 100644 (file)
--- a/src/documents/models/book.py
+++ b/src/documents/models/book.py
@@ -6,7 +6,7 @@ from django.contrib.sites.models import Site
 from django.db import connection, models, transaction
 from django.template.loader import render_to_string
 from django.urls import reverse
 from django.db import connection, models, transaction
 from django.template.loader import render_to_string
 from django.urls import reverse

-from django.utils.translation import ugettext_lazy as _
+from django.utils.translation import gettext_lazy as _

 from django.conf import settings
 from slugify import slugify
 
 from django.conf import settings
 from slugify import slugify
 


@@ -52,6 +52,7 @@ class Book(models.Model):
         related_name='document_books',
         related_query_name='document_book',
     )
         related_name='document_books',
         related_query_name='document_book',
     )

+    legimi_id = models.CharField(max_length=255, blank=True)

 
     class NoTextError(BaseException):
         pass
 
     class NoTextError(BaseException):
         pass


@@ -277,6 +278,9 @@ class Book(models.Model):
         except IndexError:
             return None
 
         except IndexError:
             return None
 

+    def last_legimi_publish(self):
+        return self.legimibookpublish_set.order_by('-created_at').first()
+

     def assert_publishable(self):
         assert self.chunk_set.exists(), _('No chunks in the book.')
         try:
     def assert_publishable(self):
         assert self.chunk_set.exists(), _('No chunks in the book.')
         try: