fix pdf rendering on production
[librarian.git] / librarian / formats / cover / evens / __init__.py
index 4207d46..9081464 100644 (file)
@@ -18,18 +18,21 @@ class EvensCover(Cover):
     title_top = 30
     logo_bottom = 100
 
-    def __init__(self, doc, format=None, width=None, height=None):
-        super(EvensCover, self).__init__(doc, format=format, width=width, height=height)
-        self.doc = doc
-
     def set_images(self, ctx):
-        cover_url = self.doc.meta.get(DCNS('relation.coverimage.url'))[0]
-        if cover_url.startswith('file://'):
-            cover_url = ctx.files_path + urllib.quote(cover_url[7:])
         try:
-            self.background_img = urlopen(cover_url)
-        except URLError:
-            raise BuildError('Cannot open the cover image: %s' % cover_url)
+            cover_url = self.doc.meta.get(DCNS('relation.coverimage.url'))[0]
+        except IndexError:
+            cover_url = False
+        if cover_url:
+            if cover_url.startswith('file://'):
+                cover_url = ctx.files_path + urllib.quote(cover_url[7:])
+            IMG_EXT = ('png', 'jpg', 'jpeg', 'gif')
+            if '.' not in cover_url or cover_url.rsplit('.', 1)[1].lower() not in IMG_EXT:
+                raise BuildError('Wrong cover format, should be PNG, JPG, GIF or SVG')
+            try:
+                self.background_img = urlopen(cover_url)
+            except URLError:
+                raise BuildError('Cannot open the cover image: %s' % cover_url)
 
         if getattr(ctx, 'cover_logo', None):
             self.logo_width = 150