1 # -*- coding: utf-8 -*-
3 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
6 from __future__ import unicode_literals
9 from PIL import Image, ImageFont, ImageDraw, ImageFilter
10 from six import BytesIO
11 from librarian import get_resource, OutputFile, URLOpener
15 """Gets metrics from an object, scaling it by a factor."""
16 def __init__(self, obj, scale):
18 self._scale = float(scale)
20 def __getattr__(self, name):
21 src = getattr(self._obj, name)
22 if src and self._scale:
23 return type(src)(self._scale * src)
28 class TextBox(object):
29 """Creates an Image with a series of centered strings."""
35 def __init__(self, max_width, max_height, padding_x=None, padding_y=None, bar_width=0, bar_color=None):
37 padding_x = self.SHADOW_X + self.SHADOW_BLUR
39 padding_y = self.SHADOW_Y + self.SHADOW_BLUR
41 self.max_width = max_width
42 self.bar_width = bar_width
43 self.bar_color = bar_color
44 self.max_text_width = max_width - 2 * padding_x - bar_width
45 self.padding_x = padding_x
46 self.padding_y = padding_y
47 self.height = padding_y
48 self.img = Image.new('RGBA', (max_width, max_height))
49 self.draw = ImageDraw.Draw(self.img)
50 self.shadow_img = None
51 self.shadow_draw = None
53 def skip(self, height):
54 """Skips some vertical space."""
57 def text(self, text, color='#000', font=None, line_height=20,
58 shadow_color=None, centering=True):
59 """Writes some centered text."""
60 text = re.sub(r'\s+', ' ', text)
62 if not self.shadow_img:
63 self.shadow_img = Image.new('RGBA', self.img.size)
64 self.shadow_draw = ImageDraw.Draw(self.shadow_img)
67 line_width = self.draw.textsize(line, font=font)[0]
68 while line_width > self.max_text_width:
69 parts = line.rsplit(' ', 1)
71 line_width = self.max_text_width
78 line_width = self.draw.textsize(line, font=font)[0]
79 line = line.strip() + ' '
82 pos_x = (self.max_width - line_width) // 2
84 pos_x = self.bar_width + self.padding_x
87 self.shadow_draw.text(
88 (pos_x + self.SHADOW_X, self.height + self.SHADOW_Y),
89 line, font=font, fill=shadow_color
92 self.draw.text((pos_x, self.height), line, font=font, fill=color)
93 self.height += line_height
95 text = text[len(line):]
98 """Creates the actual Image object."""
99 image = Image.new('RGBA', (self.max_width,
100 int(round(self.height + self.padding_y))))
103 shadow = self.shadow_img.filter(ImageFilter.BLUR)
104 image.paste(shadow, (0, 0), shadow)
105 image.paste(self.img, (0, 0), self.img)
107 image.paste(self.img, (0, 0))
112 """Abstract base class for cover images generator."""
115 background_color = '#fff'
116 background_img = None
119 author_margin_left = 20
120 author_margin_right = 20
122 author_color = '#000'
124 author_font_ttf = get_resource('fonts/DejaVuSerif.ttf')
125 author_font_size = 30
128 title_margin_left = 20
129 title_margin_right = 20
133 title_font_ttf = get_resource('fonts/DejaVuSerif.ttf')
138 uses_dc_cover = False
150 'JPEG': 'image/jpeg',
154 def __init__(self, book_info, format=None, width=None, height=None, cover_logo=None):
155 self.authors = [auth.readable() for auth in book_info.authors]
156 self.title = book_info.title
157 if format is not None:
160 self.height = int(round(height * self.width / width))
161 scale = max(float(width or 0) / self.width,
162 float(height or 0) / self.height)
166 self.scale_after = scale
168 def pretty_authors(self):
169 """Allows for decorating authors' names."""
170 return [self.authors]
172 def pretty_title(self):
173 """Allows for decorating title."""
177 metr = Metric(self, self.scale)
178 img = Image.new('RGB', (metr.width, metr.height),
179 self.background_color)
181 if self.background_img:
182 background = Image.open(self.background_img)
183 img.paste(background, None, background)
188 logo = Image.open(get_resource('res/wl-logo.png'))
191 int(round(logo.size[1] * metr.logo_width / logo.size[0]))
194 (metr.width - metr.logo_width) // 2,
195 img.size[1] - logo.size[1] - metr.logo_bottom
198 top = metr.author_top
200 metr.width - metr.author_margin_left - metr.author_margin_right,
204 author_font = ImageFont.truetype(
205 self.author_font_ttf, metr.author_font_size,
206 layout_engine=ImageFont.LAYOUT_BASIC)
207 for pa in self.pretty_authors():
208 tbox.text(pa, self.author_color, author_font, metr.author_lineskip,
210 text_img = tbox.image()
211 img.paste(text_img, (metr.author_margin_left, top), text_img)
213 top += text_img.size[1] + metr.title_top
215 metr.width - metr.title_margin_left - metr.title_margin_right,
218 title_font = ImageFont.truetype(
219 self.title_font_ttf, metr.title_font_size,
220 layout_engine=ImageFont.LAYOUT_BASIC)
221 tbox.text(self.pretty_title(), self.title_color, title_font,
222 metr.title_lineskip, self.title_shadow)
223 text_img = tbox.image()
224 img.paste(text_img, (metr.title_margin_left, top), text_img)
228 def final_image(self):
230 if self.scale_after != 1:
232 int(round(img.size[0] * self.scale_after)),
233 int(round(img.size[1] * self.scale_after))),
238 return self.mime_types[self.format]
241 return self.exts[self.format]
243 def save(self, *args, **kwargs):
245 'format': self.format,
248 default_kwargs.update(kwargs)
249 return self.final_image().save(*args, **default_kwargs)
251 def output_file(self, *args, **kwargs):
253 self.save(imgstr, *args, **kwargs)
254 return OutputFile.from_bytes(imgstr.getvalue())
257 class WLCover(Cover):
258 """Wolne Lektury cover without logos."""
262 author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
263 author_font_size = 20
265 author_centering = True
266 title_font_ttf = get_resource('fonts/DejaVuSerif-Bold.ttf')
269 title_box_width = 350
270 title_centering = True
273 box_bottom_margin = 100
288 box_position = 'middle'
289 background_color = '#444'
290 author_color = '#444'
291 background_img = get_resource('res/cover.png')
292 background_top = False
296 u'Starożytność': '#9e3610',
297 u'Średniowiecze': '#564c09',
298 u'Renesans': '#8ca629',
300 u'Oświecenie': '#f2802e',
301 u'Romantyzm': '#db4b16',
302 u'Pozytywizm': '#961060',
303 u'Modernizm': '#7784e0',
304 u'Dwudziestolecie międzywojenne': '#3044cf',
305 u'Współczesność': '#06393d',
307 set_title_color = True
309 kind_box_position = {
314 def __init__(self, book_info, format=None, width=None, height=None,
315 bleed=0, cover_logo=None):
316 super(WLCover, self).__init__(book_info, format=format, width=width,
319 self.box_position = book_info.cover_box_position or \
320 self.kind_box_position.get(book_info.kind, self.box_position)
322 if book_info.cover_bar_color == 'none':
325 self.bar_color = book_info.cover_bar_color or \
326 self.get_variable_color(book_info) or self.bar_color
328 if self.set_title_color:
329 self.title_color = self.get_variable_color(book_info) or self.title_color
332 self.box_top_margin += bleed
333 self.box_bottom_margin += bleed
334 self.bar_width += bleed
336 if book_info.cover_url:
337 url = book_info.cover_url
340 bg_src = URLOpener().open(url)
341 self.background_img = BytesIO(bg_src.read())
344 def get_variable_color(self, book_info):
345 return self.epoch_colors.get(book_info.epoch, None)
347 def pretty_authors(self):
348 return [a.upper() for a in self.authors]
350 def add_box(self, img):
351 if self.box_position == 'none':
354 metr = Metric(self, self.scale)
357 box = TextBox(metr.title_box_width, metr.height,
358 padding_y=metr.box_padding_y,
359 padding_x=metr.box_padding_x,
360 bar_width=metr.box_bar_width,
361 bar_color=self.bar_color,
363 author_font = ImageFont.truetype(
364 self.author_font_ttf, metr.author_font_size,
365 layout_engine=ImageFont.LAYOUT_BASIC)
366 for pa in self.pretty_authors():
367 box.text(pa, font=author_font, line_height=metr.author_lineskip,
368 color=self.author_color, shadow_color=self.author_shadow,
369 centering=self.author_centering
372 box.skip(metr.box_above_line)
374 (metr.box_line_left, box.height, metr.box_line_right, box.height),
375 fill=self.author_color, width=metr.box_line_width
377 box.skip(metr.box_below_line)
380 title_font = ImageFont.truetype(
381 self.title_font_ttf, metr.title_font_size,
382 layout_engine=ImageFont.LAYOUT_BASIC)
383 box.text(self.pretty_title(),
384 line_height=metr.title_lineskip,
386 color=self.title_color,
387 shadow_color=self.title_shadow,
388 centering=self.title_centering
391 box_img = box.image()
394 if self.box_position == 'bottom' or box_img.size[1] + metr.box_top_margin + metr.box_bottom_margin > metr.height:
395 box_top = metr.height - metr.box_bottom_margin - box_img.size[1]
396 elif self.box_position == 'top':
397 box_top = metr.box_top_margin
399 box_top = (metr.height - box_img.size[1]) // 2
401 box_left = metr.bar_width + (
402 metr.width - metr.bar_width - box_img.size[0]
405 # Draw the white box.
406 img_draw = ImageDraw.Draw(img)
411 box_left + box_img.size[0],
412 box_top + box_img.size[1]
416 # Paste the contents into the white box.
417 img.paste(box_img, (box_left, box_top), box_img)
418 if self.box_bar_width:
423 box_left + metr.box_bar_width,
424 box_top + box_img.size[1]
430 def add_cut_lines(self, img):
434 metr = Metric(self, self.scale)
435 draw = ImageDraw.Draw(img)
436 for corner_x, corner_y in (
437 (0, 0), (metr.width, 0),
438 (0, metr.height), (metr.width, metr.height)
440 dir_x = 1 if corner_x == 0 else -1
441 dir_y = 1 if corner_y == 0 else -1
442 for offset in (-1, 0, 1):
446 corner_y + dir_y * metr.bleed + offset,
447 corner_x + dir_x * metr.bleed * line_ratio,
448 corner_y + dir_y * metr.bleed + offset
450 fill='black' if offset == 0 else 'white',
455 corner_x + dir_x * metr.bleed + offset,
457 corner_x + dir_x * metr.bleed + offset,
458 corner_y + dir_y * metr.bleed * line_ratio
460 fill='black' if offset == 0 else 'white',
466 metr = Metric(self, self.scale)
467 img = Image.new('RGB', (metr.width, metr.height),
468 self.background_color)
469 draw = ImageDraw.Draw(img)
471 draw.rectangle((0, 0, metr.bar_width, metr.height),
474 if self.background_img:
475 src = Image.open(self.background_img)
476 trg_size = (metr.width - metr.bar_width, metr.height)
477 if src.size[0] * trg_size[1] < src.size[1] * trg_size[0]:
480 int(round(src.size[1] * trg_size[0] / src.size[0]))
482 if self.background_top:
485 cut = (resized[1] - trg_size[1]) // 2
486 src = src.resize(resized, Image.ANTIALIAS)
487 src = src.crop((0, cut, src.size[0], src.size[1] - cut))
490 int(round(src.size[0] * trg_size[1] / src.size[1])),
493 cut = (resized[0] - trg_size[0]) // 2
494 src = src.resize(resized, Image.ANTIALIAS)
495 src = src.crop((cut, 0, src.size[0] - cut, src.size[1]))
497 img.paste(src, (metr.bar_width, 0))
500 img = self.add_box(img)
502 # img = self.add_cut_lines(img)
507 class WLNoBoxCover(WLCover):
508 def add_box(self, img):
512 class LogoWLCover(WLCover):
515 gradient_logo_height = 57
516 gradient_logo_max_width = 200
517 gradient_logo_margin_right = 30
518 gradient_logo_spacing = 40
519 gradient_color = '#000'
520 gradient_opacity = .6
522 'res/wl-logo-white.png',
523 'res/fnp-logo-white.png',
526 annotation_height = 10
529 gradient_logo_centering = False
532 def __init__(self, book_info, *args, cover_logo=None, **kwargs):
533 super(LogoWLCover, self).__init__(book_info, *args, **kwargs)
534 self.gradient_height += self.bleed
535 self.gradient_logo_margin_right += self.bleed
537 self.additional_cover_logos = [
538 BytesIO(URLOpener().open(cover_logo_url).read())
539 for cover_logo_url in book_info.cover_logo_urls
541 self.end_cover_logos = []
543 self.additional_cover_logos.append(
544 open(cover_logo, 'rb')
548 def has_gradient_logos(self):
549 return self.gradient_logos or self.additional_cover_logos or self.end_cover_logos or self.annotation
551 def add_gradient_logos(self, img, metr):
552 gradient = Image.new(
554 (metr.width - metr.bar_width, metr.gradient_height),
557 gradient_mask = Image.new(
559 (metr.width - metr.bar_width, metr.gradient_height)
561 draw = ImageDraw.Draw(gradient_mask)
562 for line in range(0, metr.gradient_easing):
564 (0, line, metr.width - metr.bar_width, line),
566 255 * self.gradient_opacity * line / metr.gradient_easing
570 0, metr.gradient_easing,
571 metr.width - metr.bar_width, metr.gradient_height
572 ), fill=int(255 * self.gradient_opacity))
576 (metr.bar_width, metr.height - metr.gradient_height),
580 cursor = metr.width - metr.gradient_logo_margin_right
582 cursor = metr.gradient_logo_margin_right
585 - metr.gradient_height / 2
586 - metr.gradient_logo_height / 2 - metr.bleed / 2
590 get_resource(logo_path)
591 for logo_path in self.gradient_logos
594 logos = self.additional_cover_logos + logos + self.end_cover_logos
600 Image.open(logo_bytes).convert('RGBA')
601 for logo_bytes in logos
604 # See if logos fit into the gradient. If not, scale down accordingly.
608 - 2 * metr.gradient_logo_margin_right
612 metr.gradient_logo_max_width,
613 logo.size[0] * metr.gradient_logo_height / logo.size[1]
616 taken_space = sum(widths) + (len(logos) - 1) * metr.gradient_logo_spacing
617 if taken_space > space_for_logos:
618 logo_scale = space_for_logos / taken_space
621 if self.gradient_logo_centering:
622 cursor += int((space_for_logos - taken_space) / 2)
624 space_for_logos / taken_space
625 if taken_space > space_for_logos else 1
627 #logo_top += int(metr.gradient_logo_height * (1 - logo_scale) / 2)
629 for i, logo in enumerate(logos):
634 L_top = logo_top + int(metr.gradient_logo_height * (1 - L_scale) / 2)
638 int(round(widths[i] * L_scale)),
640 logo.size[1] * widths[i] / logo.size[0] * L_scale
645 cursor -= logo.size[0]
652 #int(round(logo_top + (metr.gradient_logo_height - logo.size[1]) * L_scale / 2))
657 cursor -= int(round(metr.gradient_logo_spacing * logo_scale))
659 cursor += int(round(metr.gradient_logo_spacing * logo_scale)) + logo.size[0]
662 img2 = Image.new('RGBA', (metr.height, metr.height), color=None)
663 draw = ImageDraw.Draw(img2)
664 author_font = ImageFont.truetype(
665 self.author_font_ttf, metr.annotation_height,
666 layout_engine=ImageFont.LAYOUT_BASIC)
667 draw.text((self.annotation_height, self.annotation_height), self.annotation, font=author_font, fill='#FFFFFF')
669 img2 = img2.rotate(90)
672 img.alpha_composite(img2, (0, 0))
673 img = img.convert('RGB')
678 img = super(LogoWLCover, self).image()
679 metr = Metric(self, self.scale)
680 if self.has_gradient_logos:
681 img = self.add_gradient_logos(img, metr)
685 class LegimiCover(LogoWLCover):
691 author_font_ttf = get_resource('fonts/Lora-Regular.ttf')
692 author_font_size = 15
693 author_lineskip = 19.5
694 author_centering = False
695 title_font_ttf = get_resource('fonts/Lora-Bold.ttf')
697 title_lineskip = 19.5
698 title_centering = False
700 title_box_width = 210
702 box_bottom_margin = 20
704 box_padding_y = 10 # do baseline
718 box_position = 'bottom'
719 background_color = '#444'
720 author_color = '#000'
722 set_title_color = False
724 kind_box_position = {}
726 box_bottom_margin_logos_add = 10
729 gradient_logo_height = 20
730 gradient_logo_max_width = 200
731 gradient_logo_margin_right = 10
732 gradient_logo_spacing = 20
733 gradient_color = '#000'
734 gradient_opacity = .5
736 'res/wl-logo-white.png',
739 gradient_logo_centering = True
740 background_top = True
743 'Artykuł': '#bf001a',
744 'Artykuł naukowy': '#bf001a',
745 'Dziennik': '#bf001a',
747 'Felieton': '#bf001a',
748 'Kronika': '#bf001a',
750 'Manifest': '#bf001a',
752 'Pamiętnik': '#bf001a',
753 'Poradnik': '#bf001a',
754 'Praca naukowa': '#bf001a',
755 'Publicystyka': '#bf001a',
756 'Reportaż': '#bf001a',
757 'Reportaż podróżniczy': '#bf001a',
758 'Rozprawa': '#bf001a',
759 'Rozprawa polityczna': '#bf001a',
760 'Traktat': '#bf001a',
768 def __init__(self, *args, **kwargs):
769 super().__init__(*args, **kwargs)
770 if self.has_gradient_logos:
771 self.box_bottom_margin += self.box_bottom_margin_logos_add
773 def pretty_authors(self):
776 def get_variable_color(self, book_info):
777 for genre in book_info.genres:
778 if genre in self.genre_colors:
779 return self.genre_colors[genre]
780 for kind in book_info.kinds:
781 if kind in self.kind_colors:
782 return self.kind_colors[kind]
784 class LegimiCornerCover(LegimiCover):
787 corner_image = get_resource('res/book-band.png')
790 image = super().image()
791 metr = Metric(self, self.scale)
793 if self.corner_image:
794 corner = Image.open(self.corner_image).convert('RGBA')
795 corner = corner.resize(
797 int(round(metr.corner_width)),
798 int(round(corner.size[1] / corner.size[0] * metr.corner_width))
801 image.paste(corner, (
802 metr.width - int(round(metr.corner_width)),
807 class LegimiAudiobookCover(LegimiCornerCover):
809 corner_image = get_resource('res/audiobook-sticker.png')
813 class EbookpointCover(LogoWLCover):
814 gradient_logo_height = 58
815 gradient_logo_spacing = 25
817 'res/ebookpoint-logo-white.png',
818 'res/wl-logo-white.png',
819 'res/fnp-logo-white.png',
823 class VirtualoCover(Cover):
832 class PrestigioCover(Cover):
835 background_img = get_resource('res/cover-prestigio.png')
838 author_margin_left = 118
839 author_margin_right = 62
841 author_color = '#fff'
842 author_shadow = '#000'
843 author_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
844 author_font_size = 50
847 title_margin_left = 118
848 title_margin_right = 62
851 title_shadow = '#000'
852 title_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
855 def pretty_title(self):
856 return u"„%s”" % self.title
859 class BookotekaCover(Cover):
862 background_img = get_resource('res/cover-bookoteka.png')
865 author_margin_left = 307
866 author_margin_right = 233
867 author_lineskip = 156
868 author_color = '#d9d919'
869 author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
870 author_font_size = 130
873 title_margin_left = 307
874 title_margin_right = 233
876 title_color = '#d9d919'
877 title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
878 title_font_size = 140
883 class GandalfCover(Cover):
886 background_img = get_resource('res/cover-gandalf.png')
887 author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
888 author_font_size = 30
889 title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
896 class KMLUCover(LogoWLCover):
897 gradient_logo_height = 58
898 gradient_logo_spacing = 25
900 'res/kmlu-logo-white.png',
901 'res/wl-logo-white.png',
902 'res/fnp-logo-white.png',
906 class MPWCover(LogoWLCover):
907 gradient_logo_height = 57
908 gradient_logo_spacing = 25
910 'res/mpw-logo-white.png',
911 'res/wl-logo-white.png',
912 'res/fnp-logo-white.png',
916 class AtriumCover(LogoWLCover):
917 gradient_logo_height = 58
918 gradient_logo_spacing = 25
920 'res/atrium-logo.png',
921 'res/wl-logo-white.png',
922 'res/fnp-logo-white.png',
926 class BNCover(LogoWLCover):
928 'res/dofinansowano.png',
931 'res/wl-logo-white.png',
933 # annotation = 'Zadanie „Udostępnienie publikacji w formatach cyfrowych” w ramach Narodowego Programu Rozwoju Czytelnictwa. Dofinansowano ze środków Ministra Kultury, Dziedzictwa Narodowego i Sportu.'
937 'default': LogoWLCover,
940 'atrium': AtriumCover,
942 'legimi': LegimiCover,
943 'legimi-corner': LegimiCornerCover,
944 'legimi-audiobook': LegimiAudiobookCover,
948 def make_cover(book_info, *args, **kwargs):
949 if 'cover_class' in kwargs:
950 cover_class_name = kwargs.pop('cover_class')
952 cover_class_name = book_info.cover_class
953 cover_class = COVER_CLASSES[cover_class_name]
954 return cover_class(book_info, *args, **kwargs)