Switch cover for legimi.

[redakcja.git] / src / depot / legimi.py

diff --git a/src/depot/legimi.py b/src/depot/legimi.py
index a41902a..75ddaa1 100644 (file)
--- a/src/depot/legimi.py
+++ b/src/depot/legimi.py
@@ -4,7 +4,7 @@ from django.conf import settings
 from librarian.functions import lang_code_3to2
 from librarian.html import transform_abstrakt
 from librarian.builders import EpubBuilder, MobiBuilder
-from librarian.cover import LegimiCornerCover, LegimiCover
+from librarian.covers.marquise import MarquiseCover, LabelMarquiseCover
 import requests
 from slugify import slugify
 
@@ -77,9 +77,9 @@ class Legimi:
         wlbook = book.wldocument(librarian2=True)
         meta = wlbook.meta
 
-        cover = LegimiCornerCover(meta, width=1200).output_file()
-        epub_file = EpubBuilder(cover=LegimiCover, fundraising=fundraising).build(wlbook).get_file()
-        mobi_file = MobiBuilder(cover=LegimiCover, fundraising=fundraising).build(wlbook).get_file()
+        cover = LabelMarquiseCover(meta, width=1200).output_file()
+        epub_file = EpubBuilder(cover=MarquiseCover, fundraising=fundraising).build(wlbook).get_file()
+        mobi_file = MobiBuilder(cover=MarquiseCover, fundraising=fundraising).build(wlbook).get_file()
 
         book_data = {
             "Title": meta.title,
@@ -185,7 +185,7 @@ class Legimi:
 
         if wlbook.meta.audience:
             description += '<p><em>{}</em> to lektura szkolna.'.format(wlbook.meta.title)
-            if wlbook.tree.find('//pe'):
+            if wlbook.tree.find('//pe') is not None:
                 description += '<br>Ebook <em>{title}</em> zawiera przypisy opracowane specjalnie dla uczennic i uczniów {school}.'.format(
                     title=wlbook.meta.title,
                     school='szkoły podstawowej' if wlbook.meta.audience == 'SP' else 'liceum i technikum'
@@ -195,16 +195,16 @@ class Legimi:
 
     def get_genre(self, wlbook):
         epoch_map = {
-            'Starożytność': 12,
-            'Średniowiecze': 16, 
-            'Renesans': 20,
-            'Barok': 13,
-            'Oświecenie': 14,
-            'Romantyzm': 21,
-            'Pozytywizm': 19,
-            'Modernizm': 18,
-            'Dwudziestolecie międzywojenne': 15,
-            'Współczesność': 17,
+            'Starożytność': 80,
+            'Średniowiecze': 81,
+            'Renesans': 82,
+            'Barok': 83,
+            'Oświecenie': 84,
+            'Romantyzm': 85,
+            'Pozytywizm': 86,
+            'Modernizm': 87,
+            'Dwudziestolecie międzywojenne': 88,
+            'Współczesność': 90,
         }
 
         for epoch in wlbook.meta.epochs: