Allow multiple sources.

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

diff --git a/src/depot/publishers/legimi.py b/src/depot/publishers/legimi.py
index 6802d14..da22245 100644 (file)
--- a/src/depot/publishers/legimi.py
+++ b/src/depot/publishers/legimi.py
@@ -130,12 +130,16 @@ class Legimi(BasePublisher):
             })
 
     def can_publish(self, site, book):
-        meta = book.wldocument(librarian2=True).meta
         d = {
             'errors': [],
             'warnings': [],
             'info': []
         }
+        try:
+            meta = book.wldocument(librarian2=True).meta
+        except:
+            d['errors'].append('Nieprawidłowy dokument.')
+            return d
         thema = self.get_thema(meta)
         if thema:
             d['info'].append(mark_safe(
@@ -261,7 +265,7 @@ class Legimi(BasePublisher):
                 site_book.external_id = legimi_id
                 site_book.save(update_fields=['external_id'])
 
-        self.edit_sale(book)
+        self.edit_sale(site_book)
 
     def get_genre(self, wlbook):
         if wlbook.meta.legimi and wlbook.meta.legimi in self.CATEGORIES: