Legimi categories.

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

diff --git a/src/depot/legimi.py b/src/depot/legimi.py
index a41902a..5fa040c 100644 (file)
--- a/src/depot/legimi.py
+++ b/src/depot/legimi.py
@@ -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: