receipts

[wolnelektury.git] / src / catalogue / fields.py

diff --git a/src/catalogue/fields.py b/src/catalogue/fields.py
index 4487549..581bf38 100644 (file)
--- a/src/catalogue/fields.py
+++ b/src/catalogue/fields.py
@@ -56,7 +56,6 @@ class EbookField(models.FileField):
     """Represents an ebook file field, attachable to a model."""
     attr_class = EbookFieldFile
     registry = []
     """Represents an ebook file field, attachable to a model."""
     attr_class = EbookFieldFile
     registry = []

-    librarian2_api = False

 
     def __init__(self, format_name, *args, **kwargs):
         super(EbookField, self).__init__(*args, **kwargs)
 
     def __init__(self, format_name, *args, **kwargs):
         super(EbookField, self).__init__(*args, **kwargs)


@@ -125,6 +124,8 @@ class EbookField(models.FileField):
 
 
 class BuildEbook(Task):
 
 
 class BuildEbook(Task):

+    librarian2_api = False
+

     formats = {}
 
     @classmethod
     formats = {}
 
     @classmethod


@@ -219,9 +220,15 @@ class BuildEpub(BuildEbook):
 @BuildEbook.register('mobi')
 @task(ignore_result=True)
 class BuildMobi(BuildEbook):
 @BuildEbook.register('mobi')
 @task(ignore_result=True)
 class BuildMobi(BuildEbook):

+    librarian2_api = True
+

     @staticmethod
     def transform(wldoc, fieldfile):
     @staticmethod
     def transform(wldoc, fieldfile):

-        return wldoc.as_mobi(cover=True, base_url=absolute_url(gallery_url(wldoc.book_info.url.slug)))
+        from librarian.builders import MobiBuilder
+        return MobiBuilder(
+                base_url='file://' + os.path.abspath(gallery_path(wldoc.meta.url.slug)) + '/',
+                fundraising=settings.EPUB_FUNDRAISING
+            ).build(wldoc)

 
 
 @BuildEbook.register('html')
 
 
 @BuildEbook.register('html')