X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/25af49b6c63e1c005129856e107143864ad5b245..0560b4a83f947a4d08f087d85759f05516f6e580:/librarian/formats/cover/evens/__init__.py diff --git a/librarian/formats/cover/evens/__init__.py b/librarian/formats/cover/evens/__init__.py index 4207d46..9081464 100644 --- a/librarian/formats/cover/evens/__init__.py +++ b/librarian/formats/cover/evens/__init__.py @@ -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